Update video_processing.py

video_processing.py

@@ -215,21 +215,19 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
                     progress(0.95, desc="Generating video with heatmap")
                 
                 output_folder = "output"
-                os.makedirs(output_folder, exist_ok=True)
                 heatmap_video_path = create_video_with_heatmap(
                     video_path, df, mse_embeddings, mse_posture, mse_voice,
                     output_folder, original_fps, largest_cluster,
                     progress=progress
                 )
-                print(f"Heatmap video path from create_video_with_heatmap: {heatmap_video_path}")
-
-                if not os.path.exists(heatmap_video_path):
-                    raise FileNotFoundError(f"Heatmap video file not found at {heatmap_video_path}")
-
-        
+                
+                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, desc="Video processing complete")
-                    
             except Exception as e:
                 print(f"Error in create_video_with_heatmap: {str(e)}")
                 import traceback