Update module

app.py

@@ -150,7 +150,7 @@ def extract_faces(video_file):
             # version 2
             h, w, _ = frame.shape
             x1, y1, x2, y2 = box.xyxy[0].cpu().numpy()
-            pad_ratio = 0.3  # 30% padding
+            pad_ratio = 0.5  # 30% padding
 
             dx = (x2 - x1) * pad_ratio
             dy = (y2 - y1) * pad_ratio
@@ -167,12 +167,19 @@ def extract_faces(video_file):
             if face_crop.size != 0:
                 #resized = cv2.resize(face_crop, (224, 224))
                 #frames.append(resized)
-                from PIL import Image, ImageOps
-
-                face_crop = Image.fromarray(face_crop)
-                resized = ImageOps.pad(face_crop, (224, 224), color=0)
-                resized = np.array(resized)
-                frames.append(resized)
+                h_crop, w_crop = face_crop.shape[:2]
+                cy, cx = h_crop // 2, w_crop // 2
+                half = 112
+
+                start_y = max(0, cy - half)
+                end_y = start_y + 224
+                start_x = max(0, cx - half)
+                end_x = start_x + 224
+
+                # Make sure we don't go out of bounds
+                if end_y <= h_crop and end_x <= w_crop:
+                    center_crop = face_crop[start_y:end_y, start_x:end_x]
+                    frames.append(center_crop)
                 break  # only one face per frame
 
     cap.release()