Update app.py

app.py

@@ -1,31 +1,85 @@
-def chat(message, history):
-    """دالة الدردشة المعدلة"""
-    if generator is None:
-        return "النموذج غير متاح حالياً"
+import gradio as gr
+from transformers import pipeline
+import torch
+
+# تحميل النموذج
+model_name = "kawkabelaloom/astramind"
+
+print("🚀 جاري تهيئة النموذج...")
+
+try:
+    # استخدم pipeline مع إعدادات Qwen
+    generator = pipeline(
+        "text-generation",
+        model=model_name,
+        tokenizer=model_name,
+        device=-1,  # CPU
+        torch_dtype=torch.float32,
+        model_kwargs={
+            "trust_remote_code": True,
+            "low_cpu_mem_usage": True
+        }
+    )
+    print("✅ النموذج جاهز!")
     
-    # بناء رسالة بالصيغة الصحيحة (ChatML format)
-    formatted_message = f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
+except Exception as e:
+    print(f"❌ خطأ: {e}")
+    generator = None
+
+def simple_chat(message):
+    """دردشة مبسطة بدون chat format"""
+    if not generator:
+        return "النموذج غير متاح"
     
     try:
-        # توليد الرد
+        # رسالة بسيطة بدون تنسيق معقد
+        prompt = f"المستخدم: {message}\nالمساعد:"
+        
         result = generator(
-            formatted_message,
-            max_new_tokens=200,
+            prompt,
+            max_new_tokens=150,
             temperature=0.7,
             do_sample=True,
-            stop_strings=["<|im_end|>"]
+            repetition_penalty=1.1
         )
         
-        # استخراج الرد فقط
-        response_text = result[0]['generated_text']
+        # استخراج الرد
+        full_text = result[0]['generated_text']
         
-        # إزالة prompt الأصلي
-        if formatted_message in response_text:
-            response_text = response_text.replace(formatted_message, "")
+        # أخذ جزء المساعد فقط
+        if "المساعد:" in full_text:
+            response = full_text.split("المساعد:")[-1].strip()
+        else:
+            response = full_text.replace(prompt, "").strip()
         
-        # إزالة أي tags متبقية
-        response_text = response_text.replace("<|im_end|>", "").strip()
+        return response
         
-        return response_text
     except Exception as e:
-        return f"حدث خطأ: {str(e)}"
+        return f"عذراً، حدث خطأ: {str(e)}"
+
+# إنشاء واجهة بسيطة
+interface = gr.Interface(
+    fn=simple_chat,
+    inputs=gr.Textbox(
+        label="اكتب رسالتك",
+        placeholder="السلام عليكم...",
+        lines=3
+    ),
+    outputs=gr.Textbox(
+        label="رد المساعد",
+        lines=5
+    ),
+    title="🤖 Astramind - مساعد عربي ذكي",
+    description="اسأل أي سؤال وسأحاول الإجابة عليه باللغة العربية",
+    examples=[
+        ["مرحباً، من أنت؟"],
+        ["كيف يمكنك مساعدتي؟"],
+        ["أخبرني نكتة عربية"],
+        ["ما هي عاصمة مصر؟"]
+    ],
+    theme="soft"
+)
+
+# تشغيل
+if __name__ == "__main__":
+    interface.launch(debug=True, share=False)