Update app.py

app.py

@@ -26,9 +26,7 @@ questions = [
     "What treatment plan do you recommend? Choose only from Options: (Scaling, Filling, Pulp therapy/RCT, Extraction, Medication, Referral) and nothing else"
 ]
 
-
 oral_health_assessment_form = [
-    "Doctor’s Name",
     "Patient’s Name",
     "Age",
     "Gender",
@@ -90,21 +88,24 @@ def fill_textboxes(context):
 supabase: Client = create_client(url, key)
 
 # Main Gradio app function
-def main(audio):
+def main(audio, doctor_name, location):
     context = transcribe_audio(audio)
     
     if "Error" in context:
-        return [context] * (len(oral_health_assessment_form) - 1)
+        return [context] * (len(oral_health_assessment_form) + 1)
     
     answers = fill_textboxes(context)
-    answers.insert(0, "")  # for "Child's Name"
-    answers.insert(0, "Dr.Alvi")  # for "Doctor’s Name"
+    answers.insert(0, "")  # for "Patient’s Name"
+    answers.append("")  # for "Referred to"
+    answers.extend([doctor_name, location])
     
     return answers
 
 def save_answers(*args):
     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["Doctor’s Name"] = args[-2]
+    answers_dict["Location"] = args[-1]
     answers_dict["Submission Date and Time"] = current_datetime
     print("Saved answers:", answers_dict)
     
@@ -158,25 +159,32 @@ def gradio_download():
 # Create the Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# OHA Form Filler App")
-    audio_input = gr.Audio(type="filepath", label="Record your audio", elem_id="audio_input")
-    transcribe_button = gr.Button("Transcribe and Generate Form", elem_id="transcribe_button")
     
-    with gr.Row(elem_id="textboxes_row"):
-        with gr.Column():
-            textboxes_left = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(0, 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)
+    with gr.Page():
+        doctor_name = gr.Textbox(label="Doctor’s Name", value="Dr. Alvi", interactive=True)
+        location = gr.Textbox(label="Location", value="", interactive=True)
+        next_button = gr.Button("Next")
     
-    submit_button = gr.Button("Submit", elem_id="submit_button")
-    output_html = gr.HTML(label="Submitted Answers")
-
-    transcribe_button.click(fn=main, inputs=audio_input, outputs=textboxes_left + textboxes_right)
-    submit_button.click(fn=save_answers, inputs=textboxes_left + textboxes_right + [dropdown_referred], outputs=output_html)
-
-    download_button = gr.Button("Download Table as CSV")
-    download_csv_output = gr.File(label="Download CSV")
-
+    with gr.Page() as main_page:
+        audio_input = gr.Audio(type="filepath", label="Record your audio", elem_id="audio_input")
+        transcribe_button = gr.Button("Transcribe and Generate Form", elem_id="transcribe_button")
+        
+        with gr.Row(elem_id="textboxes_row"):
+            with gr.Column():
+                textboxes_left = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(0, 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)
+        
+        submit_button = gr.Button("Submit", elem_id="submit_button")
+        output_html = gr.HTML(label="Submitted Answers")
+        
+        download_button = gr.Button("Download Table as CSV")
+        download_csv_output = gr.File(label="Download CSV")
+    
+    next_button.click(fn=None, inputs=[doctor_name, location], outputs=main_page)
+    transcribe_button.click(fn=main, inputs=[audio_input, doctor_name, location], outputs=textboxes_left + textboxes_right + [dropdown_referred])
+    submit_button.click(fn=save_answers, inputs=textboxes_left + textboxes_right + [dropdown_referred, doctor_name, location], outputs=output_html)
     download_button.click(fn=gradio_download, inputs=[], outputs=download_csv_output)
 
 # Launch the app