Update video_processing.py

video_processing.py

@@ -197,21 +197,12 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             mse_heatmap_posture = plot_mse_heatmap(mse_posture, "Body Posture MSE Heatmap", df)
             mse_heatmap_voice = plot_mse_heatmap(mse_voice, "Voice MSE Heatmap", df)
 
+            stacked_heatmap = plot_stacked_mse_heatmaps(mse_embeddings, mse_posture, mse_voice, df, "Combined MSE Heatmaps")
+
             correlation_heatmap = plot_correlation_heatmap(mse_embeddings, mse_posture, mse_voice)
 
-            try:
-                if progress is not None:
-                    progress(0.95, desc="Generating video with heatmap")
-                
-                heatmap_video_path = create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_voice, output_folder, desired_fps, largest_cluster)
-                
-                if heatmap_video_path is None:
-                    print("Failed to create heatmap video")
-                else:
-                    print(f"Heatmap video path from create_video_with_heatmap: {heatmap_video_path}")
-                
                 if progress is not None:
-                    progress(1.0, "Video processing complete")
+                    progress(0.95, "Finishing generating graphs")
 
             except Exception as e:
                 print(f"Error in create_video_with_heatmap: {str(e)}")
@@ -223,7 +214,7 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             print(f"Error details: {str(e)}")
             import traceback
             traceback.print_exc()
-            return (f"Error in video processing: {str(e)}",) + (None,) * 26
+            return (f"Error in video processing: {str(e)}",) + (None,) * 27
 
         progress(1.0, "Preparing results")
         results = f"Number of persons detected: {num_clusters}\n\n"
@@ -290,7 +281,7 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             anomaly_frames_posture_images,
             aligned_faces_folder,
             frames_folder,
-            heatmap_video_path,
+            stacked_heatmap,
             correlation_heatmap
         )