Update app.py

app.py

@@ -4,111 +4,54 @@ from fastapi import FastAPI, Request
 from transformers import pipeline
 import torch
 
-# ----------------------
-# 1) إعداد توكن البوت
-# ----------------------
-BOT_TOKEN = "8513655100:AAH5bgPDpXioJNW-o5IiNy6sqOVQvjvQXS0"  # يفضل بعدين تحطيه من Secrets
+BOT_TOKEN = "8513655100:AAH5bgPDpXioJNW-o5IiNy6sqOVQvjvQXS0"
 TELEGRAM_API_URL = f"https://api.telegram.org/bot{BOT_TOKEN}"
 
-# ----------------------
-# 2) تحميل المودل مرة وحدة
-# ----------------------
-print("Loading model...")
+MODEL_REPO = "maryaa4/my-arabic-sentiment-model"
 
 device = 0 if torch.cuda.is_available() else -1
+
 sentiment_pipeline = pipeline(
     "sentiment-analysis",
-    model="maryaa4/my-arabic-sentiment-model",
-    device=device,
-    trust_remote_code=True
+    model=MODEL_REPO,
+    trust_remote_code=True,
+    device=device
 )
 
-print(f"Model loaded. Device = {'cuda' if device == 0 else 'cpu'}")
+app = FastAPI()
 
-# ----------------------
-# 3) دوال التحليل والإرسال
-# ----------------------
-def analyze_text(text: str) -> str:
-    text = text.strip()
-    if not text:
-        return "رجاءً أرسل لي نص عربي عشان أحلل مشاعره 🌟"
+def analyze_text(text):
+    if not text.strip():
+        return "أرسل لي نص عربي عشان أحلل لك المشاعر 🌟"
 
     result = sentiment_pipeline(text)[0]
     label = result["label"]
     score = result["score"]
 
-    # نحاول نترجم الليبل للعربي + ايموجي لطيف
-    lower_label = label.lower()
-    if "pos" in lower_label:
-        mood = "إيجابي"
-        emoji = "😊✨"
-    elif "neg" in lower_label:
-        mood = "سلبي"
-        emoji = "😔"
-    else:
-        mood = "محايد"
-        emoji = "😐"
+    return f"التصنيف: {label}\nدرجة الثقة: {score:.3f}"
 
-    reply = (
-        f"🧠 *تحليل المشاعر:*\n"
-        f"- التصنيف: {mood} ({label})\n"
-        f"- نسبة الثقة: {score:.3f}\n\n"
-        f"{emoji}"
+def send_reply(chat_id, text):
+    requests.post(
+        f"{TELEGRAM_API_URL}/sendMessage",
+        json={"chat_id": chat_id, "text": text}
     )
 
-    # تيليجرام ما يحب الماركداون الغلط، فخليك حاليًا بدون parse_mode
-    return reply
-
-
-def send_message(chat_id: int, text: str):
-    try:
-        requests.post(
-            TELEGRAM_API_URL + "/sendMessage",
-            json={
-                "chat_id": chat_id,
-                "text": text
-            },
-            timeout=10
-        )
-    except Exception as e:
-        print("Error sending message:", e)
-
-
-# ----------------------
-# 4) تطبيق FastAPI + /telegram
-# ----------------------
-app = FastAPI()
-
 @app.post("/telegram")
 async def telegram_webhook(request: Request):
     update = await request.json()
-    print("Incoming update:", update)
+    print(update)
 
     if "message" not in update:
         return {"ok": True}
 
-    message = update["message"]
-    chat_id = message["chat"]["id"]
-    text = message.get("text", "")
-
-    if text == "/start":
-        send_message(
-            chat_id,
-            "هلا 🤍\n"
-            "أرسل لي أي جملة بالعربي، وأنا أقولك إذا إحساسها إيجابي، سلبي، أو محايد ✨"
-        )
-        return {"ok": True}
+    chat_id = update["message"]["chat"]["id"]
+    text = update["message"].get("text", "")
 
     reply = analyze_text(text)
-    send_message(chat_id, reply)
+    send_reply(chat_id, reply)
 
     return {"ok": True}
 
-
-# ----------------------
-# 5) تشغيل السيرفر داخل الـ Space
-# ----------------------
-if __name__ == "__main__":
-    import uvicorn
-    # مهم جداً: هذا اللي يخلي السيرفر يشتغل وما يطلع Runtime error
-    uvicorn.run("app:app", host="0.0.0.0", port=7860)
+@app.get("/")
+def home():
+    return {"status": "running", "message": "Telegram bot is alive"}