Update app.py

app.py

@@ -24,7 +24,7 @@ except (ImportError, AttributeError):
 
 
 
-def get_face_mask_box(img, feather, padding=0):
+def get_face_mask_box(img, feather_pct, padding_pct):
     h, w = img.shape[:2]
     mask = np.zeros((h, w), dtype=np.uint8)
     results = face_mesh.process(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
@@ -34,16 +34,18 @@ def get_face_mask_box(img, feather, padding=0):
     hull = cv2.convexHull(pts)
     cv2.fillConvexPoly(mask, hull, 255)
     x, y, bw, bh = cv2.boundingRect(hull)
-    # apply padding
-    x_pad = max(x - padding, 0)
-    y_pad = max(y - padding, 0)
-    x2 = min(x + bw + padding, w)
-    y2 = min(y + bh + padding, h)
+    # calculate padding and feather in pixels
+    pad = int(max(bw, bh) * padding_pct)
+    x_pad = max(x - pad, 0)
+    y_pad = max(y - pad, 0)
+    x2 = min(x + bw + pad, w)
+    y2 = min(y + bh + pad, h)
     mask_roi = mask[y_pad:y2, x_pad:x2]
-    # inside feather
-    if feather > 0 and mask_roi.size > 0:
-        k = int(feather)
-        mask_roi = cv2.GaussianBlur(mask_roi, (k*2+1, k*2+1), 0)
+    # inside feather: kernel proportional to face size
+    if feather_pct > 0 and mask_roi.size > 0:
+        k = int(min(mask_roi.shape[0], mask_roi.shape[1]) * feather_pct)
+        if k % 2 == 0: k += 1
+        mask_roi = cv2.GaussianBlur(mask_roi, (k, k), 0)
     return mask_roi, (x_pad, y_pad, x2 - x_pad, y2 - y_pad)
 
 
@@ -166,14 +168,14 @@ def morph_faces(img1, img2, alpha, dim, step):
     return (out*255).astype(np.uint8)
 
 
-def process_video(video_path, ref_img, trans, res, step, feather, padding):
+def process_video(video_path, ref_img, trans, res, step, feather_pct, padding_pct):
     cap = cv2.VideoCapture(video_path)
     fps = cap.get(cv2.CAP_PROP_FPS) or 24
     total = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
 
     # Prepare masked reference
     ref_bgr = cv2.cvtColor(ref_img, cv2.COLOR_RGB2BGR)
-    mask_ref, ref_box = get_face_mask_box(ref_bgr, feather, padding)
+    mask_ref, ref_box = get_face_mask_box(ref_bgr, feather_pct, padding_pct)
     if mask_ref is None:
         return None, None, None, None
     x_r, y_r, w_r, h_r = ref_box
@@ -196,26 +198,21 @@ def process_video(video_path, ref_img, trans, res, step, feather, padding):
     for i in range(total):
         ret, frame = cap.read()
         if not ret: break
-        mask_roi, box = get_face_mask_box(frame, feather, padding)
+        mask_roi, box = get_face_mask_box(frame, feather_pct, padding_pct)
         if mask_roi is None:
             out_vid.write(frame)
             continue
         x, y, w, h = box
-        # Crop and resize original ROI
         crop = frame[y:y+h, x:x+w]
         crop_resized = cv2.resize(crop, (res, res))
-        # Morph
         alpha = float(np.clip((trans+1)/2, 0, 1))
         mor = morph_faces(crop_resized, ref_morph, alpha, res, step)
-        # Store first
         if i == 0:
             first_crop = crop_resized.copy()
             first_ref = ref_morph.copy()
             first_mask = cv2.resize(mask_roi, (res, res), interpolation=cv2.INTER_LINEAR)
             first_morphed = mor.copy()
-        # Resize morphed back
         mor_back = cv2.resize(mor, (w, h))
-        # Composite with shape mask
         mask_n = (mask_roi.astype(np.float32)[..., None] / 255.0)
         region = frame[y:y+h, x:x+w].astype(np.float32)
         blended = region * (1-mask_n) + mor_back.astype(np.float32) * mask_n
@@ -224,13 +221,12 @@ def process_video(video_path, ref_img, trans, res, step, feather, padding):
 
     cap.release(); out_vid.release()
 
-    # Apply mask to first_morphed for preview
     if first_morphed is not None and first_mask is not None:
         mask_n0 = first_mask.astype(np.float32)[..., None] / 255.0
         first_morphed = (first_morphed.astype(np.float32) * mask_n0).astype(np.uint8)
     else:
-        first_morphed = np.zeros((res, res, 3), dtype=np.uint8)
-        first_crop = first_crop if first_crop is not None else np.zeros((res, res,3),np.uint8)
+        first_morphed = np.zeros((res, res,3),dtype=np.uint8)
+        first_crop = first_crop if first_crop is not None else np.zeros((res, res,3),dtype=np.uint8)
         first_ref = first_ref if first_ref is not None else ref_morph.copy()
 
     # Convert for Gradio
@@ -239,15 +235,15 @@ def process_video(video_path, ref_img, trans, res, step, feather, padding):
 # --- Gradio App ---
 css = """video, img { object-fit: contain !important; }"""
 with gr.Blocks(css=css) as iface:
-    gr.Markdown("# Morph with Face-Shaped Composite and Padding")
+    gr.Markdown("# Morph with Face-Shaped Composite and Padding Percentage")
     with gr.Row():
         vid = gr.Video(label='Input Video')
         ref = gr.Image(type='numpy', label='Reference Image')
     with gr.Row():
         res = gr.Dropdown([256,384,512,768], value=512, label='Resolution')
         step = gr.Slider(1,4,value=4,step=1,label='Landmark Sub-sampling')
-        feather = gr.Slider(0,50,value=10,step=1,label='Feather Radius')
-        padding = gr.Slider(0,100,value=24,step=1,label='Crop Padding (px)')
+        feather = gr.Slider(0.0,0.5,value=0.1,step=0.01,label='Feather (%)')
+        padding = gr.Slider(0.0,0.5,value=0.24,step=0.01,label='Padding (%)')
     trans = gr.Slider(-1.0,1.0,value=-0.35,step=0.05,label='Transition Level')
     btn = gr.Button('Generate Morph 🚀')
     out_vid = gr.Video(label='Morphed Video')
@@ -261,7 +257,7 @@ with gr.Blocks(css=css) as iface:
         outputs=[out_vid,out_crop,out_ref,out_morph],
         show_progress=True
     )
-    gr.Markdown("---\n*Added padding to the face crop for better framing.*")
+    gr.Markdown("---\n*Default values set and feather/padding are now relative percentages.*")
 
 if __name__=='__main__':
     iface.launch(debug=True)