Update app.py

app.py

@@ -159,44 +159,38 @@ def gradio_download():
 with gr.Blocks() as demo:
     gr.Markdown("# OHA Form Filler App")
     
-    # Default page for Doctor's Name and Location
-    with gr.Page(title="Doctor Info"):
-        with gr.Row():
+    with gr.Tabs() as tabs:
+        # Default tab for Doctor's Name and Location
+        with gr.Tab("Doctor Info"):
             doctor_name_input = gr.Textbox(label="Doctor's Name", interactive=True)
             location_input = gr.Textbox(label="Location", interactive=True)
-        submit_info_button = gr.Button("Submit")
-        info_output = gr.HTML()
-
-        def handle_info_submission(doctor_name, location):
-            return f"Doctor: {doctor_name}, Location: {location}"
-
-        submit_info_button.click(fn=handle_info_submission, inputs=[doctor_name_input, location_input], outputs=info_output)
-
-    # Page for OHA Form
-    with gr.Page(title="OHA Form"):
-        with gr.Row():
-            audio_input = gr.Audio(type="filepath", label="Record your audio")
-            transcribe_button = gr.Button("Transcribe and Generate Form")
+            submit_button = gr.Button("Submit")
+            info_output = gr.HTML(label="Submitted Info")
+            
+            submit_button.click(fn=lambda name, loc: f"Doctor's Name: {name}<br>Location: {loc}", inputs=[doctor_name_input, location_input], outputs=info_output)
         
-        with gr.Row():
-            with gr.Column():
-                textboxes_left = [gr.Textbox(label=oral_health_assessment_form[i], value="", interactive=True) for i in range(2, 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)
+        # Second tab for OHA Form
+        with gr.Tab("OHA Form"):
+            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(2, 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)
+            
+            transcribe_button.click(fn=main, inputs=[audio_input, doctor_name_input, location_input], outputs=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=textboxes_left + textboxes_right + [dropdown_referred], outputs=output_html)
         
-        submit_button = gr.Button("Submit")
-        output_html = gr.HTML(label="Submitted Answers")
-
-        transcribe_button.click(fn=main, inputs=[audio_input, doctor_name_input, location_input], outputs=textboxes_left + textboxes_right)
-        submit_button.click(fn=save_answers, inputs=textboxes_left + textboxes_right + [dropdown_referred], outputs=output_html)
-
-    # Page for downloading CSV
-    with gr.Page(title="Download CSV"):
-        download_button = gr.Button("Download Table as CSV")
-        download_csv_output = gr.File(label="Download CSV")
-
-        download_button.click(fn=gradio_download, inputs=[], outputs=download_csv_output)
+        # Third tab for CSV download
+        with gr.Tab("Download CSV"):
+            download_button = gr.Button("Download Table as CSV")
+            download_csv_output = gr.File(label="Download CSV")
+            download_button.click(fn=gradio_download, inputs=[], outputs=download_csv_output)
 
 # Launch the app
 demo.launch()