Update app.py

app.py

@@ -227,36 +227,20 @@ with gr.Blocks() as demo:
                     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)-4)]
-                    clinical_findings_input = gr.Textbox(label="Clinical Findings", interactive=True)
-                    dropdown_referred = gr.Dropdown(choices=["NONE", "ORAL MEDICINE & RADIOLOGY", "PERIODONTICS", "ORAL SURGERY", "CONSERVATIVE AND ENDODONTICS", "PROSTHODONTICS", "PEDODONTICS", "ORTHODONTICS"], value="NONE", label="Referred to", interactive=True)
-                    dropdown_Calculus = gr.Dropdown(choices=["+", "++", "+++"], label="Calculus", interactive=True)
-                    dropdown_stains = gr.Dropdown(choices=["+", "++", "+++"], label="Stains", interactive=True)
-                    dropdown_treatment_plan = gr.Dropdown(choices=["Scaling", "Filling", "Pulp therapy/RCT", "Extraction", "Medication", "Referral"], label="Treatment plan", interactive=True)
-
-            transcribe_button.click(
-                fn=main, 
-                inputs=[audio_input, doctor_name_input, location_input], 
-                outputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right + [clinical_findings_input, dropdown_referred, dropdown_treatment_plan, dropdown_Calculus, dropdown_stains]
-            )
-            
-            save_button = gr.Button("Save")
-            save_output = gr.HTML()
-            
-            save_button.click(
-                fn=save_answers,
-                inputs=[doctor_name_display, location_display, patient_name_input] + textboxes_left + textboxes_right + [clinical_findings_input, dropdown_referred, dropdown_treatment_plan, dropdown_Calculus, dropdown_stains],
-                outputs=save_output
-            )
-
+                    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)-3)]
+                    textboxes_right.append(gr.Textbox(label="Clinical Findings", interactive=True))
+                    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", "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])
+        
+        with gr.Tab("Download Data"):
             download_button = gr.Button("Download CSV")
-            download_file = gr.File(label="Download CSV")
-            
-            download_button.click(
-                fn=gradio_download,
-                inputs=[],
-                outputs=download_file
-            )
+            download_output = gr.File()
+            download_button.click(fn=gradio_download, outputs=download_output)
 
-# Launch the Gradio app
 demo.launch()