update text to speech

chat_app.py

@@ -2,7 +2,11 @@ import os
 from groq import Groq
 import gradio as gr
 from dotenv import load_dotenv
+from gtts import gTTS
+import tempfile
+
 load_dotenv()
+
 # Load API key from environment variable
 API_KEY = os.getenv("GROQ_API_KEY")
 if not API_KEY:
@@ -45,6 +49,18 @@ def get_agriculture_response(input_text):
     except Exception as e:
         return f"Error: {e}"
 
+def text_to_speech(text):
+    if not text.strip():
+        return None
+    try:
+        tts = gTTS(text=text, lang='en')
+        temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+        tts.save(temp_file.name)
+        return temp_file.name
+    except Exception as e:
+        print(f"TTS error: {e}")
+        return None
+
 def groq_chatbot(input_text, chat_history):
     validation_result = validate_input(input_text)
     
@@ -55,9 +71,16 @@ def groq_chatbot(input_text, chat_history):
     else:
         response = f"⚠️ Unexpected response: {validation_result}"
 
-    # Append to chat history
     chat_history.append((input_text, response))
-    return chat_history, ""  # Clears input field after submission
+    # Return chat history, clear input, reset audio output
+    return chat_history, "", None
+
+def on_tts_click(chat_history):
+    if chat_history and chat_history[-1][1]:
+        last_response = chat_history[-1][1]
+        audio_path = text_to_speech(last_response)
+        return audio_path
+    return None
 
 def launch_gradio_interface():
     with gr.Blocks() as demo:
@@ -67,16 +90,21 @@ def launch_gradio_interface():
         chatbot = gr.Chatbot(height=400)
         msg = gr.Textbox(placeholder="Ask a question about agriculture...", label="Your Question")
         clear = gr.Button("Clear Chat")
+        tts_button = gr.Button("Text-to-Speech")
+        audio_player = gr.Audio(label="Response Audio", interactive=False)
+
+        chat_history_state = gr.State([])
 
-        chat_history_state = gr.State([])  # Stores chat history
+        # Submit user question
+        msg.submit(fn=groq_chatbot, inputs=[msg, chat_history_state], outputs=[chatbot, msg, audio_player])
 
-        # ✅ Enter key submits the question
-        msg.submit(fn=groq_chatbot, inputs=[msg, chat_history_state], outputs=[chatbot, msg])
+        # Clear chat and reset audio
+        clear.click(lambda: ([], "", None), None, [chatbot, msg, audio_player], queue=False)
 
-        # ✅ Clicking "Clear Chat" resets history
-        clear.click(lambda: ([], ""), None, [chatbot, msg], queue=False)
+        # Generate TTS audio of last response
+        tts_button.click(fn=on_tts_click, inputs=chat_history_state, outputs=audio_player)
 
     demo.launch(share=True)
 
 if __name__ == "__main__":
-    launch_gradio_interface()
+    launch_gradio_interface()