w

app.py

@@ -12,57 +12,51 @@ from roop.utilities import normalize_output_path
 import os
 from PIL import Image
 
+processed_images = []
 
-def swap_face(source_file, target_file, doFaceEnhancer):
-
+def swap_face(source_file, target_files, doFaceEnhancer):
     source_path = "input.jpg"
-    target_path = "target.jpg"
-
     source_image = Image.fromarray(source_file)
     source_image.save(source_path)
-    target_image = Image.fromarray(target_file)
-    target_image.save(target_path)
-
-    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
-    )
-    if doFaceEnhancer:
-        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
-    roop.globals.keep_frames = False
-    roop.globals.many_faces = False
-    roop.globals.video_encoder = "libx264"
-    roop.globals.video_quality = 18
-    roop.globals.max_memory = suggest_max_memory()
-    roop.globals.execution_providers = decode_execution_providers(["cuda"])
-    roop.globals.execution_threads = suggest_execution_threads()
+    results = []
+    for i, target_file in enumerate(target_files):
+        target_path = f"target_{i}.jpg"
+        target_image = Image.fromarray(target_file)
+        target_image.save(target_path)
 
-    print(
-        "start process",
-        roop.globals.source_path,
-        roop.globals.target_path,
-        roop.globals.output_path,
-    )
+        roop.globals.source_path = source_path
+        roop.globals.target_path = target_path
+        output_path = f"output_{i}.jpg"
+        roop.globals.output_path = normalize_output_path(
+            roop.globals.source_path, roop.globals.target_path, output_path
+        )
+        if doFaceEnhancer:
+            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
+        roop.globals.keep_frames = False
+        roop.globals.many_faces = False
+        roop.globals.video_encoder = "libx264"
+        roop.globals.video_quality = 18
+        roop.globals.max_memory = suggest_max_memory()
+        roop.globals.execution_providers = decode_execution_providers(["cuda"])
+        roop.globals.execution_threads = suggest_execution_threads()
 
-    for frame_processor in get_frame_processors_modules(
-        roop.globals.frame_processors
-    ):
-        if not frame_processor.pre_check():
-            return
+        for frame_processor in get_frame_processors_modules(
+            roop.globals.frame_processors
+        ):
+            if not frame_processor.pre_check():
+                return
 
-    start()
-    return output_path
+        start()
+        results.append(output_path)
+        processed_images.append(output_path)
 
+    return results
 
 html_section_1 = "<div><h1>Welcome to the Face Swap App</h1></div>"
 html_section_2 = '<div><p>Upload your source and target images to swap faces. Optionally, use the face enhancer feature for HD Results.</p><h2><br /><strong>For fast and bulk Swap:</strong>&nbsp;<a href="https://picfy.xyz/" target="_blank" rel="noopener">https://picfy.xyz/</a><br /> <strong>Support me USDT (TRC-20): TAe7hsSVWtMEYz3G5V1UiUdYPQVqm28bKx</h2></div>'
@@ -72,10 +66,14 @@ app = gr.Blocks()
 with app:
     gr.HTML(html_section_1)
     gr.HTML(html_section_2)
-    gr.Interface(
-        fn=swap_face,
-        inputs=[gr.Image(), gr.Image(), gr.Checkbox(label="face_enhancer?", info="do face enhancer?")],
-        outputs="image"
-    )
+    with gr.Tabs():
+        with gr.TabItem("Face Swap"):
+            gr.Interface(
+                fn=swap_face,
+                inputs=[gr.Image(), gr.Image(type="numpy", label="Target Images", multiple=True), gr.Checkbox(label="face_enhancer?", info="do face enhancer?")],
+                outputs=gr.Gallery(label="Processed Images")
+            )
+        with gr.TabItem("History"):
+            gr.Gallery(value=processed_images, label="Processed Images History")
 
-app.launch()
+app.launch()