Update video_processing.py

video_processing.py

@@ -199,13 +199,22 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             mse_heatmap_embeddings = plot_mse_heatmap(mse_embeddings, "Facial Features MSE Heatmap", df)
             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)
-
-            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))
+            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,
+                    heatmap_video_path, original_fps, largest_cluster,
+                    progress=progress
+                )
+                
+                if progress is not None:
+                    progress(1.0, desc="Video processing complete")
+            except Exception as e:
+                print(f"Error in create_video_with_heatmap: {str(e)}")
+                heatmap_video_path = None
 
             
         except Exception as e: