Update app.py

app.py

@@ -1,62 +1,79 @@
 import os
 import requests
 from fastapi import FastAPI, Request
-from transformers import pipeline
 
-# 1) تحميل نفس موديلك من الريبو حقك
-MODEL_ID = "maryaa4/my-arabic-sentiment-model"
+# رابط الـ Inference API لموديلك
+HF_MODEL_URL = "https://api-inference.huggingface.co/models/maryaa4/my-arabic-sentiment-model"
 
-sentiment_pipe = pipeline(
-    "sentiment-analysis",
-    model=MODEL_ID,
-    trust_remote_code=True,
-)
-
-# 2) تطبيق FastAPI
-app = FastAPI()
-
-# 3) توكن البوت (بنحطه في Secrets باسم TELEGRAM_TOKEN)
+HF_TOKEN = os.getenv("HF_TOKEN")
 BOT_TOKEN = os.getenv("TELEGRAM_TOKEN")
+
 TELEGRAM_API = (
     f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage"
     if BOT_TOKEN
     else None
 )
 
-def build_reply(text: str) -> str:
-    result = sentiment_pipe(text)[0]
-    label = result["label"]
-    score = result["score"]
+app = FastAPI()
+
+def call_hf_model(text: str) -> str:
+    if not HF_TOKEN:
+        return "❌ مفقود HF_TOKEN في Secrets حق السبيس"
+
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    payload = {"inputs": text}
+
+    resp = requests.post(HF_MODEL_URL, headers=headers, json=payload)
 
-    # ترجمة بسيطة للّيبَل
-    if label.lower() == "positive":
+    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}]
+    try:
+        result = data[0]
+        label = result["label"]
+        score = result["score"]
+    except Exception:
+        return f"⚠️ Unexpected HF response: {data}"
+
+    label_lower = label.lower()
+    if "pos" in label_lower:
         label_ar = "إيجابي 🌿"
-    elif label.lower() == "negative":
+    elif "neg" in label_lower:
         label_ar = "سلبي 💔"
-    elif label.lower() == "neutral":
+    elif "neu" in label_lower:
         label_ar = "محايد 😐"
     else:
         label_ar = label
 
     return f"التصنيف: {label_ar}\nنسبة الثقة: {score:.2f}"
 
+@app.get("/")
+def root():
+    return {"status": "ok", "message": "Telegram bot webhook for Arabic sentiment 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 = build_reply(user_text)
+        reply_text = call_hf_model(user_text)
 
         if TELEGRAM_API:
             try:
-                requests.post(
+                r = requests.post(
                     TELEGRAM_API,
                     json={"chat_id": chat_id, "text": reply_text},
                 )
+                print("📤 Sent to Telegram, status:", r.status_code, r.text)
             except Exception as e:
                 print("Error sending message to Telegram:", e)
+        else:
+            print("❌ TELEGRAM_API is None (ما لقيت TELEGRAM_TOKEN في Secrets)")
 
     return {"ok": True}