debug

app.py

@@ -36,46 +36,47 @@ def process_frame(frame):
     global overlay
     height, width = frame.shape[:2]
     face_detector.setInputSize((width, height))
+    
+    # Detect faces
     _, faces = face_detector.detect(frame)
-
     if faces is not None:
         for face in faces:
+            # Extract bounding box and facial landmarks
             x, y, w, h = face[:4].astype(int)
-            face_landmarks = face[4:14].reshape(5, 2).astype(int)  # Facial landmarks
+            face_landmarks = face[4:14].reshape(5, 2).astype(int)  # Adjust if needed
 
-            # Get the nose position
-            nose_x, nose_y = face_landmarks[2].astype(int)
-            # left eye postion
-            left_eye_x, left_eye_y = face_landmarks[0].astype(int)
-            # right eye postion
-            right_eye_x, right_eye_y = face_landmarks[1].astype(int)
+            # Get the nose and eye positions
+            nose_x, nose_y = face_landmarks[2]
+            left_eye_x, left_eye_y = face_landmarks[0]
+            right_eye_x, right_eye_y = face_landmarks[1]
 
-            # Calculate the midpoint between the eyes
+            # Calculate eye center and rotation angle
             eye_center_x = (left_eye_x + right_eye_x) // 2
             eye_center_y = (left_eye_y + right_eye_y) // 2
-
-            # Calculate the angle of rotation
             delta_x = right_eye_x - left_eye_x
             delta_y = right_eye_y - left_eye_y
             angle = np.degrees(np.arctan2(delta_y, delta_x))
 
-            # Resize the overlay
+            # Resize and rotate the overlay
             overlay_resize = cv2.resize(overlay, (int(w * 1.15), int(h * 0.8)))
-
-            # Rotate the overlay
             overlay_center = (overlay_resize.shape[1] // 2, overlay_resize.shape[0] // 2)
             rotation_matrix = cv2.getRotationMatrix2D(overlay_center, angle, 1.0)
-            overlay_rotated = cv2.warpAffine(overlay_resize, rotation_matrix, (overlay_resize.shape[1], overlay_resize.shape[0]), flags=cv2.INTER_LINEAR, borderMode=cv2.BORDER_CONSTANT, borderValue=(0, 0, 0, 0))
+            overlay_rotated = cv2.warpAffine(
+                overlay_resize, rotation_matrix, 
+                (overlay_resize.shape[1], overlay_resize.shape[0]), 
+                flags=cv2.INTER_LINEAR, borderMode=cv2.BORDER_CONSTANT, borderValue=(0, 0, 0, 0)
+            )
 
-            # Calculate the position to center the glasses on the eyes
-            overlay_x = eye_center_x - overlay_rotated.shape[1] // 2
-            overlay_y = eye_center_y - overlay_rotated.shape[0] // 2  # Adjust vertical position
+            # Calculate position for overlay
+            overlay_x = max(0, eye_center_x - overlay_rotated.shape[1] // 2)
+            overlay_y = max(0, eye_center_y - overlay_rotated.shape[0] // 2)
 
-            # Overlay the glasses
+            # Overlay the glasses using cvzone
             try:
                 frame = cvzone.overlayPNG(frame, overlay_rotated, [overlay_x, overlay_y])
             except Exception as e:
                 print(f"Error overlaying glasses: {e}")
+
     return frame