Update app.py

app.py

@@ -57,9 +57,22 @@ def extract_aspects(text, model):
 
 def plot_sentiment_distribution(df):
     fig, ax = plt.subplots()
-    df['label'].value_counts().loc[list(RATING_MAP.values())].plot.pie(
+    
+    # Get counts for all possible ratings
+    counts = df['label'].value_counts()
+    
+    # Ensure all rating categories are present (even with 0 counts)
+    for rating in RATING_MAP.values():
+        if rating not in counts.index:
+            counts[rating] = 0
+    
+    # Sort by the predefined rating order
+    counts = counts.loc[list(RATING_MAP.values())]
+    
+    # Plot with consistent colors
+    counts.plot.pie(
         autopct='%1.1f%%',
-        colors=['#ff9999','#66b3ff','#99ff99'],
+        colors=['#ff9999','#66b3ff','#99ff99'],  # Negative, Neutral, Positive
         ax=ax
     )
     ax.set_ylabel('')