Update video_processing.py

video_processing.py

@@ -214,17 +214,26 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
                 if progress is not None:
                     progress(0.95, desc="Generating video with heatmap")
                 
-                output_folder = "output"  # You can change this to your desired output folder
+                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 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
+                traceback.print_exc()
                 heatmap_video_path = None