Update video_processing.py

video_processing.py

@@ -120,10 +120,11 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
 
         progress(1, "Frame extraction complete")
         progress(0.3, "Processing frames")
-        embeddings_by_frame, posture_scores_by_frame, posture_landmarks_by_frame, aligned_face_paths = process_frames(
+        embeddings_by_frame, posture_scores_by_frame, posture_landmarks_by_frame, aligned_face_paths, facial_landmarks_by_frame = process_frames(
             frames_folder, aligned_faces_folder,
             frame_count,
             progress)
+    
 
         if not aligned_face_paths:
             raise ValueError("No faces were extracted from the video.")
@@ -209,6 +210,8 @@ def process_video(video_path, anomaly_threshold, desired_fps, progress=None):
                 face_img = cv2.imread(face_path)
                 if face_img is not None:
                     face_img = cv2.cvtColor(face_img, cv2.COLOR_BGR2RGB)
+                    if frame in facial_landmarks_by_frame:
+                        face_img = draw_facial_landmarks(face_img, facial_landmarks_by_frame[frame])
                     timecode = df[df['Frame'] == frame]['Timecode'].iloc[0]
                     face_img_with_timecode = add_timecode_to_image(face_img, timecode)
                     anomaly_faces_embeddings.append(face_img_with_timecode)