Update app.py

app.py

@@ -92,8 +92,8 @@ def extract_and_align_faces_from_video(video_path, aligned_faces_folder, desired
     for frame_num in range(0, frame_count, int(original_fps / desired_fps)):
         video.set(cv2.CAP_PROP_POS_FRAMES, frame_num)
         ret, frame = video.read()
-        if not ret or frame is None:
-            print(f"Error: Could not read frame {frame_num}")
+        if not ret or frame is None or frame.size == 0:
+            print(f"Skipping frame {frame_num}: Frame is empty or couldn't be read")
             continue
         try:
             boxes, probs = mtcnn.detect(frame)
@@ -102,6 +102,9 @@ def extract_and_align_faces_from_video(video_path, aligned_faces_folder, desired
                 if probs[0] >= 0.99:
                     x1, y1, x2, y2 = [int(b) for b in box]
                     face = frame[y1:y2, x1:x2]
+                    if face.size == 0:
+                        print(f"Skipping frame {frame_num}: Detected face region is empty")
+                        continue
                     aligned_face = alignFace(face)
                     if aligned_face is not None:
                         aligned_face_resized = cv2.resize(aligned_face, (160, 160))