Update app.py

app.py

@@ -1,4 +1,4 @@
-from transformers import pipeline
+import transformers
 import gradio as gr
 import matplotlib.pyplot as plt
 import io
@@ -12,10 +12,15 @@ def analyze_emotions(text):
     results = emotion_pipeline(text, top_k=None)  # Get probabilities for all emotions
     emotions = {res['label']: res['score'] for res in results}
 
+    # Extract emotion labels and probabilities
+    emotion_labels = list(emotions.keys())
+    emotion_probabilities = list(emotions.values())
+
     # Generate a pie chart for the emotions
-    plt.figure(figsize=(8, 4))
-    plt.pie(emotions.values(), labels=emotions.keys(), autopct='%1.1f%%')
-    plt.title("Emotion Analysis")
+    plt.figure(figsize=(8, 6))
+    plt.pie(emotion_probabilities, labels=emotion_labels, autopct='%1.1f%%', startangle=140)
+    plt.title("Emotion Distribution")
+    plt.axis('equal')
 
     # Save the plot to a bytes buffer
     buf = io.BytesIO()