Update video_processing.py

video_processing.py

@@ -224,6 +224,11 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
                     frame_img_with_timecode = add_timecode_to_image(frame_img, timecode)
                     anomaly_frames_posture_images.append(frame_img_with_timecode)
 
+        # Create annotated video
+        annotated_video_path = os.path.join(output_folder, "annotated_video.mp4")
+        create_annotated_video(video_path, df, mse_embeddings, largest_cluster, annotated_video_path)
+
+        
         return (
             execution_time,
             results,
@@ -241,7 +246,8 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
             anomaly_faces_embeddings,
             anomaly_frames_posture_images,
             aligned_faces_folder,
-            frames_folder
+            frames_folder,
+            annotated_video_path
         )
 
 def is_frontal_face(landmarks, threshold=50):