Update app.py

app.py

@@ -1,65 +1,71 @@
 import os
 import requests
-import torch   # ← أضفناه هنا
+import torch  # نتأكد إنه متوفر قبل استخدامه
 from fastapi import FastAPI, Request
 from transformers import pipeline
 
-deployed_repo_id = "maryaa4/my-arabic-sentiment-model"
+# ============= إعدادات البوت =============
+
+# ⚠️ مهم: لا تحطي التوكن الحقيقي في الكود النهائي
+# حطيه مؤقتًا الآن لو حابة تجربي، وبعدها جددي التوكن وحطيه كـ Secret من الإعدادات
+BOT_TOKEN = "PUT_YOUR_TELEGRAM_BOT_TOKEN_HERE"  # <-- غيريه بالتوكن حقك مؤقتًا
+
+TELEGRAM_API_URL = f"https://api.telegram.org/bot{BOT_TOKEN}"
+
+# ============= تحميل مودل المشاعر =============
+
+MODEL_REPO_ID = "maryaa4/my-arabic-sentiment-model"
 
 sentiment_pipeline = pipeline(
     "sentiment-analysis",
-    model=deployed_repo_id,
+    model=MODEL_REPO_ID,
     trust_remote_code=True
 )
 
-def predict_sentiment(text: str) -> str:
-    if not text or text.strip() == "":
-        return "رجاءً أرسل نص عربي عشان أقدر أحلله 🙂"
-    result = sentiment_pipeline(text)
-    label = result[0]["label"]
-    score = result[0]["score"]
-    return f"التصنيف: {label}\nدرجة الثقة: {score:.2f}"
+# ============= إنشاء FastAPI app =============
 
-# ========== 2) FastAPI app ==========
 app = FastAPI()
 
-# توكن تيليجرام من Secrets
-BOT_TOKEN = os.getenv("TELEGRAM_TOKEN")
-if BOT_TOKEN:
-    TELEGRAM_API = f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage"
-else:
-    TELEGRAM_API = None
-    print("⚠️ TELEGRAM_TOKEN is not set. Telegram replies will fail.")
 
 @app.get("/")
-async def root():
-    return {"status": "ok", "message": "ArabicSentimentBoot is running."}
+def read_root():
+    return {"status": "ok", "message": "Telegram sentiment bot is running 🥰"}
+
 
 @app.post("/telegram")
-async def telegram_webhook(request: Request):
-    """
-    هذا المسار اللي تيليجرام يستدعيه كـ Webhook
-    """
-    data = await request.json()
-    print("📩 Incoming update:", data)
-
-    if "message" not in data or "text" not in data["message"]:
+async def telegram_webhook(update: Request):
+    data = await update.json()
+
+    # بعض أنواع التحديثات ما فيها "message" (مثل callback_query)
+    if "message" not in data:
         return {"ok": True}
 
-    chat_id = data["message"]["chat"]["id"]
-    user_text = data["message"]["text"]
+    message = data["message"]
+    chat = message.get("chat", {})
+    chat_id = chat.get("id")
 
-    reply_text = predict_sentiment(user_text)
+    text = message.get("text", "")
 
-    if TELEGRAM_API:
+    if not chat_id:
+        return {"ok": False, "error": "No chat_id found"}
+
+    if not text:
+        reply_text = "أرسل لي نص عربي عشان أحلل لك المشاعر 🌟"
+    else:
         try:
-            r = requests.post(
-                TELEGRAM_API,
-                json={"chat_id": chat_id, "text": reply_text},
-                timeout=10,
-            )
-            print("✅ Sent to Telegram:", r.status_code, r.text)
+            result = sentiment_pipeline(text)[0]
+            label = result["label"]
+            score = result["score"]
+            # لو اللابل عندك بالعربي من الكود السابق خليه زي ما هو
+            reply_text = f"التصنيف: {label}\nدرجة الثقة: {score:.4f}"
         except Exception as e:
-            print("❌ Error sending message to Telegram:", e)
+            reply_text = f"صار خطأ أثناء التحليل: {e}"
+
+    # إرسال الرد لتيلقرام
+    requests.post(
+        f"{TELEGRAM_API_URL}/sendMessage",
+        json={"chat_id": chat_id, "text": reply_text}
+    )
 
+    # لازم نرجع أي JSON عشان تيليجرام يعتبر الطلب ناجح
     return {"ok": True}