Update app.py

app.py

@@ -4,12 +4,13 @@ import json
 import os
 from datetime import datetime
 
-# إعدادات
-MODEL_NAME = "kawkabelaloom/astramind"  # نموذجك
-HF_TOKEN = os.environ.get("HF_TOKEN", "")  # ضعه في Space Secrets
+# === إعدادات آمنة ===
+# استخدم Environment Variables (Secrets) بدل وضع Token في الكود
+MODEL_NAME = os.environ.get("MODEL_NAME", "kawkabelaloom/astramind")
+HF_TOKEN = os.environ.get("HF_TOKEN", "")  # سيأتي من Secrets
 API_URL = f"https://router.huggingface.co/hf-inference/models/{MODEL_NAME}"
 
-# إعداد headers إذا كان هناك token
+# إعداد headers بشكل آمن
 headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
 
 class AstramindChatbot:
@@ -20,256 +21,189 @@ class AstramindChatbot:
     
     def format_message(self, user_message):
         """تنسيق الرسالة للنموذج"""
-        # إضافة الرسالة للتاريخ
-        self.conversation_history.append({"role": "user", "content": user_message})
-        
-        # بناء prompt مع التاريخ
-        prompt = self.system_prompt + "\n\n"
+        if not self.conversation_history:
+            prompt = f"{self.system_prompt}\n\nالمستخدم: {user_message}\nأنت:"
+        else:
+            prompt = self.system_prompt + "\n\n"
+            for msg in self.conversation_history[-3:]:
+                role = "المستخدم" if msg["role"] == "user" else "أنت"
+                prompt += f"{role}: {msg['content']}\n"
+            prompt += f"المستخدم: {user_message}\nأنت:"
         
-        # إضافة آخر 5 رسائل من التاريخ
-        for msg in self.conversation_history[-5:]:
-            role = "المستخدم" if msg["role"] == "user" else "أنت"
-            prompt += f"{role}: {msg['content']}\n"
-        
-        prompt += "أنت:"
+        self.conversation_history.append({"role": "user", "content": user_message})
         return prompt
     
-    def send_to_model(self, prompt):
-        """إرسال للنموذج"""
+    def send_to_model(self, prompt, max_tokens=200, temperature=0.7):
+        """إرسال للنموذج - بدون token إذا مش متاح"""
         try:
             payload = {
                 "inputs": prompt,
                 "parameters": {
-                    "max_new_tokens": 300,
-                    "temperature": 0.7,
+                    "max_new_tokens": max_tokens,
+                    "temperature": temperature,
                     "top_p": 0.9,
                     "return_full_text": False,
                     "do_sample": True
                 }
             }
             
-            response = requests.post(API_URL, headers=headers, json=payload, timeout=30)
+            # إذا بدون token، جرب نموذج عام بديل
+            if not HF_TOKEN:
+                # استخدم نموذج Qwen العام بدون token
+                alt_url = "https://router.huggingface.co/hf-inference/models/Qwen/Qwen2.5-7B-Instruct"
+                response = requests.post(alt_url, json=payload, timeout=20)
+            else:
+                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
+                        # استخراج الرد الأخير
+                        lines = full_response.split('\n')
+                        for line in reversed(lines):
+                            if line.strip() and not line.startswith('المستخدم:'):
+                                bot_response = line.replace('أنت:', '').strip()
+                                if bot_response:
+                                    self.conversation_history.append({"role": "assistant", "content": bot_response})
+                                    return bot_response
                 
                 return "عذراً، لم أفهم الرد."
                 
+            elif response.status_code in [401, 403]:
+                return "⚠️ يبدو أن هناك مشكلة في التوكن. جاري استخدام نموذج بديل..."
             elif response.status_code == 503:
-                return "🔄 النموذج بيتحمل الآن، جرب مرة أخرى بعد قليل..."
+                return "🔄 النموذج مشغول الآن، جرب مرة أخرى بعد قليل."
             else:
-                return f"⚠️ خطأ: {response.status_code}"
+                return f"⚠️ حدث خطأ: {response.status_code}"
                 
-        except requests.exceptions.Timeout:
-            return "⏳ النموذج يأخذ وقتاً، يرجى الانتظار..."
         except Exception as e:
-            return f"❌ خطأ: {str(e)}"
+            return f"❌ خطأ في الاتصال: {str(e)[:100]}"
     
-    def chat(self, user_message, history):
+    def chat(self, user_message, max_tokens=200, temperature=0.7):
         """الدردشة مع البوت"""
         if not user_message.strip():
-            return history, "⚠️ الرجاء كتابة رسالة."
+            return "⚠️ الرجاء كتابة رسالة."
         
-        # تنسيق الرسالة
         prompt = self.format_message(user_message)
-        
-        # إرسال للنموذج
-        bot_response = self.send_to_model(prompt)
-        
-        # تحديث التاريخ
-        history.append((user_message, bot_response))
-        
-        return history, ""
+        bot_response = self.send_to_model(prompt, max_tokens, temperature)
+        return bot_response
 
 # إنشاء البوت
 chatbot = AstramindChatbot()
 
-# واجهة Gradio متقدمة
+# === واجهة 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; }
-    """
+    title="🤖 Astramind - مساعد الدردشة",
+    theme=gr.themes.Soft()
 ) 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. انتظر رد البوت
-            """)
+        gr.Markdown(f"""
+        **المعلومات:**
+        - النموذج: `{MODEL_NAME if HF_TOKEN else 'Qwen/Qwen2.5-7B-Instruct (بديل)'}`
+        - الحالة: {'🔐 باستخدام Token' if HF_TOKEN else '🔓 بدون Token (بديل)'}
+        - الوقت: {datetime.now().strftime('%H:%M')}
+        """)
     
     # منطقة المحادثة
-    chatbot_ui = gr.Chatbot(
-        label="المحادثة",
-        height=400,
-        avatar_images=(None, "https://cdn-icons-png.flaticon.com/512/4712/4712035.png")
-    )
+    chatbot_display = gr.Chatbot(label="المحادثة", height=400)
     
     # منطقة الإدخال
     with gr.Row():
-        msg = gr.Textbox(
-            label="اكتب رسالتك هنا",
-            placeholder="مثال: السلام عليكم، كيف حالك؟",
-            scale=4,
-            lines=2
+        user_input = gr.Textbox(
+            label="رسالتك",
+            placeholder="اكتب سؤالك هنا...",
+            lines=2,
+            scale=4
         )
-        
-        with gr.Column(scale=1):
-            submit_btn = gr.Button("إرسال", variant="primary", size="lg")
-            clear_btn = gr.Button("🗑️ مسح", variant="secondary")
+        send_btn = gr.Button("إرسال", variant="primary")
+        clear_btn = gr.Button("مسح", variant="secondary")
     
-    # منطقة التحكم
-    with gr.Accordion("⚙️ الإعدادات المتقدمة", open=False):
+    # إعدادات
+    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)
+            max_tokens = gr.Slider(50, 500, value=200, label="طول الرد")
+            temperature = gr.Slider(0.1, 1.0, value=0.7, label="الإبداعية")
     
-    # أمثلة سريعة
+    # أمثلة
     gr.Markdown("### 💡 أمثلة سريعة:")
     
     examples = gr.Examples(
         examples=[
-            ["السلام عليكم، من أنت؟"],
-            ["كيف يمكنني التسجيل في الدورة؟"],
-            ["أخبرني نكتة عربية"],
-            ["ما هي عاصمة مصر؟"],
-            ["اكتب لي قصيدة صغيرة عن العلم"],
-            ["كيف أتعلم البرمجة؟"]
+            ["السلام عليكم"],
+            ["ما هو اسمك؟"],
+            ["كيف يمكنني التسجيل؟"],
+            ["أخبرني نكتة"]
         ],
-        inputs=msg,
-        label="اضغط على أي مثال لتجربته"
+        inputs=user_input,
+        label="جرب أحد هذه الأمثلة"
     )
     
-    # وظائف التحكم
-    def process_message(message, history, max_tokens_val, temp_val):
-        """معالجة الرسالة"""
+    # متغير حالة
+    status = gr.Textbox(label="الحالة", interactive=False)
+    
+    # وظائف
+    def respond(message, history, tokens, temp):
         if not message.strip():
-            return history, "", "⚠️ الرجاء كتابة رسالة."
+            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
-        )
+        response = chatbot.chat(message, tokens, temp)
         
-        # الدردشة
-        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)}"
+        # تحديث المحادثة
+        if history is None:
+            history = []
+        history.append((message, response))
+        
+        # تحديث الحالة
+        model_used = MODEL_NAME if HF_TOKEN else "Qwen (بديل)"
+        status_msg = f"✅ تم الرد باستخدام: {model_used}"
+        
+        return history, "", status_msg
     
-    def clear_chat():
-        """مسح المحادثة"""
+    def clear_history():
         chatbot.conversation_history = []
-        return [], "تم مسح المحادثة."
+        return [], "تم مسح المحادثة"
     
     # ربط الأحداث
-    msg.submit(
-        process_message,
-        [msg, chatbot_ui, max_tokens, temperature],
-        [chatbot_ui, msg, status_text]
+    user_input.submit(
+        respond,
+        [user_input, chatbot_display, max_tokens, temperature],
+        [chatbot_display, user_input, status]
     )
     
-    submit_btn.click(
-        process_message,
-        [msg, chatbot_ui, max_tokens, temperature],
-        [chatbot_ui, msg, status_text]
+    send_btn.click(
+        respond,
+        [user_input, chatbot_display, max_tokens, temperature],
+        [chatbot_display, user_input, status]
     )
     
     clear_btn.click(
-        clear_chat,
-        outputs=[chatbot_ui, status_text]
+        clear_history,
+        outputs=[chatbot_display, status]
     )
     
-    test_btn.click(
-        test_connection,
-        outputs=status_text
-    )
+    # رسالة ترحيب
+    def welcome_message():
+        if HF_TOKEN:
+            return f"🚀 جاهز! النموذج: {MODEL_NAME}"
+        else:
+            return "🚀 جاهز! باستخدام نموذج Qwen البديل (بدون Token)"
     
-    # رسالة ترحيبية
-    demo.load(
-        fn=lambda: "🚀 جاهز للدردشة! ابدأ بالكتابة...",
-        outputs=status_text
-    )
+    demo.load(welcome_message, outputs=status)
 
-# تشغيل التطبيق
+# التشغيل
 if __name__ == "__main__":
     demo.launch(
         debug=True,
         server_name="0.0.0.0",
-        server_port=7860,
-        share=True  # لينشئ رابط عام
+        server_port=7860
     )