Update video_processing.py

video_processing.py

@@ -182,9 +182,13 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
 
             progress(0.9, "Generating video with heatmap")
 
+            # Convert heatmaps to images
+            mse_heatmap_embeddings_img = fig_to_img(mse_heatmap_embeddings)
+            mse_heatmap_posture_img = fig_to_img(mse_heatmap_posture)
+
             # Create video with heatmap
             heatmap_video_path = os.path.join(output_folder, "video_with_heatmap.mp4")
-            create_video_with_heatmap(video_path, mse_heatmap_embeddings, mse_heatmap_posture, heatmap_video_path, desired_fps)
+            create_video_with_heatmap(video_path, mse_heatmap_embeddings_img, mse_heatmap_posture_img, heatmap_video_path, desired_fps)
 
             
         except Exception as e: