Update app.py

app.py

@@ -24,7 +24,7 @@ except (ImportError, AttributeError):
 
 
 
-def get_face_mask_box(img, feather):
+def get_face_mask_box(img, feather, padding=0):
     h, w = img.shape[:2]
     mask = np.zeros((h, w), dtype=np.uint8)
     results = face_mesh.process(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
@@ -34,11 +34,17 @@ def get_face_mask_box(img, feather):
     hull = cv2.convexHull(pts)
     cv2.fillConvexPoly(mask, hull, 255)
     x, y, bw, bh = cv2.boundingRect(hull)
-    mask_roi = mask[y:y+bh, x:x+bw]
+    # 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)
+    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)
-    return mask_roi, (x, y, bw, bh)
+    return mask_roi, (x_pad, y_pad, x2 - x_pad, y2 - y_pad)
 
 
 def cut_and_feather(img, feather):
@@ -160,14 +166,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):
+def process_video(video_path, ref_img, trans, res, step, feather, padding):
     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)
+    mask_ref, ref_box = get_face_mask_box(ref_bgr, feather, padding)
     if mask_ref is None:
         return None, None, None, None
     x_r, y_r, w_r, h_r = ref_box
@@ -190,7 +196,7 @@ def process_video(video_path, ref_img, trans, res, step, feather):
     for i in range(total):
         ret, frame = cap.read()
         if not ret: break
-        mask_roi, box = get_face_mask_box(frame, feather)
+        mask_roi, box = get_face_mask_box(frame, feather, padding)
         if mask_roi is None:
             out_vid.write(frame)
             continue
@@ -233,31 +239,31 @@ def process_video(video_path, ref_img, trans, res, step, feather):
 # --- Gradio App ---
 css = """video, img { object-fit: contain !important; }"""
 with gr.Blocks(css=css) as iface:
-    gr.Markdown("# Enhanced Face Morph with Composite Placement")
+    gr.Markdown("# Morph with Face-Shaped Composite and Padding")
     with gr.Row():
         vid = gr.Video(label='Input Video')
-        ref = gr.Image(type='numpy', label='Reference Face Image')
+        ref = gr.Image(type='numpy', label='Reference Image')
     with gr.Row():
-        res = gr.Dropdown([256, 384, 512, 768], value=512, label='Morph Resolution')
-        step = gr.Slider(1, 4, value=1, step=1, label='Landmark Sub-sampling')
-        feather = gr.Slider(0, 50, value=10, step=1, label='Feather Radius')
-    trans = gr.Slider(-1.0, 1.0, value=0.0, step=0.05, label='Transition Level')
+        res = gr.Dropdown([256,384,512,768], value=512, label='Resolution')
+        step = gr.Slider(1,4,value=1,step=1,label='Landmark Sub-sampling')
+        feather = gr.Slider(0,50,value=10,step=1,label='Feather Radius')
+        padding = gr.Slider(0,100,value=10,step=1,label='Crop Padding (px)')
+    trans = gr.Slider(-1.0,1.0,value=0.0,step=0.05,label='Transition Level')
     btn = gr.Button('Generate Morph 🚀')
     out_vid = gr.Video(label='Morphed Video')
-    out_central = gr.Image(label='Centralized Frame[0]')
-    out_ref = gr.Image(label='Cropped Reference')
-    out_morph0 = gr.Image(label='Morphed Frame[0]')
+    out_crop = gr.Image(label='First Frame Crop')
+    out_ref = gr.Image(label='Masked Reference')
+    out_morph = gr.Image(label='Masked Morphed First Frame')
 
     btn.click(
         fn=process_video,
-        inputs=[vid, ref, trans, res, step, feather],
-        outputs=[out_vid, out_central, out_ref, out_morph0],
+        inputs=[vid,ref,trans,res,step,feather,padding],
+        outputs=[out_vid,out_crop,out_ref,out_morph],
         show_progress=True
     )
-    gr.Markdown("---\n*Now the Frame[0] crop fits the face before resizing.*")
+    gr.Markdown("---\n*Added padding to the face crop for better framing.*")
 
-if __name__ == '__main__':
+if __name__=='__main__':
     iface.launch(debug=True)
 
 
-