revised app.py

app.py

@@ -72,7 +72,10 @@ ensure_codeformer()
 # -------------------------------------------------
 # Pipeline Function
 # -------------------------------------------------
-def face_swap_and_enhance(src_img, tgt_img, fidelity=0.7, background_enhance=True, face_upsample=True):
+# -------------------------------------------------
+# Pipeline Function (defaults applied)
+# -------------------------------------------------
+def face_swap_and_enhance(src_img, tgt_img):
     try:
         src_bgr = cv2.cvtColor(src_img, cv2.COLOR_RGB2BGR)
         tgt_bgr = cv2.cvtColor(tgt_img, cv2.COLOR_RGB2BGR)
@@ -94,12 +97,7 @@ def face_swap_and_enhance(src_img, tgt_img, fidelity=0.7, background_enhance=Tru
         cv2.imwrite(swapped_path, swapped_bgr)
 
         # ---------------- Run CodeFormer ----------------
-        cmd = f"python {CODEFORMER_PATH} -w {fidelity:.2f} --input_path {swapped_path} --output_path {RESULT_DIR}"
-        if background_enhance:
-            cmd += " --bg_upsampler realesrgan"
-        if face_upsample:
-            cmd += " --face_upsample"
-
+        cmd = f"python {CODEFORMER_PATH} -w 0.7 --input_path {swapped_path} --output_path {RESULT_DIR} --bg_upsampler realesrgan --face_upsample"
         result = subprocess.run(cmd, shell=True, capture_output=True, text=True)
         if result.returncode != 0:
             return None, None, f"❌ CodeFormer failed:\n{result.stderr}"
@@ -123,28 +121,25 @@ def face_swap_and_enhance(src_img, tgt_img, fidelity=0.7, background_enhance=Tru
 
 
 # -------------------------------------------------
-# Gradio Interface
+# Gradio Interface (simplified)
 # -------------------------------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("## 🧑‍🤝‍🧑 Face Swap + CodeFormer Enhancement")
+    gr.Markdown("Face Swap")
+
     with gr.Row():
-        src_input = gr.Image(type="numpy", label="Upload Source Face")
+        src_input = gr.Image(type="numpy", label="Upload Your Face")
         tgt_input = gr.Image(type="numpy", label="Upload Target Image")
-    with gr.Row():
-        fidelity = gr.Slider(0, 1, value=0.7, step=0.01, label="CodeFormer Fidelity")
-        bg = gr.Checkbox(value=True, label="Enhance Background")
-        face_up = gr.Checkbox(value=True, label="Face Upsample")
-    btn = gr.Button("🚀 Run Face Swap + Enhance")
+
+    btn = gr.Button("Swap Face")
 
     output_img = gr.Image(type="numpy", label="Enhanced Output")
     download = gr.File(label="⬇️ Download Enhanced Image")
     error_box = gr.Textbox(label="Logs / Errors", interactive=False)
 
-    def process(src, tgt, f, b, fu):
-        img, path, err = face_swap_and_enhance(src, tgt, f, b, fu)
+    def process(src, tgt):
+        img, path, err = face_swap_and_enhance(src, tgt)
         return img, path, err
 
-    btn.click(process, [src_input, tgt_input, fidelity, bg, face_up],
-              [output_img, download, error_box])
+    btn.click(process, [src_input, tgt_input], [output_img, download, error_box])
 
-demo.launch()
+demo.launch()