Update module

app.py

@@ -61,6 +61,7 @@ import tempfile
 from ultralytics import YOLO
 from moviepy import ImageSequenceClip
 from scipy.io import wavfile
+from PIL import Image, ImageOps
 
 # Load face detector
 model = YOLO("yolov8n-face.pt").cuda()  # assumes CUDA available
@@ -165,7 +166,18 @@ def extract_faces(video_file):
             y2 = int(min(max(0, y2 + shift_down), h))
             face_crop = frame[y1:y2, x1:x2]
             if face_crop.size != 0:
-                resized = cv2.resize(face_crop, (224, 224))
+                #resized = cv2.resize(face_crop, (224, 224))
+                def pad_to_square(img):
+                    h, w = img.shape[:2]
+                    size = max(h, w)
+                    pad_top = (size - h) // 2
+                    pad_bottom = size - h - pad_top
+                    pad_left = (size - w) // 2
+                    pad_right = size - w - pad_left
+                    return cv2.copyMakeBorder(img, pad_top, pad_bottom, pad_left, pad_right, borderType=cv2.BORDER_CONSTANT, value=0)
+
+                square_crop = pad_to_square(face_crop)
+                resized = cv2.resize(square_crop, (224, 224))
                 frames.append(resized)
                 break  # only one face per frame