Update app.py

app.py

@@ -46,40 +46,40 @@ print("✅ Model ready!\n")
 # ======================================================
 @spaces.GPU(duration=60)
 def chat(message, history):
-    # تجهيز سجل المحادثة
+    # إعداد سياق المحادثة
     messages = []
     for msg in history:
-        if msg["role"] == "user":
-            messages.append({"role": "user", "content": msg["content"]})
-        elif msg["role"] == "assistant":
-            messages.append({"role": "assistant", "content": msg["content"]})
+        messages.append({"role": msg["role"], "content": msg["content"]})
+
+    # system prompt محسّن
+    system_prompt = (
+        "أنت مساعد ذكي تحچي باللهجة العراقية الطبيعية. "
+        "جاوب بحرّية وخلّك مرتاح بالكلام، مثل إنسان حقيقي. "
+        "تذكّر شنو انحچا قبله وكمّل الحديث بدون ما تكون رسمي. "
+        "لا تقول 'مرحباً' أو 'بالتأكيد'، بس جاوب مباشرة."
+    )
 
-    # إضافة السؤال الحالي مع الـ system prompt
-    messages.append({"role": "user", "content": f"{SYSTEM_PROMPT}\n\nالسؤال: {message}"})
+    messages.append({"role": "system", "content": system_prompt})
+    messages.append({"role": "user", "content": message})
 
-    # تجهيز الإدخال
     input_ids = tokenizer.apply_chat_template(
         messages,
         return_tensors="pt",
-        add_generation_prompt=True,
+        add_generation_prompt=True
     ).to(model.device)
 
-    # إعداد البث المباشر للنص
-    streamer = TextIteratorStreamer(
-        tokenizer, skip_prompt=True, skip_special_tokens=True
-    )
+    streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
 
     generation_kwargs = dict(
         input_ids=input_ids,
         streamer=streamer,
-        max_new_tokens=256,
-        temperature=0.8,
-        top_p=0.95,
+        max_new_tokens=1024,        # يكتب براحة
+        temperature=0.95,          # كلام طبيعي
+        top_p=0.95,                # توازن بالاحتمالات
         do_sample=True,
-        repetition_penalty=1.15,
+        repetition_penalty=1.07,   # يقلل التكرار الطفيف
     )
 
-    # إنشاء Thread للبث التدريجي
     thread = Thread(target=model.generate, kwargs=generation_kwargs)
     thread.start()