Update app.py

app.py

@@ -47,8 +47,7 @@ def process_video(
     tmp_vid = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4').name
     out_vid = cv2.VideoWriter(tmp_vid, cv2.VideoWriter_fourcc(*'mp4v'), fps, (w_o, h_o))
 
-    first_crop = first_ref = first_mask = first_morphed = None
-
+    
     # --- Frame-by-frame processing ---
     for i in range(total):
         ret, frame = cap.read()
@@ -66,11 +65,7 @@ def process_video(
         alpha = float(np.clip((trans+1)/2, 0, 1))
         mor = ff.morph_faces(crop_resized, ref_morph, alpha, res, step)
 
-        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()
+        
 
         mor_back = cv2.resize(mor, (w, h))
         mask_n = (mask_roi.astype(np.float32)[..., None] / 255.0)
@@ -82,49 +77,35 @@ def process_video(
     cap.release()
     out_vid.release()
 
-    # --- First-frame outputs ---
-    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:
-        zero = np.zeros((res, res, 3), dtype=np.uint8)
-        first_crop = first_crop or zero
-        first_ref = first_ref or ref_morph
-        first_morphed = zero
+    
 
     progress(1.0, desc="Done")
-    return tmp_vid, \
-           cv2.cvtColor(first_crop, cv2.COLOR_BGR2RGB), \
-           cv2.cvtColor(first_ref, cv2.COLOR_BGR2RGB), \
-           cv2.cvtColor(first_morphed, cv2.COLOR_BGR2RGB)
-
+    return tmp_vid
+        
 # --- 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 Percentage")
+    gr.Markdown("# FaceFlux: Super Fast face swap")
     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.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')
+        res = gr.Dropdown([256,384,512,768], value=512, label='Face Resolution')
+        step = gr.Slider(1,4,value=1,step=1,label='Quality')
+        feather = gr.Slider(0.1,0.24,value=0.1,step=0.01,label='Feather (%)')
+        
+    trans = gr.Slider(-1.0,1.0,value=-0.35,step=0.05,label='Strength')
     btn = gr.Button('Generate Morph 🚀')
     out_vid = gr.Video(label='Morphed Video')
-    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, padding],
-        outputs=[out_vid, out_crop, out_ref, out_morph],
+        inputs=[vid, ref, trans, res, (5 - step), feather, 0.24],
+        outputs=[out_vid],
         show_progress=True
     )
 
-    gr.Markdown("---\n*Default values set and feather/padding are now relative percentages.*")
 
 # Enable queueing so progress updates render
 iface.queue().launch(debug=True)