Update app.py

app.py

@@ -711,12 +711,12 @@ with gr.Blocks(css=css) as demo:
                     generate_with_audio_button = gr.Button("Generate Report", variant="primary")
 
             # Shared output containers
-            audio_patient_name_box = gr.Textbox(label="Patient Name", interactive=True, placeholder="Generated Patient Name", lines=1)
+            patient_name_box_text = gr.Textbox(label="Patient Name", interactive=True, placeholder="Generated Patient Name", lines=1)
             with gr.Row():
                 with gr.Column():
-                    audio_soap_report_box = gr.Textbox(label="SOAP Report", interactive=True, placeholder="Generated SOAP Report", lines=10)
+                    soap_report_box_text = gr.Textbox(label="SOAP Report", interactive=True, placeholder="Generated SOAP Report", lines=10)
                 with gr.Column():
-                    audio_sbar_report_box = gr.Textbox(label="SBAR Report", interactive=True, placeholder="Generated SBAR Report", lines=10)
+                    sbar_report_box_text = gr.Textbox(label="SBAR Report", interactive=True, placeholder="Generated SBAR Report", lines=10)
 
             audio_doctor_recommendations_box = gr.Textbox(label="Doctor Recommendations", interactive=False, placeholder="Recommendations", lines=5)
             audio_transcription_box = gr.Textbox(label="Transcription Text", interactive=False, placeholder="Transcribed Text", lines=5)
@@ -726,14 +726,27 @@ with gr.Blocks(css=css) as demo:
                 fn=report_audio,
                 inputs=[audio_file, transcription_service],
                 outputs=[
-                    audio_patient_name_box,
-                    audio_soap_report_box,
-                    audio_sbar_report_box,
+                    patient_name_box_text,
+                    soap_report_box_text,
+                    sbar_report_box_text,
                     audio_doctor_recommendations_box,
                     audio_transcription_box
                 ]
             )
 
+            # Add Save Report Button
+            with gr.Row():
+                save_button = gr.Button("Save Report", variant="secondary")
+                save_message = gr.Textbox(label="Save Status", interactive=False, placeholder="Status of the save operation", lines=1)
+
+            # Click event for Save Report Button using `patient_name_box_text` as the file name
+            save_button.click(
+                fn=save_reports_main,
+                inputs=[patient_name_box_text, soap_report_box_text, sbar_report_box_text],
+                outputs=[save_message]
+            )
+
+
         # Tab for generating from text input
         with gr.Tab("From Transcript"):
             with gr.Column():
@@ -762,17 +775,18 @@ with gr.Blocks(css=css) as demo:
                         doctor_recommendations_box_text
                     ]
                 )
-        # Add Save Report Button
-        with gr.Row():
-            save_button = gr.Button("Save Report", variant="secondary")
-            save_message = gr.Textbox(label="Save Status", interactive=False, placeholder="Status of the save operation", lines=1)
-
-        # Click event for Save Report Button using `patient_name_box_text` as the file name
-        save_button.click(
-            fn=save_reports_main,
-            inputs=[patient_name_box_text, soap_report_box_text, sbar_report_box_text],
-            outputs=[save_message]
-        )
+                
+            # Add Save Report Button
+            with gr.Row():
+                save_button = gr.Button("Save Report", variant="secondary")
+                save_message = gr.Textbox(label="Save Status", interactive=False, placeholder="Status of the save operation", lines=1)
+
+            # Click event for Save Report Button using `patient_name_box_text` as the file name
+            save_button.click(
+                fn=save_reports_main,
+                inputs=[patient_name_box_text, soap_report_box_text, sbar_report_box_text],
+                outputs=[save_message]
+            )
 
 # Page 2----------------------------------------------------------------------
 ####|
@@ -891,3 +905,4 @@ with gr.Blocks(css=css) as demo:
                 docs2_delete_button.click(delete2_docs, inputs=None, outputs=docs2_delete_output)
 
 demo.launch(debug=True)
+