ermm

app.py

@@ -176,23 +176,17 @@ def plot_average_emotion_barplot(heatmap_data):
     labels = [id2label[i] for i in range(len(mean_scores))]
     scores = mean_scores
 
-    # Sort for better visualization
-    sorted_data = sorted(
-        zip(labels_filtered, scores), key=lambda x: x[1], reverse=True
-    )
-    sorted_labels, sorted_scores = zip(*sorted_data)
-
-    colors = [emotion_colors[label] for label in sorted_labels]
+    colors = [emotion_colors[label] for label in labels]
 
     fig, ax = plt.subplots(figsize=(8, 6))
     bars = sns.barplot(
-        x=list(sorted_scores), y=list(sorted_labels), palette=colors, ax=ax
+        x=list(scores), y=list(labels), palette=colors, ax=ax
     )
 
     ax.xaxis.set_major_formatter(PercentFormatter(xmax=1.0, decimals=0))
 
     # Add percentage labels
-    for i, score in enumerate(sorted_scores):
+    for i, score in enumerate(scores):
         ax.text(score + 0.01, i, f"{score*100:.1f}%", va="center")
 
     ax.set_title("Average Emotion Confidence Across Sentences", fontsize=14)