Update app.py

app.py

@@ -3,43 +3,41 @@
 from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 import gradio as gr
 
-# موديل التحليل الفارسي
+# تحميل الموديل الفارسي
 model_name = "HooshvareLab/bert-fa-base-uncased-sentiment-snappfood"
 
-# تحميل الموديل والتوكنيزر
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
-# إنشاء الـ pipeline مع عرض كل النتائج
-sentiment_analyzer = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer, return_all_scores=True)
+sentiment_analyzer = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
 
-# دالة التحليل
 def analyze_sentiment(text):
-    results = sentiment_analyzer(text)[0]
-
-    # احتمال تكون التسميات LABEL_0 / LABEL_1 / LABEL_2
-    labels_map = {
-        "LABEL_0": "منفی (سلبي)",
-        "LABEL_1": "خنثی (محايد)",
-        "LABEL_2": "مثبت (إيجابي)"
-    }
-
-    output = ""
-    for r in results:
-        label = r["label"]
-        label_fa = labels_map.get(label, label)
-        score = round(r["score"], 3)
-        output += f"{label_fa}: {score}\n"
+    result = sentiment_analyzer(text)[0]
 
+    label = result["label"]
+    score = result["score"]
+
+    # نحسب الاحتمالات بشكل يدوي لو الموديل فيه فئتين فقط
+    if label.upper() == "HAPPY":
+        positive = round(score, 3)
+        negative = round(1 - score, 3)
+    else:
+        negative = round(score, 3)
+        positive = round(1 - score, 3)
+
+    neutral = round(1 - (positive + negative), 3)
+    if neutral < 0:
+        neutral = 0.0  # علشان ميطلعش سالب بالخطأ
+
+    output = f"مثبت (إيجابي): {positive}\nمنفی (سلبي): {negative}\nخنثی (محايد): {neutral}"
     return output
 
-# واجهة Gradio
 iface = gr.Interface(
     fn=analyze_sentiment,
     inputs=gr.Textbox(lines=2, placeholder="متن خود را وارد کنید..."),
     outputs="text",
-    title="تحلیل احساسات متون فارسی",
-    description="نمایش احتمال مثبت، منفی و خنثی بودن متن فارسی با مدل Snapfood."
+    title="تحلیل احساسات فارسی",
+    description="تحلیل متن به فارسی و نمایش احتمال مثبت، منفی و خنثی."
 )
 
 iface.launch()