Update app.py

app.py

@@ -93,8 +93,8 @@ def generate_soap(
 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(
+    # Then generate SOAP notes based on the transcription and collect all generated text into a single string
+    generated_text = "".join(generate_soap(
         transcribed_text,
         system_prompt,
         max_new_tokens,
@@ -102,7 +102,8 @@ def transcribe_and_generate_soap(audio, task, system_prompt, max_new_tokens, tem
         top_p,
         top_k,
         repetition_penalty
-    )
+    ))
+    return generated_text
 
 # Gradio Interface combining transcription and SOAP note generation
 demo = gr.Blocks(theme=gr.themes.Ocean())