Update app.py

app.py

@@ -1,104 +1,75 @@
 import os
 import requests
+import gradio as gr
 from fastapi import FastAPI, Request
+from transformers import pipeline
 
-# ========= 1) إعدادات Hugging Face Model =========
+# ============ 1) تحميل الموديل من هجينغ فيس ============
+deployed_repo_id = "maryaa4/my-arabic-sentiment-model"
 
-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
-    else None
+loaded_sentiment_pipeline = pipeline(
+    "sentiment-analysis",
+    model=deployed_repo_id,
+    trust_remote_code=True,
 )
 
-app = FastAPI()
-
-
-def analyze_sentiment(text: str) -> str:
-    """
-    ينادي Hugging Face Inference API ويحلل المشاعر
-    """
-    if not HF_TOKEN:
-        return "❌ HF_TOKEN مفقود في Secrets حق السبيس"
-
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
-    payload = {"inputs": text}
+def predict_sentiment(text: str) -> str:
+    if not text or text.strip() == "":
+        return "رجاءً أدخل نص عربي عشان أقدر أحلله 📝"
 
-    resp = requests.post(HF_API_URL, headers=headers, json=payload)
+    result = loaded_sentiment_pipeline(text)
+    label = result[0]["label"]
+    score = result[0]["score"]
 
-    # لو رجع خطأ من API
-    if resp.status_code != 200:
-        return f"❌ Error from HF API: {resp.status_code} - {resp.text}"
+    return f"التصنيف: {label}\nنسبة الثقة: {score:.2f}"
 
-    data = resp.json()
-
-    # متوقع يرجع زي: [[{"label": "POSITIVE", "score": 0.95}]]
-    try:
-        # بعض الموديلات ترجع 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 content: {data}"
-
-    label_lower = label.lower()
-    if "pos" in label_lower:
-        label_ar = "إيجابي 🌿"
-    elif "neg" in label_lower:
-        label_ar = "سلبي 💔"
-    elif "neu" in label_lower:
-        label_ar = "محايد 😐"
-    else:
-        label_ar = label
-
-    return f"التصنيف: {label_ar}\nنسبة الثقة: {score:.2f}"
+# ============ 2) واجهة Gradio (عشان السبيس يشتغل طبيعي) ============
+demo = gr.Interface(
+    fn=predict_sentiment,
+    inputs=gr.Textbox(lines=5, placeholder="أدخل النص العربي هنا..."),
+    outputs="text",
+    title="Arabic Sentiment Analysis",
+    description="تحليل مشاعر للنصوص العربية باستخدام الموديل maryaa4/my-arabic-sentiment-model",
+)
 
+# ============ 3) FastAPI + Telegram Webhook ============
 
-@app.get("/")
-def root():
-    return {
-        "status": "ok",
-        "message": "Telegram webhook for Arabic sentiment bot is running ✅",
-    }
+app = FastAPI()
 
+BOT_TOKEN = os.getenv("TELEGRAM_TOKEN")  # حطي التوكن في Secrets باسم TELEGRAM_TOKEN
+if BOT_TOKEN:
+    TELEGRAM_API = f"https://api.telegram.org/bot{BOT_TOKEN}/sendMessage"
+else:
+    TELEGRAM_API = None
+    print("⚠️ TELEGRAM_TOKEN مش موجود في Secrets، البوت ما راح يقدر يرسل ردود لتيليجرام.")
 
-@app.post("/")
+@app.post("/telegram")
 async def telegram_webhook(request: Request):
     """
-    هذا هو الويبهوك اللي تيليجرام يناديه
+    هذا المسار اللي تيليجرام بيناديه لما يجي رسالة للبوت.
     """
     data = await request.json()
-    print("📩 Incoming update:", data)
+    print("📩 Webhook data:", data)
 
-    # نتأكد إن فيه رسالة نصية
     if "message" in data and "text" in data["message"]:
         chat_id = data["message"]["chat"]["id"]
         user_text = data["message"]["text"]
 
-        reply_text = analyze_sentiment(user_text)
+        reply = predict_sentiment(user_text)
 
         if TELEGRAM_API:
             try:
                 r = requests.post(
                     TELEGRAM_API,
-                    json={"chat_id": chat_id, "text": reply_text},
+                    json={"chat_id": chat_id, "text": reply},
+                    timeout=10
                 )
-                print("📤 Sent to Telegram, status:", r.status_code, r.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_TOKEN مفقود في Secrets (TELEGRAM_API = None)")
+                print("❌ Error sending to Telegram:", e)
+
+    return {"ok": True}
 
-    return {"ok": True}
+# ============ 4) ربط Gradio مع FastAPI ============
+# يخلي واجهة Gradio على "/" والويبهوك على "/telegram"
+app = gr.mount_gradio_app(app, demo, path="/")