Update app.py

app.py

@@ -162,7 +162,7 @@ def enhance_image_with_codeformer(rgb_img, temp_dir=None):
     python_cmd = sys.executable if sys.executable else "python3"
     cmd = (
         f"{python_cmd} {CODEFORMER_PATH} "
-        f"-w 0.7 "
+        f"-w 0.4 "
         f"--input_path {input_path} "
         f"--output_path {temp_dir} "
         f"--bg_upsampler realesrgan "
@@ -366,14 +366,16 @@ def face_swap_and_enhance(src_img, tgt_img, temp_dir=None):
                 return None, None, "❌ Face not detected in one of the images"
 
             swapped_path = os.path.join(temp_dir, f"swapped_{uuid.uuid4().hex[:8]}.jpg")
-            swapped_bgr = swapper.get(tgt_bgr, tgt_faces[0], src_faces[0])
+            swapped_rgb = multi_face_swap_industry(src_img, tgt_img)
+            swapped_bgr = cv2.cvtColor(swapped_rgb, cv2.COLOR_RGB2BGR)
+
             if swapped_bgr is None:
                 return None, None, "❌ Face swap failed"
 
             cv2.imwrite(swapped_path, swapped_bgr)
 
             python_cmd = sys.executable if sys.executable else "python3"
-            cmd = f"{python_cmd} {CODEFORMER_PATH} -w 0.7 --input_path {swapped_path} --output_path {temp_dir} --bg_upsampler realesrgan --face_upsample"
+            cmd = f"{python_cmd} {CODEFORMER_PATH} -w 0.4 --input_path {swapped_path} --output_path {temp_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}"
@@ -474,7 +476,7 @@ def build_multi_faceswap_gradio():
 
         def process(src_img, tgt_img):
             try:
-                swapped = multi_face_swap(src_img, tgt_img)
+                swapped = multi_face_swap_industry(src_img, tgt_img)
                 enhanced = enhance_image_with_codeformer(swapped)
                 return enhanced, ""
             except Exception as e: