Update app.py

app.py

@@ -120,21 +120,33 @@ def fill_textboxes(context: str) -> dict:
 supabase: Client = create_client(url, key)
 
 # Main Gradio app function
-def main(audio: str, doctor_name: str, location: str) -> dict:
+def handle_transcription(audio: str, doctor_name: str, location: str) -> dict:
     context = transcribe_audio(audio)
-    
     if "Error" in context:
-        return {field: context for field in form_fields}  # Adjust for the number of fields
+        return {field: context for field in form_fields}
     
     answers = fill_textboxes(context)
-    # Fill in the textboxes with the generated answers
     answers.update({
         "Doctor’s Name": doctor_name,
         "Location": location
     })
-    return answers
+    
+    return {
+        "Age": answers.get("Age", ""),
+        "Gender": answers.get("Gender", ""),
+        "Chief complaint": answers.get("Chief complaint", ""),
+        "Medical history": answers.get("Medical history", ""),
+        "Dental history": answers.get("Dental history", ""),
+        "Clinical Findings": answers.get("Clinical Findings", ""),
+        "Referred to": answers.get("Referred to", ""),
+        "Treatment plan": answers.get("Treatment plan", ""),
+        "Calculus": answers.get("Calculus", ""),
+        "Stains": answers.get("Stains", ""),
+        "Doctor’s Name": doctor_name,
+        "Location": location
+    }
 
-def save_answers(doctor_name: str, location: str, patient_name: str, age: str, gender: str, chief_complaint: str, medical_history: str, dental_history: str, clinical_findings: str, treatment_plan: str, referred_to: str, Calculus: str, stains: str) -> str:
+def save_answers(doctor_name: str, location: str, patient_name: str, age: str, gender: str, chief_complaint: str, medical_history: str, dental_history: str, clinical_findings: str, treatment_plan: str, referred_to: str, calculus: str, stains: str) -> str:
     current_datetime = datetime.now().isoformat()
     answers_dict = {
         "Doctor’s Name": doctor_name,
@@ -148,7 +160,7 @@ def save_answers(doctor_name: str, location: str, patient_name: str, age: str, g
         "Clinical Findings": clinical_findings,
         "Treatment plan": treatment_plan,
         "Referred to": referred_to,
-        "Calculus": Calculus,
+        "Calculus": calculus,
         "Stains": stains,
         "Submission Date and Time": current_datetime
     }
@@ -250,13 +262,30 @@ with gr.Blocks() as demo:
             def handle_transcription(audio, doctor_name, location):
                 context = transcribe_audio(audio)
                 if "Error" in context:
-                    return {field: context for field in form_fields}  # Return error message for all fields
-                answers = fill_textboxes(context)
+                    # Return error message for all fields
+                    return [context] * len(textboxes_left + textboxes_right) + [context]
+                
+                answers = handle_transcription(audio, doctor_name, location)
                 answers.update({
                     "Doctor’s Name": doctor_name,
                     "Location": location
                 })
-                return answers
+            
+                # Return the answers in the order of textboxes and dropdowns
+                return [
+                    answers.get("Age", ""),
+                    answers.get("Gender", ""),
+                    answers.get("Chief complaint", ""),
+                    answers.get("Medical history", ""),
+                    answers.get("Dental history", ""),
+                    answers.get("Clinical Findings", ""),
+                    answers.get("Referred to", ""),
+                    answers.get("Treatment plan", ""),
+                    answers.get("Calculus", ""),
+                    answers.get("Stains", ""),
+                    doctor_name,
+                    location
+                ]
             
             def save_to_supabase(doctor_name, location, patient_name, age, gender, chief_complaint, medical_history, dental_history, clinical_findings, treatment_plan, referred_to, calculus, stains):
                 return save_answers(doctor_name, location, patient_name, age, gender, chief_complaint, medical_history, dental_history, clinical_findings, treatment_plan, referred_to, calculus, stains)