Update visualization.py

visualization.py

@@ -209,6 +209,7 @@ def plot_posture(df, posture_scores, color='blue', anomaly_threshold=3):
 
 
 def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_voice, output_path, desired_fps, largest_cluster, progress=None):
+    print(f"Creating heatmap video. Output path: {output_path}")
     # Filter the DataFrame to only include frames from the largest cluster
     df_largest_cluster = df[df['Cluster'] == largest_cluster]
     
@@ -230,6 +231,7 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
     
     fourcc = cv2.VideoWriter_fourcc(*'mp4v')
     out = cv2.VideoWriter(output_path, fourcc, original_fps, (width, height + 200))
+    print(f"VideoWriter initialized. FPS: {original_fps}, Size: {(width, height + 200)}")
     
     # Ensure all MSE arrays have the same length as total_frames
     mse_embeddings = np.interp(np.linspace(0, len(mse_embeddings) - 1, total_frames), 
@@ -301,5 +303,5 @@ def create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_v
     cap.release()
     out.release()
     plt.close(fig)
-    
+    print(f"Heatmap video created at: {output_path}")
     return output_path