Update app.py

app.py

@@ -8,7 +8,7 @@ from datetime import datetime
 # استخدم 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}"
+API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 
 # إعداد headers بشكل آمن
 headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
@@ -19,85 +19,144 @@ class AstramindChatbot:
         self.system_prompt = """أنت مساعد عربي ذكي اسمك "أسترا". 
 أجب بلغة عربية واضحة ومفيدة. كن ودوداً ومتعاوناً."""
     
-    def format_message(self, user_message):
-        """تنسيق الرسالة للنموذج"""
-        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أنت:"
+    def format_messages_for_api(self):
+        """تنسيق الرسائل لـ Hugging Face Inference API"""
+        messages = []
         
-        self.conversation_history.append({"role": "user", "content": user_message})
-        return prompt
+        # إضافة الرسالة النظامية أولاً
+        messages.append({"role": "system", "content": self.system_prompt})
+        
+        # إضافة تاريخ المحادثة
+        for msg in self.conversation_history[-6:]:  # آخر 3 تبادلات
+            messages.append({"role": msg["role"], "content": msg["content"]})
+        
+        return messages
     
-    def send_to_model(self, prompt, max_tokens=200, temperature=0.7):
-        """إرسال للنموذج - بدون token إذا مش متاح"""
+    def send_to_model(self, user_message, max_tokens=200, temperature=0.7):
+        """إرسال للنموذج باستخدام Hugging Face Inference API"""
         try:
+            # تحديث تاريخ المحادثة أولاً
+            self.conversation_history.append({"role": "user", "content": user_message})
+            
+            # تنسيق الرسائل للـ API
+            messages = self.format_messages_for_api()
+            
             payload = {
-                "inputs": prompt,
+                "inputs": messages,
                 "parameters": {
                     "max_new_tokens": max_tokens,
                     "temperature": temperature,
                     "top_p": 0.9,
-                    "return_full_text": False,
                     "do_sample": True
                 }
             }
             
-            # إذا بدون 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)
+            # إرسال الطلب
+            response = requests.post(
+                API_URL, 
+                headers=headers, 
+                json=payload, 
+                timeout=60
+            )
+            
+            print(f"Status Code: {response.status_code}")  # للتصحيح
             
             if response.status_code == 200:
                 result = response.json()
                 
-                if isinstance(result, list) and len(result) > 0:
+                # معالجة الرد بناءً على تنسيق النموذج
+                if isinstance(result, list):
+                    # تنسيق الرد القديم
                     if 'generated_text' in result[0]:
-                        full_response = result[0]['generated_text']
-                        # استخراج الرد الأخير
-                        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
+                        bot_response = result[0]['generated_text']
+                    elif 'generated_texts' in result[0]:
+                        bot_response = result[0]['generated_texts'][0]
+                    else:
+                        # جرب استخراج ��لنص مباشرة
+                        bot_response = str(result[0]).strip()
+                elif isinstance(result, dict):
+                    # تنسيق الـ chat models
+                    if 'choices' in result:
+                        bot_response = result['choices'][0]['message']['content']
+                    elif 'generated_text' in result:
+                        bot_response = result['generated_text']
+                    else:
+                        bot_response = json.dumps(result)[:500]
+                else:
+                    bot_response = str(result)[:500]
+                
+                # تنظيف الرد
+                bot_response = bot_response.strip()
+                
+                # إضافة رد البوت للتاريخ
+                self.conversation_history.append({"role": "assistant", "content": bot_response})
                 
-                return "عذراً، لم أفهم الرد."
+                return bot_response
                 
-            elif response.status_code in [401, 403]:
-                return "⚠️ يبدو أن هناك مشكلة في التوكن. جاري استخدام نموذج بديل..."
             elif response.status_code == 503:
-                return "🔄 النموذج مشغول الآن، جرب مرة أخرى بعد قليل."
+                # النموذج يبدأ Loading
+                return "🔄 النموذج يبدأ التشغيل... جرب مرة أخرى خلال 30 ثانية."
+            elif response.status_code == 401:
+                return "🔒 خطأ في التوكن. الرجاء التحقق من HF_TOKEN."
             else:
-                return f"⚠️ حدث خطأ: {response.status_code}"
+                error_msg = f"⚠️ خطأ {response.status_code}: {response.text[:200]}"
+                print(error_msg)  # للتصحيح
+                return error_msg
                 
         except Exception as e:
-            return f"❌ خطأ في الاتصال: {str(e)[:100]}"
+            error_msg = f"❌ خطأ في الاتصال: {str(e)[:100]}"
+            print(f"Exception: {str(e)}")  # للتصحيح
+            return error_msg
     
     def chat(self, user_message, max_tokens=200, temperature=0.7):
         """الدردشة مع البوت"""
         if not user_message.strip():
             return "⚠️ الرجاء كتابة رسالة."
         
-        prompt = self.format_message(user_message)
-        bot_response = self.send_to_model(prompt, max_tokens, temperature)
+        bot_response = self.send_to_model(user_message, max_tokens, temperature)
         return bot_response
 
 # إنشاء البوت
 chatbot = AstramindChatbot()
 
 # === واجهة Gradio ===
+def respond(message, history, tokens, temp):
+    """معالجة الرد وعرضه في Chatbot"""
+    if not message.strip():
+        return history, "", "اكتب رسالة أولاً"
+    
+    # الحصول على الرد من البوت
+    response = chatbot.chat(message, tokens, temp)
+    
+    # تحديث الـ history الخاص بـ Gradio
+    history = history or []  # تأكد أن history ليست None
+    history.append((message, response))
+    
+    # تحديث الحالة
+    status_msg = f"✅ تم الرد - طول: {len(response)} حرف"
+    
+    return history, "", status_msg
+
+def clear_chat():
+    """مسح المحادثة"""
+    chatbot.conversation_history = []
+    return [], "🗑️ تم مسح المحادثة"
+
+def welcome_message():
+    """رسالة ترحيبية"""
+    if HF_TOKEN:
+        return f"🤖 Astramind جاهز للدردشة! النموذج: {MODEL_NAME}"
+    else:
+        return "⚠️ الرجاء إضافة HF_TOKEN في Secrets لاستخدام النموذج الخاص"
+
+# إنشاء الواجهة
 with gr.Blocks(
     title="🤖 Astramind - مساعد الدردشة",
-    theme=gr.themes.Soft()
+    theme=gr.themes.Soft(),
+    css="""
+    .chatbot { min-height: 400px; }
+    .message { text-align: right; }
+    """
 ) as demo:
     
     gr.Markdown("""
@@ -105,105 +164,94 @@ with gr.Blocks(
     ### تحدث مع النموذج العربي الخاص بك
     """)
     
-    # معلومات
-    with gr.Row():
-        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')}
-        """)
+    # عرض معلومات النموذج
+    status_display = gr.Textbox(
+        label="حالة النظام",
+        value=welcome_message(),
+        interactive=False
+    )
+    
+    # شات بوت
+    chatbot_display = gr.Chatbot(
+        label="المحادثة",
+        height=400,
+        type="messages"
+    )
     
-    # منطقة المحادثة
-    chatbot_display = gr.Chatbot(label="المحادثة", height=400)
+    # ��ر المسح
+    clear_btn = gr.Button("🗑️ مسح المحادثة", variant="secondary", size="sm")
     
-    # منطقة الإدخال
+    # إدخال المستخدم
     with gr.Row():
         user_input = gr.Textbox(
-            label="رسالتك",
-            placeholder="اكتب سؤالك هنا...",
+            label="اكتب رسالتك",
+            placeholder="مرحباً! كيف يمكنني مساعدتك اليوم؟",
             lines=2,
             scale=4
         )
-        send_btn = gr.Button("إرسال", variant="primary")
-        clear_btn = gr.Button("مسح", variant="secondary")
+        send_btn = gr.Button("إرسال", variant="primary", size="lg")
     
     # إعدادات
-    with gr.Accordion("⚙️ الإعدادات", open=False):
+    with gr.Accordion("⚙️ إعدادات النموذج", open=False):
         with gr.Row():
-            max_tokens = gr.Slider(50, 500, value=200, label="طول الرد")
-            temperature = gr.Slider(0.1, 1.0, value=0.7, label="الإبداعية")
+            max_tokens = gr.Slider(
+                minimum=50,
+                maximum=1000,
+                value=200,
+                step=50,
+                label="طول الرد (tokens)"
+            )
+            temperature = gr.Slider(
+                minimum=0.1,
+                maximum=1.0,
+                value=0.7,
+                step=0.1,
+                label="درجة الإبداع"
+            )
     
-    # أمثلة
+    # أمثلة سريعة
     gr.Markdown("### 💡 أمثلة سريعة:")
-    
     examples = gr.Examples(
         examples=[
-            ["السلام عليكم"],
-            ["ما هو اسمك؟"],
-            ["كيف يمكنني التسجيل؟"],
-            ["أخبرني نكتة"]
+            ["السلام عليكم، كيف حالك؟"],
+            ["ما هو اسمك وماذا تفعل؟"],
+            ["أخبرني عن نفسك"],
+            ["ساعدني في كتابة رسالة بريد إلكتروني"]
         ],
         inputs=user_input,
         label="جرب أحد هذه الأمثلة"
     )
     
-    # متغير حالة
-    status = gr.Textbox(label="الحالة", interactive=False)
-    
-    # وظائف
-    def respond(message, history, tokens, temp):
-        if not message.strip():
-            return history, "", "اكتب رسالة أولاً"
-        
-        response = chatbot.chat(message, tokens, temp)
-        
-        # تحديث المحادثة
-        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_history():
-        chatbot.conversation_history = []
-        return [], "تم مسح المحادثة"
-    
     # ربط الأحداث
-    user_input.submit(
-        respond,
-        [user_input, chatbot_display, max_tokens, temperature],
-        [chatbot_display, user_input, status]
+    # إرسال بالزر
+    send_btn.click(
+        fn=respond,
+        inputs=[user_input, chatbot_display, max_tokens, temperature],
+        outputs=[chatbot_display, user_input, status_display]
     )
     
-    send_btn.click(
-        respond,
-        [user_input, chatbot_display, max_tokens, temperature],
-        [chatbot_display, user_input, status]
+    # إرسال بالإنتر
+    user_input.submit(
+        fn=respond,
+        inputs=[user_input, chatbot_display, max_tokens, temperature],
+        outputs=[chatbot_display, user_input, status_display]
     )
     
+    # مسح المحادثة
     clear_btn.click(
-        clear_history,
-        outputs=[chatbot_display, status]
+        fn=clear_chat,
+        outputs=[chatbot_display, status_display]
     )
     
-    # رسالة ترحيب
-    def welcome_message():
-        if HF_TOKEN:
-            return f"🚀 جاهز! النموذج: {MODEL_NAME}"
-        else:
-            return "🚀 جاهز! باستخدام نموذج Qwen البديل (بدون Token)"
-    
-    demo.load(welcome_message, outputs=status)
+    # تحديث الحالة عند التحميل
+    demo.load(welcome_message, outputs=status_display)
 
 # التشغيل
 if __name__ == "__main__":
+    # إعدادات التشغيل على Hugging Face Space
     demo.launch(
-        debug=True,
+        debug=False,
         server_name="0.0.0.0",
-        server_port=7860
+        server_port=7860,
+        share=False
     )