transpose heatmap

app.py

@@ -89,15 +89,29 @@ def prepare_heatmap_data(data):
     heatmap_data.columns = [item["sentence"][:18]+"..." for item in data]
     return heatmap_data
 
-def plot_emotion_heatmap(heatmap_data):
-    fig = plt.figure(figsize=(len(heatmap_data.columns) * 0.5 + 4, 5))
-    sns.heatmap(heatmap_data, annot=False, cmap="viridis", cbar=True, linewidths=0.5, linecolor='gray')
-    plt.xlabel("Sentences")
-    plt.ylabel("Emotions")
-    plt.xticks(rotation=45)
+
+def plot_emotion_heatmap(heatmap_data, cmap="viridis"):
+    heatmap_data = heatmap_data.T
+
+    fig = plt.figure(figsize=(len(heatmap_data.columns) * 0.5 + 4, len(heatmap_data.index) * 0.5 + 2))
+
+    sns.heatmap(
+        heatmap_data,
+        annot=False,
+        cmap=cmap,
+        cbar=True,
+        linewidths=0.5,
+        linecolor='gray',
+        vmin=0,
+        vmax=1
+    )
+
+    plt.xlabel("Emotions")
+    plt.ylabel("Sentences")
+    plt.xticks(rotation=0, ha='right')
     plt.yticks(rotation=0)
-    plt.subplots_adjust(left=0.2, right=0.95, top=0.9, bottom=0.2)
     plt.tight_layout()
+
     return fig
 
 def plot_sunburst_chart(heatmap_data):