Update app.py

app.py

@@ -1,56 +1,31 @@
-import gradio as gr
-from transformers import pipeline
-import torch
-
-# تحميل النموذج من صفحتك
-model_name = "kawkabelaloom/astramind"
-
-print("🚀 جاري تحميل النموذج...")
-try:
-    # إنشاء pipeline للدردشة
-    generator = pipeline(
-        'text-generation',
-        model=model_name,
-        device=-1,  # استخدام CPU
-        torch_dtype=torch.float32
-    )
-    print("✅ تم تحميل النموذج!")
-except Exception as e:
-    print(f"❌ خطأ في التحميل: {e}")
-    generator = None
-
 def chat(message, history):
-    """دالة الدردشة"""
+    """دالة الدردشة المعدلة"""
     if generator is None:
         return "النموذج غير متاح حالياً"
     
-    # توليد الرد
+    # بناء رسالة بالصيغة الصحيحة (ChatML format)
+    formatted_message = f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
+    
     try:
+        # توليد الرد
         result = generator(
-            message,
-            max_length=200,
+            formatted_message,
+            max_new_tokens=200,
             temperature=0.7,
-            do_sample=True
+            do_sample=True,
+            stop_strings=["<|im_end|>"]
         )
-        return result[0]['generated_text']
+        
+        # استخراج الرد فقط
+        response_text = result[0]['generated_text']
+        
+        # إزالة prompt الأصلي
+        if formatted_message in response_text:
+            response_text = response_text.replace(formatted_message, "")
+        
+        # إزالة أي tags متبقية
+        response_text = response_text.replace("<|im_end|>", "").strip()
+        
+        return response_text
     except Exception as e:
-        return f"حدث خطأ: {str(e)}"
-
-# واجهة Gradio
-with gr.Blocks(title="Astramind Chatbot") as demo:
-    gr.Markdown("# 🤖 Astramind - روبوت الدردشة العربي")
-    gr.Markdown("نموذج ذكي للرد على استفساراتك باللغة العربية")
-    
-    chatbot = gr.Chatbot(height=400)
-    msg = gr.Textbox(label="اكتب رسالتك هنا...")
-    clear = gr.Button("مسح المحادثة")
-    
-    def respond(message, chat_history):
-        bot_message = chat(message, chat_history)
-        chat_history.append((message, bot_message))
-        return "", chat_history
-    
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
-    clear.click(lambda: None, None, chatbot, queue=False)
-
-demo.launch(debug=True)
+        return f"حدث خطأ: {str(e)}"