Update app.py

app.py

@@ -3,14 +3,12 @@ from huggingface_hub import InferenceClient
 import random
 import re
 
-# ✅ Smart Detection: Only blocks obvious garbage (ads/links/spam), everything else goes to AI
+# ✅ تنظيف المدخلات من السبام
 OFF_TOPIC_REGEX = r"(http|www|buy now|discount|subscribe|follow me|click here)"
 
 def is_safe_to_process(text: str) -> bool:
-    # If it's a very short message with no meaning, we use a friendly nudge
     if len(text.strip()) < 2:
         return False
-    # If it matches spam patterns
     if re.search(OFF_TOPIC_REGEX, text.lower()):
         return False
     return True
@@ -23,35 +21,48 @@ def respond(
     temperature,
     top_p,
     hf_token: gr.OAuthToken,
+    request: gr.Request, # 👈 دي اللي بتسحب بيانات اليوزر من موقعك
 ):
-    # 1. Basic Safety Check
+    # 1. سحب الـ Username من الـ URL Query Parameter
+    # موقعك هيبعت الـ Iframe كدا: src=".../?username=Koko"
+    user_name = "User"
+    if request:
+        user_name = request.query_params.get("username", "User")
+
     if not is_safe_to_process(message):
-        yield "I'm here to listen and support your emotional well-being. How can I help you today?"
+        yield f"Hello {user_name}, I'm here to support your emotional well-being. How can I help you today?"
         return
 
     if not hf_token:
-        yield "Please log in via the Sidebar to start our session."
+        yield "Please log in to start our session."
         return
 
-    # 2. Using Zephyr-7B: Faster, smarter, and doesn't 'hang' on free tier
     client = InferenceClient(token=hf_token.token, model="HuggingFaceH4/zephyr-7b-beta")
 
-    # 3. Memory Management: Last 8 messages to keep the context sharp but fast
-    messages = [{"role": "system", "content": system_message}]
-    messages.extend(history[-8:])
+    # 2. حوار الـ History والـ الشخصية
+    # بنحقن الاسم في الـ System Message عشان الموديل يناديه باسمه صح
+    personalized_system = (
+        f"{system_message} "
+        f"The user's name is {user_name}. Always address them by this name naturally. "
+        f"Do not use formal headers like 'Dear'."
+    )
+    
+    # الـ History هنا هتكون اللي مبعوثة من الـ Frontend بتاعك فقط
+    # الموديل هياخد آخر 10 رسايل عشان الذاكرة تكون قوية بس سريعة
+    messages = [{"role": "system", "content": personalized_system}]
+    messages.extend(history[-10:]) 
     messages.append({"role": "user", "content": message})
 
     response = ""
     try:
-        # 4. Perfect Parameters for ChatGPT-like flow
         for msg in client.chat_completion(
             messages,
             max_tokens=max_tokens,
             stream=True,
-            temperature=0.7, # The "Sweet Spot"
-            top_p=0.9,
+            temperature=temperature,
+            top_p=top_p,
             extra_body={
-                "repetition_penalty": 1.15, # Stops the "I specialize in..." loop
+                "repetition_penalty": 1.15,
                 "presence_penalty": 0.3
             }
         ):
@@ -59,37 +70,36 @@ def respond(
             response += token
             yield response
     except Exception as e:
-        yield f"Connection error: {str(e)}. Please try refreshing the page."
+        yield f"Connection error: {str(e)}. Please try refreshing."
 
-# ✅ Professional UI Setup
-with gr.Blocks(theme=gr.themes.Default(primary_hue="blue")) as demo:
+# ✅ الـ UI الاحترافي
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
     with gr.Sidebar():
-        gr.Markdown("## 🌿 Therapy Assistant Settings")
+        gr.Markdown("## 🌿 Therapy Assistant")
         gr.LoginButton()
         gr.Markdown("---")
         sys_msg = gr.Textbox(
             value=(
-                "You are a professional mental health assistant. "
-                "1. Respond directly to the user's specific problem in the first sentence. "
-                "2. Be empathetic but professional. "
-                "3. If the user mentions symptoms (like washing hands), explain them gently as a supportive peer. "
-                "4. Never use a repetitive introductory phrase."
+                "You are a compassionate mental health assistant. "
+                "Respond directly to the user's problem. "
+                "Be empathetic and professional. "
+                "Never use formal letter formats or generic intros."
             ),
             label="System Persona",
-            lines=6
+            lines=5
         )
-        tokens = gr.Slider(128, 1024, value=512, label="Response Length")
-        temp = gr.Slider(0.1, 1.5, value=0.7, label="Empathy Level (Temperature)")
-        top_p_val = gr.Slider(0.1, 1.0, value=0.9, label="Focus (Top-p)")
+        tokens = gr.Slider(128, 1024, value=512, label="Max Response Length")
+        temp = gr.Slider(0.1, 1.5, value=0.7, label="Temperature")
+        top_p_val = gr.Slider(0.1, 1.0, value=0.9, label="Top-p")
 
     chatbot_ui = gr.ChatInterface(
         respond,
         type="messages",
         additional_inputs=[sys_msg, tokens, temp, top_p_val],
+        # خلي بالك الـ Examples لازم تطابق عدد الـ inputs
         examples=[
-            ["I think my sister has OCD, she washes her hands constantly.", sys_msg.value, 512, 0.7, 0.9],
-            ["Ana 7ases b de2 fashkh w msh 3aref anam.", sys_msg.value, 512, 0.7, 0.9],
-            ["I've been feeling very lonely lately.", sys_msg.value, 512, 0.7, 0.9]
+            ["I've been feeling very anxious lately.", sys_msg.value, 512, 0.7, 0.9],
+            ["Ana ta3ban awee w msh 3aref anam.", sys_msg.value, 512, 0.7, 0.9]
         ],
         cache_examples=False,
     )