Update app.py

app.py

@@ -35,68 +35,31 @@ async def process_img(file: UploadFile = File(...)):
 
         face = faces[0]
 
-        # =============================
-        # 1. Extract main landmarks
-        # =============================
-        pts = face.landmark_2d_106
-        
-        left_eye = pts[38]
-        right_eye = pts[88]
-        nose = pts[52]
-
-        # =============================
-        # 2. Compute roll angle
-        # =============================
-        dx = right_eye[0] - left_eye[0]
-        dy = right_eye[1] - left_eye[1]
-        roll_angle = np.degrees(np.arctan2(dy, dx))
-
-        # =============================
-        # 3. Rotate image to fix roll
-        # =============================
-        rotated = img.rotate(-roll_angle, expand=True)
-        np_rot = np.array(rotated)
-
-        # =============================
-        # 4. Detect again after rotation
-        # =============================
-        faces2 = face_app.get(np_rot)
-        if not faces2:
-            return JSONResponse({"error": "Face lost after rotation"}, status_code=500)
-
-        f = faces2[0]
-        x1, y1, x2, y2 = f.bbox.astype(int)
-
-        # =============================
-        # 5. Resize face to Abshir ratio
-        # Face height should be 70% of image
-        # =============================
-        face_h = y2 - y1
-        target_face_ratio = 0.70
-        target_h = int(face_h / target_face_ratio)
-
-        center_y = (y1 + y2) // 2
-        top = center_y - target_h // 2
-        bottom = center_y + target_h // 2
-
-        top = max(0, top)
-        bottom = min(rotated.height, bottom)
-
-        crop_img = rotated.crop((0, top, rotated.width, bottom))
-
-        # =============================
-        # 6. Apply white background
-        # =============================
-        white_bg = Image.new("RGB", crop_img.size, (255, 255, 255))
-        white_bg.paste(crop_img, (0, 0))
-
-        # =============================
-        # 7. Final export 4x6
-        # =============================
-        final_img = white_bg.resize((472, 709), Image.LANCZOS)
+        # --- Extract face bounding box ---
+        x1, y1, x2, y2 = face.bbox.astype(int)
+        cropped = img.crop((x1, y1, x2, y2))
 
+        # --- Resize cropped face with preserved aspect ratio ---
+        max_face_height = int(709 * 0.75)  # face occupies 75% of final height
+        w, h = cropped.size
+        scale_factor = max_face_height / h
+
+        new_w = int(w * scale_factor)
+        new_h = int(h * scale_factor)
+        resized_face = cropped.resize((new_w, new_h), Image.LANCZOS)
+
+        # --- Create final white 4x6 canvas ---
+        final_w, final_h = 472, 709
+        canvas = Image.new("RGB", (final_w, final_h), (255, 255, 255))
+
+        # --- Center image on canvas ---
+        paste_x = (final_w - new_w) // 2
+        paste_y = (final_h - new_h) // 2
+        canvas.paste(resized_face, (paste_x, paste_y))
+
+        # --- Output final image ---
         buf = io.BytesIO()
-        final_img.save(buf, format="JPEG")
+        canvas.save(buf, format="JPEG")
         buf.seek(0)
         return Response(buf.getvalue(), media_type="image/jpeg")