Move clear of audio to generate_text step.

app.py

@@ -161,12 +161,12 @@ def generate_text(code, prompt):
     new_code = match.group(1)
     print(new_code)
     # TODO: error handling here
-    return assistant_reply, new_code
+    return assistant_reply, new_code, None
 
 
 def transcribe(audio):
     result = whisper_pipe(audio)
-    return result["text"], None
+    return result["text"]
 
 
 def copy_notify(code):
@@ -194,7 +194,7 @@ with gr.Blocks() as demo:
             update_btn = gr.Button("Update App", variant="primary")
             update_btn.click(None, inputs=code_area, outputs=None, _js=update_iframe_js)
             in_prompt.submit(generate_text, [code_area, in_prompt], [out_text, code_area]).then(None, inputs=code_area, outputs=None, _js=update_iframe_js)
-            in_audio.stop_recording(transcribe, [in_audio], [in_prompt, in_audio]).then(generate_text, [code_area, in_prompt], [out_text, code_area]).then(None, inputs=code_area, outputs=None, _js=update_iframe_js)
+            in_audio.stop_recording(transcribe, [in_audio], [in_prompt]).then(generate_text, [code_area, in_prompt, in_audio], [out_text, code_area]).then(None, inputs=code_area, outputs=None, _js=update_iframe_js)
     with gr.Row():
         with gr.Column():
             gr.Markdown("## 3. Export your app to share!")