updatE

app.py

@@ -127,10 +127,9 @@ def detect_language(text):
 
 @spaces.GPU()
 def generate_response(input_data, chat_history, max_new_tokens, temperature, top_p, top_k, repetition_penalty):
-    # Build conversation for Iraqi model format
+    # بناء المحادثة بنمط الموديل العراقي
     conversation_parts = []
 
-    # Add chat history
     if chat_history:
         for item in chat_history:
             role = item["role"]
@@ -143,33 +142,34 @@ def generate_response(input_data, chat_history, max_new_tokens, temperature, top
             elif role == "assistant":
                 conversation_parts.append(f"[|AI|] {content}")
 
-    # Add current user message
+    # أضف الرسالة الحالية
     conversation_parts.append(f"[|Human|] {input_data}")
 
-    # Join conversation
+    # إنشاء البرومبت الكامل
     conversation = "\n".join(conversation_parts)
-
-    # Create full prompt using the Iraqi Arabic prompt template
     full_prompt = prompt_ar.format(Question=conversation)
 
     try:
-        # استخدام دالة get_response من documentation
         response = get_response(full_prompt)
 
-        # استخراج الرد الجديد فقط (بعد "### Response :")
+        # استخراج الرد الجديد فقط بعد ### Response :
         if "### Response :" in response:
             response = response.split("### Response :")[-1].strip()
 
-        if not response:
-            response = "أهلاً! أنا أليكس مساعد خدمة العملاء. كيف أقدر أساعدك اليوم؟"
+        # إزالة أي محتوى إضافي أو markers
+        if "[|Human|]" in response:
+            response = response.split("[|Human|]")[0].strip()
+        if "[|AI|]" in response:
+            response = response.replace("[|AI|]", "").strip()
 
-        yield response
+        # ✅ رجّع فقط الرد، مو البرومبت الكامل
+        return response if response else "هلا، ويّاك أليكس من دعم TechSolutions، شلون أگدر أساعدك اليوم؟"
 
     except Exception as e:
         print(f"Error in generate_response: {e}")
         import traceback
         print(traceback.format_exc())
-        yield "أهلاً! أنا أليكس مساعد خدمة العملاء. كيف أقدر أساعدك اليوم؟"
+        return "هلا، ويّاك أليكس من دعم TechSolutions، شلون أگدر أساعدك اليوم؟"
 
 demo = gr.ChatInterface(
     fn=generate_response,