Update app.py

app.py

@@ -89,13 +89,28 @@ def generate_soap(
         outputs.append(text)
         yield "".join(outputs)
 
+# Combine transcription and SOAP generation
+def transcribe_and_generate_soap(audio, task, system_prompt, max_new_tokens, temperature, top_p, top_k, repetition_penalty):
+    # First transcribe the audio
+    transcribed_text = transcribe(audio, task)
+    # Then generate SOAP notes based on the transcription
+    return generate_soap(
+        transcribed_text,
+        system_prompt,
+        max_new_tokens,
+        temperature,
+        top_p,
+        top_k,
+        repetition_penalty
+    )
+
 # Gradio Interface combining transcription and SOAP note generation
 demo = gr.Blocks(theme=gr.themes.Ocean())
 
 with demo:
     with gr.Tab("Clinical SOAP Note from Audio"):
         audio_transcribe_and_soap = gr.Interface(
-            fn=lambda inputs, task: generate_soap(transcribe(inputs, task)),
+            fn=transcribe_and_generate_soap,
             inputs=[
                 gr.Audio(sources=["microphone", "upload"], type="filepath", label="Audio Input"),
                 gr.Radio(["transcribe", "translate"], label="Task", value="transcribe"),