Upload 4 files

overlay.py

@@ -3,7 +3,7 @@ from PIL import Image
 import mediapipe as mp
 import os
 
-from segmentation import extract_hair_face_full_forehead, extract_hair
+from segmentation import get_face, extract_hair
 
 # MediaPipe Face Detection
 mp_fd = mp.solutions.face_detection.FaceDetection(model_selection=1,
@@ -182,7 +182,7 @@ def overlay_source(background: Image.Image, source: Image.Image):
     )
     
     # Step 2: Get face from background (with forehead) and overlay onto scaled source
-    bg_face = extract_hair_face_full_forehead(background)  # Extract face with forehead
+    bg_face = get_face(background)  # Extract face with forehead
     
     # Compute offset to align background face onto scaled source face
     offset = compute_offset_to_align_faces(bbox_src_scaled, bbox_bg, 1.0)
@@ -197,7 +197,7 @@ def overlay_source(background: Image.Image, source: Image.Image):
     print("✅ Overlaid background face onto scaled source")
     
     # Step 3: Extract hair+face+forehead from the result
-    hair_face_result = extract_hair_face_full_forehead(source_with_bg_face)
+    hair_face_result = get_face(source_with_bg_face)
     
     # Overlay this onto original background with smoothing -> temp variable
     bbox_result = get_face_bbox(source_with_bg_face)