Update app.py

app.py

@@ -1,114 +1,130 @@
 import gradio as gr
 import requests
 import os
+import json
 from datetime import datetime
 
-# === إعدادات ===
+# ===============================
+# 🔧 الإعدادات
+# ===============================
 MODEL_NAME = os.environ.get("MODEL_NAME", "kawkabelaloom/astramind")
 HF_TOKEN = os.environ.get("HF_TOKEN", "")
 API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 
-headers = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
+headers = {
+    "Authorization": f"Bearer {HF_TOKEN}",
+    "Content-Type": "application/json"
+} if HF_TOKEN else {}
 
+# ===============================
+# 🤖 Chatbot Class
+# ===============================
 class AstramindChatbot:
     def __init__(self):
         self.conversation_history = []
-        self.system_prompt = """أنت مساعد عربي ذكي اسمك "أسترا". أجب بلغة عربية واضحة ومفيدة. كن ودوداً ومتعاوناً."""
-    
-    def format_prompt(self, user_message):
-        """تنسيق المحادثة كـ نص بسيط"""
-        prompt = self.system_prompt + "\n\n"
-        
-        # إضافة آخر 3 محادثات
-        for msg in self.conversation_history[-6:]:
-            if msg["role"] == "user":
-                prompt += f"المستخدم: {msg['content']}\n"
-            else:
-                prompt += f"المساعد: {msg['content']}\n"
-        
-        prompt += f"المستخدم: {user_message}\nالمساعد:"
-        
-        # تحديث التاريخ
-        self.conversation_history.append({"role": "user", "content": user_message})
-        
-        return prompt
-    
-def send_to_model(self, user_message, max_tokens=200, temperature=0.7):
-    try:
-        messages = [
-            {"role": "system", "content": self.system_prompt}
-        ]
-
-        # آخر 3 رسائل
-        for msg in self.conversation_history[-6:]:
-            messages.append({
-                "role": msg["role"],
-                "content": msg["content"]
-            })
-
-        messages.append({"role": "user", "content": user_message})
-
-        payload = {
-            "messages": messages,
-            "parameters": {
-                "max_new_tokens": max_tokens,
-                "temperature": temperature,
-                "top_p": 0.9
+        self.system_prompt = (
+            "أنت مساعد عربي ذكي اسمك أسترا. "
+            "تحدث بلغة عربية واضحة، مختصرة ومفيدة."
+        )
+
+    def send_to_model(self, user_message, max_tokens=200, temperature=0.7):
+        try:
+            # ===== تجهيز الرسائل =====
+            messages = [{"role": "system", "content": self.system_prompt}]
+
+            for msg in self.conversation_history[-6:]:
+                messages.append({
+                    "role": msg["role"],
+                    "content": msg["content"]
+                })
+
+            messages.append({"role": "user", "content": user_message})
+
+            payload = {
+                "messages": messages,
+                "parameters": {
+                    "max_new_tokens": max_tokens,
+                    "temperature": temperature,
+                    "top_p": 0.9
+                }
             }
-        }
 
-        response = requests.post(
-            API_URL,
-            headers=headers,
-            json=payload,
-            timeout=60
-        )
+            # ===== Debug Logs =====
+            print("\n==============================")
+            print("📤 REQUEST PAYLOAD:")
+            print(json.dumps(payload, ensure_ascii=False, indent=2))
+            print("==============================")
+
+            response = requests.post(
+                API_URL,
+                headers=headers,
+                json=payload,
+                timeout=60
+            )
 
-        if response.status_code == 200:
-            result = response.json()
+            print(f"📥 STATUS CODE: {response.status_code}")
+            print(f"📥 RAW RESPONSE: {response.text[:500]}")
 
-            # استخراج الرد
-            if isinstance(result, dict) and "choices" in result:
-                bot_response = result["choices"][0]["message"]["content"]
-            else:
-                bot_response = str(result)
+            # ===== Success =====
+            if response.status_code == 200:
+                result = response.json()
 
-            self.conversation_history.append(
-                {"role": "assistant", "content": bot_response}
-            )
+                # HuggingFace Chat format
+                if isinstance(result, dict) and "choices" in result:
+                    bot_response = result["choices"][0]["message"]["content"]
 
-            return bot_response.strip()
+                # Fallback
+                else:
+                    bot_response = str(result)
 
-        elif response.status_code == 503:
-            return "🔄 النموذج جاري التحميل... حاول بعد دقيقة"
+                self.conversation_history.append(
+                    {"role": "assistant", "content": bot_response}
+                )
 
-        else:
-            return f"⚠️ Error {response.status_code}: {response.text[:200]}"
+                return bot_response.strip()
 
-    except Exception as e:
-        return f"❌ Exception: {str(e)}"
+            # ===== Model Loading =====
+            elif response.status_code == 503:
+                return "🔄 النموذج جاري التحميل… حاول مرة أخرى بعد دقيقة."
+
+            # ===== Any Other Error =====
+            else:
+                return (
+                    f"❌ API ERROR\n"
+                    f"Status: {response.status_code}\n"
+                    f"Response: {response.text}"
+                )
+
+        except Exception as e:
+            return (
+                "❌ PYTHON EXCEPTION\n"
+                f"{str(e)}"
+            )
 
     def chat(self, user_message, max_tokens=200, temperature=0.7):
         if not user_message.strip():
             return "⚠️ الرجاء كتابة رسالة."
-        
         return self.send_to_model(user_message, max_tokens, temperature)
 
-# إنشاء البوت
+# ===============================
+# 🚀 إنشاء البوت
+# ===============================
 chatbot = AstramindChatbot()
 
-# === واجهة Gradio ===
+# ===============================
+# 🎛️ Gradio Functions
+# ===============================
 def respond(message, history, tokens, temp):
-    if not message.strip():
-        return history, "", "اكتب رسالة أولاً"
-    
-    response = chatbot.chat(message, tokens, temp)
-    
     history = history or []
+
+    response = chatbot.chat(message, tokens, temp)
     history.append((message, response))
-    
-    status_msg = f"✅ تم الرد ({len(response)} حرف)"
-    return history, "", status_msg
+
+    status = "✅ تم التنفيذ"
+    if response.startswith("❌"):
+        status = "🚨 يوجد خطأ — راجع الرسالة"
+
+    return history, "", status
 
 def clear_chat():
     chatbot.conversation_history = []
@@ -116,99 +132,81 @@ def clear_chat():
 
 def welcome_message():
     if HF_TOKEN:
-        return f"🤖 Astramind جاهز! النموذج: {MODEL_NAME}"
-    else:
-        return "⚠️ يرجى إضافة HF_TOKEN في Secrets"
+        return f"🤖 Astramind Ready | Model: {MODEL_NAME}"
+    return "⚠️ HF_TOKEN غير موجود في Secrets"
 
-# إنشاء التطبيق
+# ===============================
+# 🖥️ Gradio UI
+# ===============================
 with gr.Blocks(title="Astramind Chatbot") as demo:
-    
-    gr.Markdown("# 🤖 Astramind - مساعد الدردشة")
-    gr.Markdown("### تحدث مع النموذج العربي")
-    
-    # حالة النظام
+    gr.Markdown("# 🤖 Astramind")
+    gr.Markdown("### مساعد دردشة عربي مبني على HuggingFace")
+
     status_box = gr.Textbox(
         value=welcome_message(),
         label="الحالة",
         interactive=False
     )
-    
-    # شات بوت
+
     chatbot_ui = gr.Chatbot(
         label="المحادثة",
-        height=400,
-        show_label=True
+        height=400
     )
-    
-    # إدخال الرسالة
+
     with gr.Row():
         message_input = gr.Textbox(
             placeholder="اكتب رسالتك هنا...",
-            label="رسالتك",
             lines=2,
             scale=4
         )
         send_btn = gr.Button("إرسال", variant="primary")
-    
-    # أزرار التحكم
+
     with gr.Row():
-        clear_btn = gr.Button("مسح المحادثة", variant="secondary")
-        gr.Button("🔄", size="sm").click(
-            fn=welcome_message,
-            outputs=status_box
-        )
-    
-    # إعدادات
+        clear_btn = gr.Button("مسح المحادثة")
+        refresh_btn = gr.Button("🔄")
+
     with gr.Accordion("⚙️ الإعدادات", open=False):
-        max_tokens = gr.Slider(
-            50, 500, value=200,
-            label="طول الرد",
-            step=10
-        )
-        temperature = gr.Slider(
-            0.1, 1.0, value=0.7,
-            label="الإبداعية",
-            step=0.1
-        )
-    
-    # أمثلة
+        max_tokens = gr.Slider(50, 500, value=200, step=10, label="طول الرد")
+        temperature = gr.Slider(0.1, 1.0, value=0.7, step=0.1, label="الإبداعية")
+
     gr.Examples(
         examples=[
             ["السلام عليكم"],
-            ["ما هو اسمك؟"],
-            ["أخبرني عن نفسك"],
-            ["كيف حالك اليوم؟"]
+            ["من أنت؟"],
+            ["احكي لي عن نفسك"],
+            ["كيف أتعلم الذكاء الاصطناعي؟"]
         ],
-        inputs=message_input,
-        label="💡 أمثلة سريعة"
+        inputs=message_input
     )
-    
-    # أحداث
-    def submit_message(message, history, tokens, temp):
-        return respond(message, history, tokens, temp)
-    
-    # ربط الأحداث
+
     send_btn.click(
-        fn=respond,
+        respond,
         inputs=[message_input, chatbot_ui, max_tokens, temperature],
         outputs=[chatbot_ui, message_input, status_box]
     )
-    
+
     message_input.submit(
-        fn=respond,
+        respond,
         inputs=[message_input, chatbot_ui, max_tokens, temperature],
         outputs=[chatbot_ui, message_input, status_box]
     )
-    
+
     clear_btn.click(
-        fn=clear_chat,
+        clear_chat,
         outputs=[chatbot_ui, status_box]
     )
 
-# التشغيل
+    refresh_btn.click(
+        fn=welcome_message,
+        outputs=status_box
+    )
+
+# ===============================
+# ▶️ Run
+# ===============================
 if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
         debug=True
-    )
+    )