Update app.py

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import pyttsx3  # For text-to-speech
+import speech_recognition as sr  # For speech-to-text
 
 # ✅ Connect to Hugging Face Zephyr-7B API
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# ✅ Define the system message for a mental health chatbot
+# ✅ Mental Health System Message
 MENTAL_HEALTH_PROMPT = (
     "You are Bandar AI, an empathetic mental health AI assistant. "
     "You provide emotional support, self-care advice, and stress management tips. "
@@ -22,7 +24,24 @@ def is_mental_health_related(message):
     ]
     return any(word in message.lower() for word in keywords)
 
-# ✅ Function to handle chat responses
+# ✅ Speech-to-Text Function (Voice Input)
+def speech_to_text(audio):
+    recognizer = sr.Recognizer()
+    with sr.AudioFile(audio) as source:
+        audio_data = recognizer.record(source)
+    try:
+        return recognizer.recognize_google(audio)  # Convert speech to text
+    except sr.UnknownValueError:
+        return "❌ Sorry, I couldn't understand the audio."
+
+# ✅ Text-to-Speech Function (Voice Output)
+def text_to_speech(response_text):
+    engine = pyttsx3.init()
+    engine.save_to_file(response_text, "response.mp3")
+    engine.runAndWait()
+    return "response.mp3"
+
+# ✅ Chat Response Function
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -62,31 +81,43 @@ def respond(
         response += token
         yield response  # ✅ Stream output correctly
 
-# ✅ Improved Chatbot UI for Mental Health
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(
+# ✅ Improved Modern UI
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown(
+        """
+        # 🧠 Bandar AI - Your Mental Health Companion
+        ### 💙 Chat about emotions, anxiety, and self-care. Get support when you need it.
+        """
+    )
+
+    with gr.Row():
+        system_message = gr.Textbox(
             value=MENTAL_HEALTH_PROMPT,
             label="System message (Defines AI behavior)",
-        ),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-p (nucleus sampling)",
-        ),
-    ],
-    title="🧠 Bandar AI - Your Mental Health Companion",
-    description=(
-        "💙 Bandar AI is here to support your mental well-being. "
-        "Chat with me about stress, anxiety, or self-care. "
-        "I offer guidance, but I'm NOT a medical professional. Always seek expert help when needed."
-    ),
-)
+            interactive=False,
+        )
+
+    chat_interface = gr.ChatInterface(
+        respond,
+        additional_inputs=[
+            gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max tokens"),
+            gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+            gr.Slider(
+                minimum=0.1,
+                maximum=1.0,
+                value=0.95,
+                step=0.05,
+                label="Top-p (nucleus sampling)",
+            ),
+        ],
+    )
+
+    with gr.Row():
+        voice_input = gr.Microphone(label="🎤 Speak", type="filepath")
+        voice_output = gr.Audio(label="🔊 AI Response", type="file")
+
+    voice_input.change(speech_to_text, inputs=voice_input, outputs=chat_interface.input_component)
+    chat_interface.output_component.change(text_to_speech, inputs=chat_interface.output_component, outputs=voice_output)
 
 # ✅ Run the chatbot
 if __name__ == "__main__":