Update app.py

app.py

@@ -99,10 +99,22 @@ def main(audio, doctor_name, location):
     
     return answers
 
-def save_answers(*args):
+def save_answers(doctor_name, location, patient_name, age, gender, chief_complaint, medical_history, dental_history, clinical_findings, treatment_plan, referred_to):
     current_datetime = datetime.now().isoformat()
-    answers_dict = {oral_health_assessment_form[i]: args[i] for i in range(len(oral_health_assessment_form))}
-    answers_dict["Submission Date and Time"] = current_datetime
+    answers_dict = {
+        "Doctor’s Name": doctor_name,
+        "Location": location,
+        "Patient’s Name": patient_name,
+        "Age": age,
+        "Gender": gender,
+        "Chief complaint": chief_complaint,
+        "Medical history": medical_history,
+        "Dental history": dental_history,
+        "Clinical Findings": clinical_findings,
+        "Treatment plan": treatment_plan,
+        "Referred to": referred_to,
+        "Submission Date and Time": current_datetime
+    }
     print("Saved answers:", answers_dict)
     
     # Insert data into Supabase
@@ -178,7 +190,8 @@ with gr.Blocks() as demo:
                 with gr.Column():
                     doctor_name_display = gr.Textbox(label="Doctor’s Name", value="", interactive=False)
                     location_display = gr.Textbox(label="Location", value="", interactive=False)
-                    textboxes_left = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(2, len(oral_health_assessment_form)//2)]
+                    patient_name_input = gr.Textbox(label="Patient’s Name", value="", interactive=True)
+                    textboxes_left = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(3, len(oral_health_assessment_form)//2)]
                 with gr.Column():
                     textboxes_right = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(len(oral_health_assessment_form)//2, len(oral_health_assessment_form)-1)]
                     dropdown_referred = gr.Dropdown(choices=["NONE","ORAL MEDICINE & RADIOLOGY", "PERIODONTICS", "ORAL SURGERY", "CONSERVATIVE AND ENDODONTICS", "PROSTHODONTICS", "PEDODONTICS", "ORTHODONTICS"], label="Referred to", interactive=True)
@@ -187,18 +200,18 @@ with gr.Blocks() as demo:
                 context = transcribe_audio(audio)
                 
                 if "Error" in context:
-                    return [context] * (len(oral_health_assessment_form) - 2)  # Adjust for the number of fields
+                    return [context] * (len(oral_health_assessment_form) - 3)  # Adjust for the number of fields
                 
                 answers = fill_textboxes(context)
                 answers.insert(0, location)  # Add location
                 answers.insert(0, doctor_name)  # Add doctor name
                 
-                return [doctor_name, location] + answers
+                return [doctor_name, location] + [""] + answers  # Patient name is left empty
 
-            transcribe_button.click(fn=update_textboxes, inputs=[audio_input, doctor_name_input, location_input], outputs=[doctor_name_display, location_display] + textboxes_left + textboxes_right)
+            transcribe_button.click(fn=update_textboxes, inputs=[audio_input, doctor_name_input, location_input], outputs=[doctor_name_display, location_display] + [patient_name_input] + textboxes_left + textboxes_right)
             submit_button = gr.Button("Submit", elem_id="submit_button")
             output_html = gr.HTML(label="Submitted Answers")
-            submit_button.click(fn=save_answers, inputs=[doctor_name_display, location_display] + textboxes_left + textboxes_right + [dropdown_referred], outputs=output_html)
+            submit_button.click(fn=save_answers, inputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right + [dropdown_referred], outputs=output_html)
         
         # Third tab for CSV download
         with gr.Tab("Download CSV"):