Update app.py

app.py

@@ -2,12 +2,14 @@ import os
 import requests
 from fastapi import FastAPI, Request
 
-# رابط الـ Inference API لموديلك
-HF_MODEL_URL = "https://api-inference.huggingface.co/models/maryaa4/my-arabic-sentiment-model"
+# ========= 1) إعدادات Hugging Face Model =========
 
-HF_TOKEN = os.getenv("HF_TOKEN")
-BOT_TOKEN = os.getenv("TELEGRAM_TOKEN")
+HF_API_URL = "https://api-inference.huggingface.co/models/maryaa4/my-arabic-sentiment-model"
+HF_TOKEN = os.getenv("HF_TOKEN")  # نجيبه من Secrets
 
+# ========= 2) إعدادات Telegram Bot =========
+
+BOT_TOKEN = os.getenv("TELEGRAM_TOKEN")  # برضه من Secrets
 TELEGRAM_API = (
     f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage"
     if BOT_TOKEN
@@ -16,26 +18,40 @@ TELEGRAM_API = (
 
 app = FastAPI()
 
-def call_hf_model(text: str) -> str:
+
+def analyze_sentiment(text: str) -> str:
+    """
+    ينادي Hugging Face Inference API ويحلل المشاعر
+    """
     if not HF_TOKEN:
-        return "❌ مفقود HF_TOKEN في Secrets حق السبيس"
+        return "❌ HF_TOKEN مفقود في Secrets حق السبيس"
 
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
     payload = {"inputs": text}
 
-    resp = requests.post(HF_MODEL_URL, headers=headers, json=payload)
+    resp = requests.post(HF_API_URL, headers=headers, json=payload)
 
+    # لو رجع خطأ من API
     if resp.status_code != 200:
         return f"❌ Error from HF API: {resp.status_code} - {resp.text}"
 
     data = resp.json()
-    # متوقع يرجع list فيها dict زي: [{"label": "...", "score": 0.9}]
+
+    # متوقع يرجع زي: [[{"label": "POSITIVE", "score": 0.95}]]
     try:
-        result = data[0]
+        # بعض الموديلات ترجع list جوّا list، نجرب الحالتين
+        if isinstance(data, list) and len(data) > 0:
+            if isinstance(data[0], list):
+                result = data[0][0]
+            else:
+                result = data[0]
+        else:
+            return f"⚠️ Unexpected HF response structure: {data}"
+
         label = result["label"]
         score = result["score"]
     except Exception:
-        return f"⚠️ Unexpected HF response: {data}"
+        return f"⚠️ Unexpected HF response content: {data}"
 
     label_lower = label.lower()
     if "pos" in label_lower:
@@ -49,20 +65,29 @@ def call_hf_model(text: str) -> str:
 
     return f"التصنيف: {label_ar}\nنسبة الثقة: {score:.2f}"
 
+
 @app.get("/")
 def root():
-    return {"status": "ok", "message": "Telegram bot webhook for Arabic sentiment is running."}
+    return {
+        "status": "ok",
+        "message": "Telegram webhook for Arabic sentiment bot is running ✅",
+    }
+
 
 @app.post("/")
 async def telegram_webhook(request: Request):
+    """
+    هذا هو الويبهوك اللي تيليجرام يناديه
+    """
     data = await request.json()
     print("📩 Incoming update:", data)
 
+    # نتأكد إن فيه رسالة نصية
     if "message" in data and "text" in data["message"]:
         chat_id = data["message"]["chat"]["id"]
         user_text = data["message"]["text"]
 
-        reply_text = call_hf_model(user_text)
+        reply_text = analyze_sentiment(user_text)
 
         if TELEGRAM_API:
             try:
@@ -74,6 +99,6 @@ async def telegram_webhook(request: Request):
             except Exception as e:
                 print("Error sending message to Telegram:", e)
         else:
-            print("❌ TELEGRAM_API is None (ما لقيت TELEGRAM_TOKEN في Secrets)")
+            print("❌ TELEGRAM_TOKEN مفقود في Secrets (TELEGRAM_API = None)")
 
     return {"ok": True}