Update app.py

app.py

@@ -1,9 +1,7 @@
-# -*- coding: UTF-8 -*-
-# !/usr/bin/env python
 
-import os
-import shutil
-from PIL import Image
+# -* coding:UTF-8 -*
+# !/usr/bin/env python
+import numpy as np
 import gradio as gr
 import roop.globals
 from roop.core import (
@@ -14,29 +12,33 @@ from roop.core import (
 )
 from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
+import os
+from PIL import Image
+
+
+def swap_face(source_file, target_file,doFaceEnhancer):
 
-def swap_face(source_file, target_file, do_face_enhancer, mode):
-    # Save source image
     source_path = "input.jpg"
-    Image.open(source_file).save(source_path)
-    roop.globals.source_path = source_path
+    target_path = "target.jpg"
 
-    # Save target file based on mode
-    if mode == "Image to Image":
-        target_path = "target.jpg"
-        Image.open(target_file).save(target_path)
-        output_path = "output.jpg"
-    else:  # Image to Video
-        target_path = "target.mp4"
-        shutil.copyfile(target_file, target_path)  # ✅ FIXED for file path
-        output_path = "output.mp4"
+    source_image = Image.fromarray(source_file)
+    source_image.save(source_path)
+    target_image = Image.fromarray(target_file)
+    target_image.save(target_path)
 
-    # Set Roop globals
+    print("source_path: ", source_path)
+    print("target_path: ", target_path)
+
+    roop.globals.source_path = source_path
     roop.globals.target_path = target_path
+    output_path = "output.jpg"
     roop.globals.output_path = normalize_output_path(
         roop.globals.source_path, roop.globals.target_path, output_path
     )
-    roop.globals.frame_processors = ["face_swapper", "face_enhancer"] if do_face_enhancer else ["face_swapper"]
+    if doFaceEnhancer == True:
+        roop.globals.frame_processors = ["face_swapper","face_enhancer"]
+    else:
+        roop.globals.frame_processors = ["face_swapper"]
     roop.globals.headless = True
     roop.globals.keep_fps = True
     roop.globals.keep_audio = True
@@ -48,40 +50,24 @@ def swap_face(source_file, target_file, do_face_enhancer, mode):
     roop.globals.execution_providers = decode_execution_providers(["cuda"])
     roop.globals.execution_threads = suggest_execution_threads()
 
-    # Validate pre-checks
-    for frame_processor in get_frame_processors_modules(roop.globals.frame_processors):
+    print(
+        "start process",
+        roop.globals.source_path,
+        roop.globals.target_path,
+        roop.globals.output_path,
+    )
+
+    for frame_processor in get_frame_processors_modules(
+        roop.globals.frame_processors
+    ):
         if not frame_processor.pre_check():
-            return None
+            return
 
-    # Start face swap
     start()
+    return output_path
 
-    return roop.globals.output_path
-
-# Gradio Interface
-with gr.Blocks() as app:
-    gr.Markdown("# 👤 Face Swap App - Roop Based")
-
-    with gr.Row():
-        mode = gr.Radio(["Image to Image", "Image to Video"], label="Swap Mode", value="Image to Image")
-        do_face_enhancer = gr.Checkbox(label="Apply Face Enhancer", value=True)
-
-    with gr.Row():
-        source_file = gr.Image(label="Source Face Image", type="filepath")
-        target_file = gr.File(label="Target Image or Video")
-
-    output = gr.File(label="Swapped Output")
 
-    swap_button = gr.Button("🔁 Swap Face")
-
-    def run_swap(source_file, target_file, do_face_enhancer, mode):
-        result = swap_face(source_file, target_file, do_face_enhancer, mode)
-        return result
-
-    swap_button.click(
-        fn=run_swap,
-        inputs=[source_file, target_file, do_face_enhancer, mode],
-        outputs=output
-    )
-
-app.launch()
+app = gr.Interface(
+    fn=swap_face, inputs=[gr.Image(), gr.Image(),gr.Checkbox(label="face_enhancer?", info="do face enhancer?")], outputs="image"
+)
+app.launch()