debug

app.py

@@ -34,52 +34,57 @@ def change_glasses():
 # Process frame for overlay
 def process_frame(frame):
     global overlay
+    # Ensure the frame is writable
+    frame = np.array(frame, copy=True)
+
     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)  # Adjust if needed
+            face_landmarks = face[4:14].reshape(5, 2).astype(int)  # Facial landmarks
 
-            # 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]
+            # Get the nose position
+            nose_x, nose_y = face_landmarks[2].astype(int)
+            # Left and right eye positions
+            left_eye_x, left_eye_y = face_landmarks[0].astype(int)
+            right_eye_x, right_eye_y = face_landmarks[1].astype(int)
 
-            # Calculate eye center and rotation angle
+            # Calculate the midpoint between the eyes
             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 and rotate the overlay
+            # Resize 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]), 
+                (overlay_resize.shape[1], overlay_resize.shape[0]),
                 flags=cv2.INTER_LINEAR, borderMode=cv2.BORDER_CONSTANT, borderValue=(0, 0, 0, 0)
             )
 
-            # 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)
+            # 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
 
-            # Overlay the glasses using cvzone
+            # Overlay the glasses
             try:
                 frame = cvzone.overlayPNG(frame, overlay_rotated, [overlay_x, overlay_y])
             except Exception as e:
                 print(f"Error overlaying glasses: {e}")
-
+                
     return frame
 
-
 # Gradio webcam input
 def webcam_input(frame):
     frame = process_frame(frame)