Update app.py

app.py

@@ -120,16 +120,19 @@ 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) -> list:
+def main(audio: str, doctor_name: str, location: str) -> dict:
     context = transcribe_audio(audio)
     
     if "Error" in context:
-        return [context] * len(oral_health_assessment_form)  # Adjust for the number of fields
+        return {field: context for field in form_fields}  # Adjust for the number of fields
     
     answers = fill_textboxes(context)
-    answers_list = [doctor_name, location, ""]  # Initial patient name field empty
-    answers_list += [answers.get(field, "") for field in form_fields]
-    return answers_list
+    # Fill in the textboxes with the generated answers
+    answers.update({
+        "Doctor’s Name": doctor_name,
+        "Location": location
+    })
+    return answers
 
 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()
@@ -239,11 +242,27 @@ with gr.Blocks() as demo:
                     textboxes_right.append(gr.Dropdown(choices=["NONE", "ORAL MEDICINE & RADIOLOGY", "PERIODONTICS", "ORAL SURGERY", "CONSERVATIVE AND ENDODONTICS", "PROSTHODONTICS", "PEDODONTICS", "ORTHODONTICS"], value="NONE", label="Referred to", interactive=True))
                     textboxes_right.append(gr.Dropdown(choices=["+", "++", "+++"], label="Calculus", interactive=True))
                     textboxes_right.append(gr.Dropdown(choices=["-", "+", "++", "+++"], label="Stains", interactive=True))
-                    textboxes_right.append(gr.Dropdown(choices=["Scaling", "Filling", "Pulp therapy/RCT", "Extraction", "Medication"], label="Treatment plan", interactive=True))
+                    textboxes_right.append(gr.Dropdown(choices=["Scaling", "Filling", "Pulp therapy/RCT", "Extraction", "Medication", "Referral"], label="Treatment plan", interactive=True))
+
             oha_output = gr.Textbox(label="OHA Output", value="", interactive=False)
             save_button = gr.Button("Save to Supabase", elem_id="save_button", interactive=True)
-            transcribe_button.click(fn=main, inputs=[audio_input, doctor_name_display, location_display], outputs=[oha_output])
-            save_button.click(fn=save_answers, inputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right, outputs=[oha_output])
+            
+            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)
+                answers.update({
+                    "Doctor’s Name": doctor_name,
+                    "Location": location
+                })
+                return answers
+            
+            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)
+
+            transcribe_button.click(fn=handle_transcription, inputs=[audio_input, doctor_name_display, location_display], outputs=textboxes_left + textboxes_right + [oha_output])
+            save_button.click(fn=save_to_supabase, inputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right, outputs=oha_output)
         
         with gr.Tab("Download Data"):
             download_button = gr.Button("Download CSV")