Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import os
 import shutil
 import openai
@@ -163,6 +164,19 @@ def assemblyai_STT(audio_url: str) -> str:
 
     return transcription_output
 
+def openai_STT(audio_url: str) -> str:
+    from openai import OpenAI
+    client = OpenAI()
+    audio = open(audio_url, "rb")
+    transcript = client.audio.transcriptions.create(
+          model="whisper-1",
+          file=audio,
+          response_format="text"
+          )
+    output = transcript
+
+    return output
+
 def generate_report(input_text: str = None, file_path: str = None) -> AI_Medical_Report:
     """
     Generates a SOAP report from text or audio input using OpenAI's GPT-4 model.
@@ -230,7 +244,7 @@ def generate_report(input_text: str = None, file_path: str = None) -> AI_Medical
         return None
 
 # driver function for making reports
-def report_main(input_text: str = None, audio_file: str = None, transcription_service: str = "OpenAI") -> tuple:
+def report_main(input_text: str = None, audio_file: str = None, transcription_service: str = "OpenAI"):
     """
     Generates a SOAP and SBAR report based on user input, either from text or audio.
 
@@ -263,11 +277,11 @@ def report_main(input_text: str = None, audio_file: str = None, transcription_se
         if transcription_service == "AssemblyAI":
             transcription_text = assemblyai_STT(audio_file)
             report = generate_report(input_text=transcription_text)
-            print(transcription_text)
+            # print(transcription_text)
         elif transcription_service == "OpenAI":
-            report = generate_report(file_path=audio_file)
-            transcription_text = "OpenAI directly accepts Audio Inputs"
-            print(transcription_text)
+            transcription_text = openai_STT(audio_file)
+            report = generate_report(input_text=transcription_text)
+            # print(transcription_text)
         else:
             raise ValueError("Invalid transcription service specified. Choose 'AssemblyAI' or 'OpenAI'.")
         print(report)
@@ -281,9 +295,18 @@ def report_main(input_text: str = None, audio_file: str = None, transcription_se
     sbar_report = report.sbar_report
     doctor_recommendations = report.recommendations_for_doc
     # Return structured output in a tuple
-    return patient_name, soap_report, sbar_report, doctor_recommendations
+    if audio_file:
+        return patient_name, soap_report, sbar_report, doctor_recommendations, transcription_text
+    else:
+        return patient_name, soap_report, sbar_report, doctor_recommendations
+
+# x=report_main(audio_file="/content/Cough.wav")
+
+# x
 
-# x=generate_report(file_path="/content/Cough.wav")
+# y=report_main(input_text="I have fever")
+
+# y
 
 def delete_dir(dir):
     try:
@@ -711,7 +734,8 @@ with gr.Blocks(css=css) as demo:
                     audio_patient_name_box,
                     audio_soap_report_box,
                     audio_sbar_report_box,
-                    audio_doctor_recommendations_box
+                    audio_doctor_recommendations_box,
+                    # audio_transcription_box
                 ]
             )
 
@@ -871,4 +895,5 @@ with gr.Blocks(css=css) as demo:
                 docs2_search_button.click(search2_docs, inputs=[docs2_prompt_input, docs2_search_input, docs2_state], outputs=[docs2_chatbot, docs2_state])
                 docs2_delete_button.click(delete2_docs, inputs=None, outputs=docs2_delete_output)
 
-demo.launch(debug=True)
+demo.launch(debug=True)
+