Update app.py

app.py

@@ -1,59 +1,275 @@
 import gradio as gr
 import requests
 import json
+import os
+from datetime import datetime
 
-# رابط النموذج
-MODEL_ID = "kawkabelaloom/astramind"
+# إعدادات
+MODEL_NAME = "kawkabelaloom/astramind"  # نموذجك
+HF_TOKEN = os.environ.get("HF_TOKEN", "")  # ضعه في Space Secrets
+API_URL = f"https://router.huggingface.co/hf-inference/models/{MODEL_NAME}"
 
-def test_model_directly():
-    """اختبار مباشر للنموذج"""
-    print("🔍 اختبار تحميل الملفات...")
-    
-    # تحقق من الملفات مباشرة
-    base_url = "https://huggingface.co/kawkabelaloom/astramind/raw/main"
+# إعداد headers إذا كان هناك token
+headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
+
+class AstramindChatbot:
+    def __init__(self):
+        self.conversation_history = []
+        self.system_prompt = """أنت مساعد عربي ذكي اسمك "أسترا". 
+أجب بلغة عربية واضحة ومفيدة. كن ودوداً ومتعاوناً."""
     
-    files_to_check = [
-        "config.json",
-        "tokenizer.json",
-        "model.safetensors.index.json"
-    ]
+    def format_message(self, user_message):
+        """تنسيق الرسالة للنموذج"""
+        # إضافة الرسالة للتاريخ
+        self.conversation_history.append({"role": "user", "content": user_message})
+        
+        # بناء prompt مع التاريخ
+        prompt = self.system_prompt + "\n\n"
+        
+        # إضافة آخر 5 رسائل من التاريخ
+        for msg in self.conversation_history[-5:]:
+            role = "المستخدم" if msg["role"] == "user" else "أنت"
+            prompt += f"{role}: {msg['content']}\n"
+        
+        prompt += "أنت:"
+        return prompt
     
-    for file in files_to_check:
+    def send_to_model(self, prompt):
+        """إرسال للنموذج"""
         try:
-            response = requests.get(f"{base_url}/{file}", timeout=10)
-            print(f"✅ {file}: {response.status_code} ({len(response.text)} chars)")
-        except:
-            print(f"❌ {file}: غير موجود")
-    
-    print("\n🔍 اختبار Inference...")
-    
-    # جرب API مباشرة
-    api_url = f"https://router.huggingface.co/hf-inference/models/{MODEL_ID}"
-    
-    # أولاً بدون token
-    response = requests.post(api_url, json={"inputs": "مرحبا"}, timeout=15)
-    print(f"📊 بدون Token: {response.status_code}")
-    
-    # إذا مش شغال، جرب مع token
-    if response.status_code != 200:
-        print("\n🔑 جرب مع Token...")
-        # أضف token هنا إذا عندك
-        token = ""  # ضع token إذا عندك
-        if token:
-            headers = {"Authorization": f"Bearer {token}"}
-            response = requests.post(api_url, headers=headers, json={"inputs": "اهلا"}, timeout=15)
-            print(f"📊 مع Token: {response.status_code}")
-    
-    return "✅ تم الاختبار - شاهد النتائج في logs"
+            payload = {
+                "inputs": prompt,
+                "parameters": {
+                    "max_new_tokens": 300,
+                    "temperature": 0.7,
+                    "top_p": 0.9,
+                    "return_full_text": False,
+                    "do_sample": True
+                }
+            }
+            
+            response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
+            
+            if response.status_code == 200:
+                result = response.json()
+                
+                # معالجة الرد
+                if isinstance(result, list) and len(result) > 0:
+                    if 'generated_text' in result[0]:
+                        full_response = result[0]['generated_text']
+                        # استخراج رد المساعد فقط
+                        if prompt in full_response:
+                            bot_response = full_response.replace(prompt, "").strip()
+                        else:
+                            bot_response = full_response
+                        
+                        # إضافة للتاريخ
+                        self.conversation_history.append({"role": "assistant", "content": bot_response})
+                        return bot_response
+                
+                return "عذراً، لم أفهم الرد."
+                
+            elif response.status_code == 503:
+                return "🔄 النموذج بيتحمل الآن، جرب مرة أخرى بعد قليل..."
+            else:
+                return f"⚠️ خطأ: {response.status_code}"
+                
+        except requests.exceptions.Timeout:
+            return "⏳ النموذج يأخذ وقتاً، يرجى الانتظار..."
+        except Exception as e:
+            return f"❌ خطأ: {str(e)}"
+    
+    def chat(self, user_message, history):
+        """الدردشة مع البوت"""
+        if not user_message.strip():
+            return history, "⚠️ الرجاء كتابة رسالة."
+        
+        # تنسيق الرسالة
+        prompt = self.format_message(user_message)
+        
+        # إرسال للنموذج
+        bot_response = self.send_to_model(prompt)
+        
+        # تحديث التاريخ
+        history.append((user_message, bot_response))
+        
+        return history, ""
 
-# واجهة بسيطة
-with gr.Blocks() as demo:
-    gr.Markdown("# 🔍 اختبار نموذج Astramind")
-    gr.Markdown("جاري اختبار النموذج على Hugging Face...")
+# إنشاء البوت
+chatbot = AstramindChatbot()
+
+# واجهة Gradio متقدمة
+with gr.Blocks(
+    title="🤖 Astramind - مساعد الدردشة الذكي",
+    theme=gr.themes.Soft(),
+    css="""
+    .chatbot { height: 500px !important; }
+    .message.user { background-color: #e3f2fd; }
+    .message.bot { background-color: #f3e5f5; }
+    """
+) as demo:
+    
+    # العنوان
+    gr.Markdown("""
+    # 🤖 Astramind - مساعد الدردشة الذكي
+    ### تحدث مع النموذج العربي الخاص بك مباشرة!
+    """)
+    
+    # معلومات النموذج
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown(f"""
+            **معلومات النموذج:**
+            - الاسم: `{MODEL_NAME}`
+            - الحالة: {'🔒 يحتاج Token' if HF_TOKEN else '🔓 Public'}
+            - آخر تحديث: {datetime.now().strftime('%Y-%m-%d %H:%M')}
+            """)
+        
+        with gr.Column(scale=2):
+            gr.Markdown("""
+            **تعليمات:**
+            1. اكتب رسالتك في المربع أدناه
+            2. اضغط Enter أو زر "إرسال"
+            3. انتظر رد البوت
+            """)
+    
+    # منطقة المحادثة
+    chatbot_ui = gr.Chatbot(
+        label="المحادثة",
+        height=400,
+        avatar_images=(None, "https://cdn-icons-png.flaticon.com/512/4712/4712035.png")
+    )
+    
+    # منطقة الإدخال
+    with gr.Row():
+        msg = gr.Textbox(
+            label="اكتب رسالتك هنا",
+            placeholder="مثال: السلام عليكم، كيف حالك؟",
+            scale=4,
+            lines=2
+        )
+        
+        with gr.Column(scale=1):
+            submit_btn = gr.Button("إرسال", variant="primary", size="lg")
+            clear_btn = gr.Button("🗑️ مسح", variant="secondary")
+    
+    # منطقة التحكم
+    with gr.Accordion("⚙️ الإعدادات المتقدمة", open=False):
+        with gr.Row():
+            max_tokens = gr.Slider(
+                minimum=50, maximum=500, value=200, step=50,
+                label="طول الرد"
+            )
+            temperature = gr.Slider(
+                minimum=0.1, maximum=1.0, value=0.7, step=0.1,
+                label="الإبداعية (Temperature)"
+            )
+        
+        with gr.Row():
+            test_btn = gr.Button("🔍 اختبار الاتصال بالنموذج")
+            status_text = gr.Textbox(label="حالة الاتصال", interactive=False)
+    
+    # أمثلة سريعة
+    gr.Markdown("### 💡 أمثلة سريعة:")
+    
+    examples = gr.Examples(
+        examples=[
+            ["السلام عليكم، من أنت؟"],
+            ["كيف يمكنني التسجيل في الدورة؟"],
+            ["أخبرني نكتة عربية"],
+            ["ما هي عاصمة مصر؟"],
+            ["اكتب لي قصيدة صغيرة عن العلم"],
+            ["كيف أتعلم البرمجة؟"]
+        ],
+        inputs=msg,
+        label="اضغط على أي مثال لتجربته"
+    )
+    
+    # وظائف التحكم
+    def process_message(message, history, max_tokens_val, temp_val):
+        """معالجة الرسالة"""
+        if not message.strip():
+            return history, "", "⚠️ الرجاء كتابة رسالة."
+        
+        # تحديث إعدادات البوت
+        chatbot.send_to_model = lambda prompt: requests.post(
+            API_URL, 
+            headers=headers, 
+            json={
+                "inputs": prompt,
+                "parameters": {
+                    "max_new_tokens": max_tokens_val,
+                    "temperature": temp_val,
+                    "top_p": 0.9,
+                    "return_full_text": False
+                }
+            },
+            timeout=30
+        )
+        
+        # الدردشة
+        new_history, _ = chatbot.chat(message, history)
+        return new_history, "", f"✅ تم الرد (الطول: {max_tokens_val}, الإبداعية: {temp_val})"
+    
+    def test_connection():
+        """اختبار اتصال بالنموذج"""
+        try:
+            test_response = requests.post(
+                API_URL,
+                headers=headers,
+                json={"inputs": "مرحبا"},
+                timeout=10
+            )
+            
+            if test_response.status_code == 200:
+                return "✅ الاتصال ناجح! النموذج شغال."
+            elif test_response.status_code == 401:
+                return "⚠️ يحتاج Token. أضف HF_TOKEN في Secrets."
+            else:
+                return f"❌ خطأ: {test_response.status_code}"
+                
+        except Exception as e:
+            return f"❌ خطأ اتصال: {str(e)}"
+    
+    def clear_chat():
+        """مسح المحادثة"""
+        chatbot.conversation_history = []
+        return [], "تم مسح المحادثة."
+    
+    # ربط الأحداث
+    msg.submit(
+        process_message,
+        [msg, chatbot_ui, max_tokens, temperature],
+        [chatbot_ui, msg, status_text]
+    )
+    
+    submit_btn.click(
+        process_message,
+        [msg, chatbot_ui, max_tokens, temperature],
+        [chatbot_ui, msg, status_text]
+    )
+    
+    clear_btn.click(
+        clear_chat,
+        outputs=[chatbot_ui, status_text]
+    )
     
-    test_btn = gr.Button("بدء الاختبار")
-    output = gr.Textbox(label="النتيجة")
+    test_btn.click(
+        test_connection,
+        outputs=status_text
+    )
     
-    test_btn.click(test_model_directly, outputs=output)
+    # رسالة ترحيبية
+    demo.load(
+        fn=lambda: "🚀 جاهز للدردشة! ابدأ بالكتابة...",
+        outputs=status_text
+    )
 
-demo.launch(debug=True)
+# تشغيل التطبيق
+if __name__ == "__main__":
+    demo.launch(
+        debug=True,
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=True  # لينشئ رابط عام
+    )