Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# ✅ Use a multilingual model for better Arabic & English support
+# ✅ Use a multilingual model for Arabic & English support
 client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.1")
 
-# ✅ Mental Health System Message (Arabic & English)
+# ✅ Mental Health System Message
 MENTAL_HEALTH_PROMPT = (
     "أنت بندر AI، مساعد صحي نفسي متعاطف. "
     "تقدم دعمًا عاطفيًا، نصائح للعناية بالنفس، وتقنيات لإدارة التوتر. "
@@ -14,7 +14,7 @@ MENTAL_HEALTH_PROMPT = (
     "You DO NOT give medical diagnoses. Always encourage users to seek professional help if needed.\n"
 )
 
-# ✅ Mental health keywords (Arabic & English)
+# ✅ Mental health keywords
 MENTAL_HEALTH_KEYWORDS = [
     "قلق", "توتر", "اكتئاب", "حزين", "علاج", "عواطف", "صحة نفسية",
     "دعم", "رعاية ذاتية", "علاج نفسي", "إرشاد", "استرخاء", "الرفاهية",
@@ -29,7 +29,7 @@ MENTAL_HEALTH_KEYWORDS = [
 def is_mental_health_related(message):
     return any(word in message.lower() for word in MENTAL_HEALTH_KEYWORDS)
 
-# ✅ Chat Response Function (Arabic & English Support)
+# ✅ Chat Response Function
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -40,7 +40,7 @@ def respond(
 ):
     # 🔴 Reject unrelated topics
     if not is_mental_health_related(message):
-        return "❌ عذرًا، يمكنني فقط مناقشة مواضيع الصحة النفسية. يرجى السؤال عن العواطف، الرعاية الذاتية، أو إدارة التوتر.\nSorry, I can only discuss mental health topics."
+        return "❌ عذرًا، يمكنني فقط مناقشة مواضيع الصحة النفسية.\nSorry, I can only discuss mental health topics."
 
     messages = [{"role": "system", "content": system_message}]
 
@@ -51,8 +51,8 @@ def respond(
         if val[1]:
             messages.append({"role": "assistant", "content": val[1]})
 
-    # ✅ Add the new user input (Ensure UTF-8 encoding)
-    messages.append({"role": "user", "content": message.encode("utf-8").decode("utf-8")})
+    # ✅ Add the new user input
+    messages.append({"role": "user", "content": message})
 
     response = ""
 
@@ -68,7 +68,7 @@ def respond(
         response += token
         yield response  # ✅ Stream output correctly
 
-# ✅ Beautiful Modern UI
+# ✅ Modern UI
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
@@ -80,36 +80,27 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         elem_id="title"
     )
 
-    with gr.Row():
-        system_message = gr.Textbox(
-            value=MENTAL_HEALTH_PROMPT,
-            label="رسالة النظام (تحديد سلوك الذكاء الاصطناعي)",
-            interactive=False,
-            visible=False,  # ✅ Hide system message box for a cleaner UI
-        )
-
     # ✅ Chat Interface with Improved Design
-    with gr.Box():
-        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
+    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 (اختيار الكلمات)",
             ),
-            type="messages",  # ✅ Fixes deprecated warning
-        )
+        ],
+        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
+    )
 
 # ✅ Run the chatbot
 if __name__ == "__main__":