Update app.py

app.py

@@ -4,7 +4,7 @@ from huggingface_hub import InferenceClient
 # ✅ Connect to Hugging Face Zephyr-7B API
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# ✅ Mental health-specific system message
+# ✅ Define the system message for a mental health chatbot
 MENTAL_HEALTH_PROMPT = (
     "You are Bandar AI, an empathetic mental health AI assistant. "
     "You provide emotional support, self-care advice, and stress management tips. "
@@ -17,7 +17,8 @@ def is_mental_health_related(message):
     keywords = [
         "anxious", "stress", "depressed", "lonely", "overwhelmed",
         "panic", "mental health", "self-care", "therapy", "sad", "burnout",
-        "emotions", "counseling", "support", "well-being"
+        "emotions", "counseling", "support", "well-being", "anxiety",
+        "depression", "happiness", "relaxation", "coping", "mindfulness"
     ]
     return any(word in message.lower() for word in keywords)
 
@@ -30,9 +31,10 @@ def respond(
     temperature,
     top_p,
 ):
-    # 🔴 Reject unrelated topics
+    # 🔴 Reject unrelated topics with a streaming response
     if not is_mental_health_related(message):
-        return "❌ Sorry, I can only discuss mental health topics. Please ask about emotions, self-care, or stress management."
+        yield "❌ Sorry, I can only discuss mental health topics. Please ask about emotions, self-care, or stress management."
+        return
 
     messages = [{"role": "system", "content": system_message}]
 
@@ -58,7 +60,7 @@ def respond(
     ):
         token = message.choices[0].delta.content
         response += token
-        yield response  # Stream output in real-time
+        yield response  # ✅ Stream output correctly
 
 # ✅ Improved Chatbot UI for Mental Health
 demo = gr.ChatInterface(