Update app.py

app.py

@@ -1,106 +1,59 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# ✅ Use a multilingual model for Arabic & English support
-client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
+# ✅ Connect to Hugging Face Zephyr-7B API
+client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
 # ✅ Mental Health System Message
 MENTAL_HEALTH_PROMPT = (
-    "أنت بندر AI، مساعد صحي نفسي متعاطف. "
-    "تقدم دعمًا عاطفيًا، نصائح للعناية بالنفس، وتقنيات لإدارة التوتر. "
-    "لا تقدم تشخيصات طبية أبدًا، ويجب عليك دائمًا تشجيع المستخدمين على طلب المساعدة المتخصصة عند الحاجة.\n"
-    "You are Bandar AI, an empathetic mental health AI assistant. "
-    "You provide emotional support, self-care advice, and stress management tips. "
-    "You DO NOT give medical diagnoses. Always encourage users to seek professional help if needed.\n"
+    "أنت مساعد ذكاء اصطناعي للصحة النفسية. "
+    "أنت تقدم الدعم العاطفي، نصائح الرعاية الذاتية، وإدارة التوتر. "
+    "لا تقدم تشخيصات طبية. شجع المستخدمين دائمًا على طلب المساعدة المهنية عند الحاجة."
 )
 
-# ✅ Mental health keywords
-MENTAL_HEALTH_KEYWORDS = [
-    "قلق", "توتر", "اكتئاب", "حزين", "علاج", "عواطف", "صحة نفسية",
-    "دعم", "رعاية ذاتية", "علاج نفسي", "إرشاد", "استرخاء", "الرفاهية",
-    "ضغط نفسي", "مساعدة", "أشعر بالحزن", "أحتاج إلى دعم", "قلق شديد",
-    "anxious", "stress", "depressed", "lonely", "overwhelmed", "panic",
-    "mental health", "self-care", "therapy", "sad", "burnout",
-    "emotions", "counseling", "support", "well-being", "anxiety",
-    "depression", "happiness", "relaxation", "coping", "mindfulness"
-]
-
 # ✅ Function to check if the message is related to mental health
 def is_mental_health_related(message):
-    return any(word in message.lower() for word in MENTAL_HEALTH_KEYWORDS)
+    keywords = [
+        "قلق", "توتر", "اكتئاب", "وحدة", "إجهاد", "نوبات هلع", 
+        "صحة نفسية", "رعاية ذاتية", "علاج", "حزن", "احتراق وظيفي",
+        "عواطف", "استشارة", "دعم", "رفاهية", "تفكير إيجابي", "راحة",
+        "coping", "therapy", "anxiety", "stress", "depression",
+    ]
+    return any(word in message.lower() for word in keywords)
 
 # ✅ Chat Response Function
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
+def respond(message, history):
     # 🔴 Reject unrelated topics
     if not is_mental_health_related(message):
-        return "❌ عذرًا، يمكنني فقط مناقشة مواضيع الصحة النفسية.\nSorry, I can only discuss mental health topics."
-
-    messages = [{"role": "system", "content": system_message}]
+        return "❌ يمكنني فقط مناقشة مواضيع الصحة النفسية. الرجاء طرح أسئلة عن العواطف، الرعاية الذاتية، أو إدارة التوتر."
 
-    # ✅ Add chat history
+    # ✅ Prepare conversation
+    messages = [{"role": "system", "content": MENTAL_HEALTH_PROMPT}]
+    
     for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
+        if val[0]: messages.append({"role": "user", "content": val[0]})
+        if val[1]: messages.append({"role": "assistant", "content": val[1]})
 
-    # ✅ Add the new user input
     messages.append({"role": "user", "content": message})
 
     response = ""
 
     # ✅ Stream responses from Hugging Face API
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
+    for msg in client.chat_completion(messages, max_tokens=512, stream=True, temperature=0.7, top_p=0.95):
+        token = msg.choices[0].delta.content
         response += token
         yield response  # ✅ Stream output correctly
 
-# ✅ Modern UI
+# ✅ Clean & Modern UI
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
-        <h1 style="text-align: center; color: #ffffff;">🧠 بندر AI - رفيقك للصحة النفسية</h1>
-        <p style="text-align: center; font-size: 18px; color: #dddddd;">
-        💙 تحدث عن مشاعرك، التوتر، والعناية الذاتية. احصل على الدعم عندما تحتاج إليه.
-        </p>
-        """,
-        elem_id="title"
+        # 🧠 **Bandar AI - Your Mental Health Companion**  
+        ### 💙 تحدث عن مشاعرك، القلق، والتوتر. نحن هنا لدعمك.  
+        """
     )
 
-    # ✅ Chat Interface with Improved Design
-    chat_interface = gr.ChatInterface(
-        respond,
-        additional_inputs=[
-            gr.Slider(minimum=1, maximum=1024, value=512, step=1, label="🔠 أقصى عدد من الكلمات"),
-            gr.Slider(minimum=0.1, maximum=2.0, value=0.7, step=0.1, label="🔥 درجة العشوائية"),
-            gr.Slider(
-                minimum=0.1,
-                maximum=1.0,
-                value=0.95,
-                step=0.05,
-                label="🎯 Top-p (اختيار الكلمات)",
-            ),
-        ],
-        chatbot=gr.Chatbot(
-            bubble_radius=10,  # ✅ Rounded bubbles
-            height=400,  # ✅ Fixed height for better layout
-            avatar_images=("🧑‍💻", "🤖"),  # ✅ User and AI icons
-        ),
-        type="messages",  # ✅ Fixes deprecated warning
-    )
+    chatbot = gr.ChatInterface(respond)
 
 # ✅ Run the chatbot
 if __name__ == "__main__":