Update app.py

app.py

@@ -4,6 +4,8 @@
 
 import gradio as gr
 from transformers import pipeline
+from gtts import gTTS
+import os
 
 # Load the model and pipeline
 model_name = "MTSAIR/Cotype-Nano"
@@ -19,7 +21,12 @@ def generate_response(history, user_input, temperature, max_new_tokens):
     generated_text = response[0]['generated_text']
     history.append({"role": "user", "content": user_input})
     history.append({"role": "assistant", "content": generated_text})
-    return history, ""
+
+    # Generate speech
+    tts = gTTS(text=generated_text, lang='en')
+    tts.save("response.mp3")
+
+    return history, "response.mp3"
 
 # Function to clear chat history
 def clear_chat():
@@ -66,12 +73,12 @@ with gr.Blocks(css=custom_css) as demo:
 
     chatbot = gr.Chatbot([], elem_id="chatbot", type='messages')
 
-    with gr.Row():
-        txt = gr.Textbox(
-            show_label=False,
-            placeholder="Type your message here...",
-        )
+    txt = gr.Textbox(
+        show_label=False,
+        placeholder="Type your message here...",
+    )
 
+    with gr.Row():
         send_btn = gr.Button("Send")
         clear_btn = gr.Button("Clear Chat")
 
@@ -79,8 +86,8 @@ with gr.Blocks(css=custom_css) as demo:
         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, txt])
-    txt.submit(generate_response, [chatbot, txt, temperature_slider, max_new_tokens_slider], [chatbot, txt])
+    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"])
     clear_btn.click(clear_chat, outputs=[chatbot, txt])
 
 # Launch the interface