Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os
 import shutil
 import openai
@@ -243,6 +242,10 @@ def generate_report(input_text: str = None, file_path: str = None) -> AI_Medical
         print(f"An error occurred: {e}")
         return None
 
+# wrapper function for audio
+def report_audio(audio_file: str = None, transcription_service: str = "OpenAI"):
+    return report_main(audio_file=audio_file,transcription_service=transcription_service)
+
 # driver function for making reports
 def report_main(input_text: str = None, audio_file: str = None, transcription_service: str = "OpenAI"):
     """
@@ -275,11 +278,11 @@ def report_main(input_text: str = None, audio_file: str = None, transcription_se
     elif audio_file:
         # Use selected transcription service for audio input
         if transcription_service == "AssemblyAI":
-            transcription_text = assemblyai_STT(audio_file)
+            transcription_text += assemblyai_STT(audio_file)
             report = generate_report(input_text=transcription_text)
             # print(transcription_text)
         elif transcription_service == "OpenAI":
-            transcription_text = openai_STT(audio_file)
+            transcription_text += openai_STT(audio_file)
             report = generate_report(input_text=transcription_text)
             # print(transcription_text)
         else:
@@ -290,7 +293,6 @@ def report_main(input_text: str = None, audio_file: str = None, transcription_se
     else:
         raise ValueError("Either input_text or audio_file must be provided.")
 
-    # patient_name = report
     patient_name = report.patient_name
     soap_report = report.soap_report
     sbar_report = report.sbar_report
@@ -301,14 +303,6 @@ def report_main(input_text: str = None, audio_file: str = None, transcription_se
     else:
         return patient_name, soap_report, sbar_report, doctor_recommendations
 
-# x=report_main(audio_file="/content/Cough.wav")
-
-# x
-
-y=report_main(input_text="I have fever")
-
-print(y)
-
 def delete_dir(dir):
     try:
         shutil.rmtree(dir)
@@ -725,18 +719,18 @@ with gr.Blocks(css=css) as demo:
                     audio_sbar_report_box = 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)
+            audio_transcription_box = gr.Textbox(label="Transcription Text", interactive=False, placeholder="Transcribed Text", lines=5)
 
             # Click event for audio
             generate_with_audio_button.click(
-                fn=report_main,
+                fn=report_audio,
                 inputs=[audio_file, transcription_service],
                 outputs=[
                     audio_patient_name_box,
                     audio_soap_report_box,
                     audio_sbar_report_box,
                     audio_doctor_recommendations_box,
-                    # audio_transcription_box
+                    audio_transcription_box
                 ]
             )
 
@@ -897,4 +891,3 @@ with gr.Blocks(css=css) as demo:
                 docs2_delete_button.click(delete2_docs, inputs=None, outputs=docs2_delete_output)
 
 demo.launch(debug=True)
-