Update app.py

app.py

@@ -159,33 +159,34 @@ def gradio_download():
 with gr.Blocks() as demo:
     gr.Markdown("# OHA Form Filler App")
     
-    # Initial page for Doctor's Name and Location
-    with gr.Page(title="Doctor Info"):
-        doctor_name_input = gr.Textbox(label="Doctor's Name", interactive=True)
-        location_input = gr.Textbox(label="Location", interactive=True)
-        next_button = gr.Button("Next")
-    
-    with gr.Page(title="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)
+    with gr.Tabs():
+        with gr.TabItem("Doctor Info"):
+            doctor_name_input = gr.Textbox(label="Doctor's Name", interactive=True)
+            location_input = gr.Textbox(label="Location", 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, 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)
-
-    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)
+        with gr.TabItem("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)
+            
+            submit_button = gr.Button("Submit", elem_id="submit_button")
+            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)
+
+        with gr.TabItem("Download"):
+            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()