Update app.py

app.py

@@ -24,9 +24,10 @@ def generate_response(history, user_input, temperature, max_new_tokens):
 
     # Generate speech
     tts = gTTS(text=generated_text, lang='en')
-    tts.save("response.mp3")
+    audio_path = "response.mp3"
+    tts.save(audio_path)
 
-    return history, "response.mp3"
+    return history, audio_path
 
 # Function to clear chat history
 def clear_chat():
@@ -82,12 +83,14 @@ with gr.Blocks(css=custom_css) as demo:
         send_btn = gr.Button("Send")
         clear_btn = gr.Button("Clear Chat")
 
+    audio_output = gr.Audio(label="Generated Speech", type="filepath")
+
     with gr.Accordion("Advanced Settings", open=False, elem_id="settings"):
         temperature_slider = gr.Slider(0, 1, 0.7, step=0.1, label="Temperature")
         max_new_tokens_slider = gr.Slider(1, 1000, 100, step=1, label="Max New Tokens")
 
-    send_btn.click(generate_response, [chatbot, txt, temperature_slider, max_new_tokens_slider], [chatbot, "audio"])
-    txt.submit(generate_response, [chatbot, txt, temperature_slider, max_new_tokens_slider], [chatbot, "audio"])
+    send_btn.click(generate_response, [chatbot, txt, temperature_slider, max_new_tokens_slider], [chatbot, audio_output])
+    txt.submit(generate_response, [chatbot, txt, temperature_slider, max_new_tokens_slider], [chatbot, audio_output])
     clear_btn.click(clear_chat, outputs=[chatbot, txt])
 
 # Launch the interface