Update app.py

app.py

@@ -1,60 +1,47 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# ✅ Connect to Hugging Face Zephyr-7B API
+# ✅ Connect to Hugging Face API (Replace with your model or API key)
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# ✅ Mental Health System Message
-MENTAL_HEALTH_PROMPT = (
-    "أنت مساعد ذكاء اصطناعي للصحة النفسية. "
-    "أنت تقدم الدعم العاطفي، نصائح الرعاية الذاتية، وإدارة التوتر. "
-    "لا تقدم تشخيصات طبية. شجع المستخدمين دائمًا على طلب المساعدة المهنية عند الحاجة."
+# ✅ System Prompt for Bandar AI (General Purpose)
+BANDAR_AI_PROMPT = (
+    "أنت مساعد ذكاء اصطناعي يدعى باندر AI. "
+    "تهدف إلى تقديم المساعدة في مجموعة متنوعة من المواضيع، مثل الإجابة على الأسئلة، تقديم النصائح، والمساعدة في المهام اليومية. "
+    "كن ودودًا ومتعاونًا دائمًا."
 )
 
-# ✅ Function to check if the message is related to mental health
-def is_mental_health_related(message):
-    keywords = [
-        "قلق", "توتر", "اكتئاب", "وحدة", "إجهاد", "نوبات هلع", 
-        "صحة نفسية", "رعاية ذاتية", "علاج", "حزن", "احتراق وظيفي",
-        "عواطف", "استشارة", "دعم", "رفاهية", "تفكير إيجابي", "راحة",
-        "coping", "therapy", "anxiety", "stress", "depression",
-    ]
-    return any(word in message.lower() for word in keywords)
-
 # ✅ Chat Response Function
 def respond(message, history):
-    # 🔴 Reject unrelated topics
-    if not is_mental_health_related(message):
-        return "❌ يمكنني فقط مناقشة مواضيع الصحة النفسية. الرجاء طرح أسئلة عن العواطف، الرعاية الذاتية، أو إدارة التوتر."
-
     # ✅ Prepare conversation
-    messages = [{"role": "system", "content": MENTAL_HEALTH_PROMPT}]
+    messages = [{"role": "system", "content": BANDAR_AI_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]})
-
+    # Add previous conversation history
+    for user_msg, bot_msg in history:
+        if user_msg: messages.append({"role": "user", "content": user_msg})
+        if bot_msg: messages.append({"role": "assistant", "content": bot_msg})
+    
+    # Add the current user message
     messages.append({"role": "user", "content": message})
-
+    
     response = ""
-
     # ✅ Stream responses from Hugging Face API
     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
+        if token:
+            response += token
+            yield response  # ✅ Stream output correctly
 
 # ✅ Clean & Modern UI
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
-        # 🧠 **Bandar AI - Your Mental Health Companion**  
-        ### 💙 تحدث عن مشاعرك، القلق، والتوتر. نحن هنا لدعمك.  
+        # 🧠 **Bandar AI - Your General-Purpose Assistant**  
+        ### 💬 تحدث معي! أنا هنا لمساعدتك في أي موضوع تحتاجه.
         """
     )
-
     chatbot = gr.ChatInterface(respond)
 
 # ✅ Run the chatbot
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()