Update app.py

app.py

@@ -1,37 +1,60 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the sentiment analysis pipeline using your fine-tuned model
+# Ensure the model is correctly loaded
 model_name = "Abduuu/ArabReview-Sentiment"
 sentiment_pipeline = pipeline("text-classification", model=model_name, tokenizer=model_name)
 
 # Define label mapping for better readability
-label_mapping = {"LABEL_0": "سلبي 😞", "LABEL_1": "إيجابي 😊"}
+label_mapping = {
+    "LABEL_0": "سـلـبـي 😞",
+    "LABEL_1": "إيـجـابـي 😊"
+}
 
-# Define a function for sentiment prediction
+# Function to predict sentiment with confidence visualization
 def predict_sentiment(review):
     result = sentiment_pipeline(review)[0]
     sentiment_label = label_mapping[result["label"]]
-    confidence = f"{(100 * result['score']):.2f}%"
-    return f"التصنيف: {sentiment_label} | الثقة: {confidence}"
-
-# Define Gradio interface
-iface = gr.Interface(
-    fn=predict_sentiment,  # Function for sentiment prediction
-    inputs=gr.Textbox(label="أدخل مراجعتك", placeholder="اكتب مراجعتك هنا..."),
-    outputs=gr.Textbox(label="تحليل المشاعر", interactive=False),
-    title="🍽️ تحليل مشاعر مراجعات المطاعم العربية 🚀",
-    description="😞أدخل مراجعة لمطعم باللغة العربية وسيقوم النموذج بتحديد ما إذا كانت **إيجابية 😊** أو **سلبية 📌**.",
-    examples=[
-        ["الطعام لذيذ جدًا والخدمة رائعة!"],
-        ["التجربة كانت مريعة، الطعام كان سيئًا جدًا!"], 
-        ["الخدمة كانت سيئة"],  
-        ["لا بأس لكن يحتاج تحسين"],  
-        ["أفضل مطعم زرته في حياتي!"],  
-    ],
-    allow_flagging="never"  # Disables user flagging for simplicity
-)
+    confidence = round(result["score"] * 100, 2)
+    
+    return {
+        "التصنيف": sentiment_label,
+        "نسبة الثقة": f"{confidence}%",
+    }
+
+# Define Gradio interface with better UI
+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>"
+    )
+
+    with gr.Row():
+        review_input = gr.Textbox(
+            label="✍️ أدخل مراجعتك", 
+            placeholder="اكتب مراجعتك هنا...", 
+            lines=2
+        )
+
+    submit_button = gr.Button("🔍 تحليل المراجعة")
+
+    with gr.Row():
+        output_label = gr.Textbox(label="🔹 التصنيف", interactive=False)
+        output_confidence = gr.Textbox(label="📊 نسبة الثقة", interactive=False)
+
+    submit_button.click(predict_sentiment, inputs=review_input, outputs=[output_label, output_confidence])
+
+    gr.Examples(
+        examples=[
+            ["🍛 الطعام لذيذ جدًا والخدمة ممتازة!"],
+            ["🤢 للأسف، التجربة كانت سيئة والطعام غير نظيف!"],
+            ["👎 الخدمة كانت بطيئة والأسعار مرتفعة جدًا."],  
+            ["👌 تجربة رائعة، سأعود مجددًا!"],  
+            ["🔥 أفضل مطعم جربته على الإطلاق!"],  
+        ],
+        inputs=review_input,
+    )
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()