Update app.py

app.py

@@ -29,18 +29,22 @@ def echo(audio):
     #     max_tokens=200,
     # )
     # prompt = response.choices[0].message.content
-    prompt = chat_with_bot(prompt)
-    for audio_chunk in tts_model.stream_tts_sync(prompt):
-        yield audio_chunk
+    bot_response = chat_with_bot(prompt)
+    audio_chunks = []
+    for audio_chunk in tts_model.stream_tts_sync(bot_response):
+        audio_chunks.append(audio_chunk)
+    return audio_chunks[0]
+    # for audio_chunk in tts_model.stream_tts_sync(prompt):
+    #     yield audio_chunk
 def process_audio(audio_input):
     # audio_input is received as a Gradio Audio object, containing a tuple of (sample_rate, numpy array)
     sample_rate, audio_data = audio_input
     
     # Process audio through the stream
-    processed_sample_rate, processed_audio = echo((sample_rate, audio_data))
+    processed_audio = echo((sample_rate, audio_data))
     
     # Return processed audio to Gradio for output
-    return processed_sample_rate, processed_audio
+    return sample_rate, processed_audio
 
 
 def chat_with_bot(user_input):