Update app.py

app.py

@@ -1,11 +1,11 @@
 import gradio as gr
 from transformers import pipeline
 
-# Ensure the model is correctly loaded
+# Load the sentiment analysis model
 model_name = "Abduuu/ArabReview-Sentiment"
 sentiment_pipeline = pipeline("text-classification", model=model_name, tokenizer=model_name)
 
-# Define label mapping (ensure it matches actual model outputs)
+# Define label mapping (handling both possible label formats)
 label_mapping = {
     "LABEL_0": "سلبي",
     "LABEL_1": "إيجابي",
@@ -13,30 +13,21 @@ label_mapping = {
     "Positive": "إيجابي"
 }
 
-# Function to predict sentiment with confidence visualization
+# Function to predict sentiment
 def predict_sentiment(review):
     result = sentiment_pipeline(review)[0]
-    sentiment_label = label_mapping.get(result["label"], result["label"])  # Handle unexpected labels
-    confidence = round(result["score"] * 100, 2)
+    sentiment_label = label_mapping.get(result["label"], "غير معروف")  # Handle unexpected labels
+    confidence = f"{(result['score'] * 100):.2f}%"
     
-    return {
-        "التصنيف": sentiment_label,
-        "نسبة الثقة": f"{confidence}%",
-    }
+    return sentiment_label, confidence  # Return as separate values
 
-# Define Gradio interface with better UI
+# Define Gradio interface
 with gr.Blocks(theme=gr.themes.Default()) as iface:
     gr.Markdown("<h1 style='text-align: center;'>🍽️ تحليل مشاعر مراجعات المطاعم 🚀</h1>")
-    gr.Markdown(
-        "<p style='text-align: center;'> 🤖 أدخل مراجعة مطعم بالعربية، وسيقوم النموذج بتحليل المشاعر وتحديدها **إيجابية** أو **سلبية**.</p>"
-    )
+    gr.Markdown("<p style='text-align: center;'> 🤖 أدخل مراجعة مطعم بالعربية، وسيقوم النموذج بتحليل المشاعر.</p>")
 
     with gr.Row():
-        review_input = gr.Textbox(
-            label="✍️ أدخل مراجعتك", 
-            placeholder="اكتب مراجعتك هنا...", 
-            lines=2
-        )
+        review_input = gr.Textbox(label="✍️ أدخل مراجعتك", placeholder="اكتب مراجعتك هنا...", lines=2)
 
     submit_button = gr.Button("🔍 تحليل المراجعة")
 
@@ -59,4 +50,4 @@ with gr.Blocks(theme=gr.themes.Default()) as iface:
 
 # Launch the app with public sharing enabled
 if __name__ == "__main__":
-    iface.launch(share=True)
+    iface.launch(share=True)