updatE

app.py

@@ -125,45 +125,47 @@ def detect_language(text):
     arabic_ratio = arabic_chars / total_chars
     return 'ar' if arabic_ratio > 0.3 else 'en'
 
+@spaces.GPU()
 @spaces.GPU()
 def generate_response(input_data, chat_history, max_new_tokens, temperature, top_p, top_k, repetition_penalty):
-    # بناء المحادثة بنمط الموديل العراقي
+    # نبني نص المحادثة مثل ما الموديل يحب
     conversation_parts = []
-
     if chat_history:
         for item in chat_history:
             role = item["role"]
             content = item["content"]
             if isinstance(content, list):
                 content = content[0]["text"] if content and "text" in content[0] else str(content)
-
             if role == "user":
                 conversation_parts.append(f"[|Human|] {content}")
             elif role == "assistant":
                 conversation_parts.append(f"[|AI|] {content}")
 
-    # أضف الرسالة الحالية
+    # أضف الرسالة الجديدة
     conversation_parts.append(f"[|Human|] {input_data}")
-
-    # إنشاء البرومبت الكامل
     conversation = "\n".join(conversation_parts)
+
+    # نحطها داخل البرومبت العراقي
     full_prompt = prompt_ar.format(Question=conversation)
 
     try:
         response = get_response(full_prompt)
 
-        # استخراج الرد الجديد فقط بعد ### Response :
+        # نقطع أي تعليمات أو وسوم من النتيجة
         if "### Response :" in response:
             response = response.split("### Response :")[-1].strip()
-
-        # إزالة أي محتوى إضافي أو markers
-        if "[|Human|]" in response:
-            response = response.split("[|Human|]")[0].strip()
         if "[|AI|]" in response:
             response = response.replace("[|AI|]", "").strip()
+        if "[|Human|]" in response:
+            response = response.split("[|Human|]")[0].strip()
+        if "### Instruction:" in response:
+            response = response.split("### Instruction:")[-1].strip()
+
+        # نرجع الرد فقط
+        if not response:
+            response = "هلا، ويّاك أليكس من دعم TechSolutions، شلون أگدر أساعدك اليوم؟"
 
-        # ✅ رجّع فقط الرد، مو البرومبت الكامل
-        return response if response else "هلا، ويّاك أليكس من دعم TechSolutions، شلون أگدر أساعدك اليوم؟"
+        return response
 
     except Exception as e:
         print(f"Error in generate_response: {e}")