Update app.py

app.py

@@ -1,57 +1,75 @@
 import os
 import requests
 from fastapi import FastAPI, Request
-from transformers import pipeline
-import torch
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, TextClassificationPipeline
 
-BOT_TOKEN = "8513655100:AAH5bgPDpXioJNW-o5IiNy6sqOVQvjvQXS0"
-TELEGRAM_API_URL = f"https://api.telegram.org/bot{BOT_TOKEN}"
-
-MODEL_REPO = "maryaa4/my-arabic-sentiment-model"
+# توكن تيلقرام (قدري تحطينه هنا مباشرة)
+TELEGRAM_BOT_TOKEN = os.getenv(
+    "TELEGRAM_BOT_TOKEN",
+    "8513655100:AAH5bgPDpXioJNW-o5IiNy6sqOVQvjvQXS0"  # استبدليه لو غيرتي التوكن
+)
 
-device = 0 if torch.cuda.is_available() else -1
+# موديل المشاعر من هقنق فيس
+MODEL_ID = "maryaa4/my-arabic-sentiment-model"
 
-sentiment_pipeline = pipeline(
-    "sentiment-analysis",
-    model=MODEL_REPO,
-    trust_remote_code=True,
-    device=device
-)
+print("Loading model...")
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
+model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID)
+pipeline = TextClassificationPipeline(model=model, tokenizer=tokenizer, return_all_scores=False)
+print("Model loaded. Device set to CPU.")
 
 app = FastAPI()
 
-def analyze_text(text):
-    if not text.strip():
-        return "أرسل لي نص عربي عشان أحلل لك المشاعر 🌟"
 
-    result = sentiment_pipeline(text)[0]
+@app.get("/")
+async def root():
+    # عشان صفحة السبيس ما تعطي 404
+    return {"status": "ok", "message": "Telegram webhook is at /telegram"}
+
+
+def analyze_sentiment(text: str) -> str:
+    if not text:
+        return "أرسل لي جملة عربية عشان أحلل مشاعرها 😊"
+
+    result = pipeline(text)[0]
     label = result["label"]
     score = result["score"]
+    return f"التصنيف: {label} — الثقة: {score:.2f}"
 
-    return f"التصنيف: {label}\nدرجة الثقة: {score:.3f}"
-
-def send_reply(chat_id, text):
-    requests.post(
-        f"{TELEGRAM_API_URL}/sendMessage",
-        json={"chat_id": chat_id, "text": text}
-    )
 
 @app.post("/telegram")
 async def telegram_webhook(request: Request):
-    update = await request.json()
-    print(update)
+    """
+    هذا الإندبوينت يستقبل تحديثات تيلقرام كـ JSON
+    ويرد على نفس الشات باستعمال sendMessage
+    """
+    try:
+        update = await request.json()
+        # ناخذ المرسلة الأساسية (لو رسالة جديدة أو معدلة)
+        message = update.get("message") or update.get("edited_message") or {}
 
-    if "message" not in update:
-        return {"ok": True}
+        chat = message.get("chat") or {}
+        chat_id = chat.get("id")
 
-    chat_id = update["message"]["chat"]["id"]
-    text = update["message"].get("text", "")
+        # لو مافي شات آي دي، نطلع بس بدون ما نسوي شيء
+        if not chat_id:
+            return {"ok": True}
 
-    reply = analyze_text(text)
-    send_reply(chat_id, reply)
+        text = message.get("text") or ""
 
-    return {"ok": True}
+        reply_text = analyze_sentiment(text)
 
-@app.get("/")
-def home():
-    return {"status": "running", "message": "Telegram bot is alive"}
+        # نرسل الرد لتيلقرام
+        url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendMessage"
+        payload = {
+            "chat_id": chat_id,
+            "text": reply_text,
+        }
+        requests.post(url, json=payload)
+
+    except Exception as e:
+        # تطبعين الخطأ في اللوق لو حبيتي تشوفين وش صار
+        print("Error in /telegram webhook:", e)
+
+    # لازم نرجع 200 لتيلقرام
+    return {"ok": True}