Update visualization.py

visualization.py

@@ -216,7 +216,6 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
     plt.close()
     return fig
 
-
 def create_heatmap(t, mse_embeddings, mse_posture, mse_voice, video_fps, total_frames, video_width):
     frame_count = int(t * video_fps)
     
@@ -230,14 +229,16 @@ def create_heatmap(t, mse_embeddings, mse_posture, mse_voice, video_fps, total_f
     combined_mse[1] = mse_posture_norm
     combined_mse[2] = mse_voice_norm
 
-    fig, ax = plt.subplots(figsize=(video_width / 50, 0.5))
+    fig, ax = plt.subplots(figsize=(video_width / 25, 0.2))  # Much thinner height, wider width
     ax.imshow(combined_mse, aspect='auto', cmap='hot', vmin=0, vmax=1, extent=[0, total_frames, 0, 3])
     ax.set_yticks([0.5, 1.5, 2.5])
-    ax.set_yticklabels(['Voice', 'Posture', 'Face'])
+    ax.set_yticklabels(['Voice', 'Posture', 'Face'], fontsize=6)  # Smaller font size
     ax.set_xticks([])
 
-    ax.axvline(x=frame_count, color='black', linewidth=3)
+    ax.axvline(x=frame_count, color='black', linewidth=2)  # Thinner line for smaller heatmap
 
+    plt.tight_layout(pad=0.1)  # Reduce padding around the plot
+    
     canvas = FigureCanvas(fig)
     canvas.draw()
     heatmap_img = np.frombuffer(canvas.tostring_rgb(), dtype='uint8')