Update video_processing.py

video_processing.py

@@ -200,11 +200,13 @@ 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)
 
-            progress(0.95, "Generating video with heatmap")
+            if progress:
+                progress(0.95, "Generating video with heatmap")
+        
+            heatmap_video_path = create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_voice, 
+                                                           heatmap_video_path, original_fps, largest_cluster, 
+                                                           progress=lambda p, desc: progress(0.95 + 0.05 * p, desc))
 
-            # Create video with heatmap
-            heatmap_video_path = os.path.join(output_folder, "heatmap_video.mp4")
-            heatmap_video_path = create_video_with_heatmap(video_path, df, mse_embeddings, mse_posture, mse_voice, heatmap_video_path, original_fps, largest_cluster)
             
         except Exception as e:
             print(f"Error details: {str(e)}")