ew

app.py

@@ -1,79 +1,50 @@
-import numpy as np
+import os
 import gradio as gr
-import roop.globals
-from roop.core import (
-    start,
-    decode_execution_providers,
-    suggest_max_memory,
-    suggest_execution_threads,
-)
+import numpy as np
+from roop.core import start, decode_execution_providers, suggest_max_memory, suggest_execution_threads
 from roop.processors.frame.core import get_frame_processors_modules
 from roop.utilities import normalize_output_path
-import os
-from PIL import Image
-
-processed_images = []
-
-def swap_face(source_file, target_files, doFaceEnhancer):
-    source_path = "input.jpg"
-    source_image = Image.fromarray(source_file)
-    source_image.save(source_path)
-
-    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)
-
-        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
-
-        start()
-        results.append(output_path)
-        processed_images.append(output_path)
 
-    return results
+html_section_1 = '<div><h1>Face Swap App</h1></div>'
 
-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>'
 
 app = gr.Blocks()
 
 with app:
     gr.HTML(html_section_1)
     gr.HTML(html_section_2)
-    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")
+    source_image = gr.Image(label="Source Image")
+    target_images = gr.File(label="Target Images", file_count="multiple")
+    do_face_enhancer = gr.Checkbox(label="Face Enhancer")
+    with gr.Tab("Result"):
+        result_images = gr.Gallery(label="Result Images")
+    with gr.Tab("History"):
+        history_images = gr.Gallery(label="Processed Images")
+
+    def face_swap(source_image, target_images, do_face_enhancer):
+        # Your face swap function here
+        # Assuming you have a function that takes source image, target images, and do_face_enhancer as input and returns a list of result images
+        result_images = []
+        for target_image in target_images:
+            # Call your face swap function for each target image
+            result_image = face_swap_function(source_image, target_image, do_face_enhancer)
+            result_images.append(result_image)
+        return result_images
+
+    gr.Button("Swap Faces").click(face_swap, inputs=[source_image, target_images, do_face_enhancer], outputs=result_images)
+
+    def save_history(result_images):
+        # Save the result images to a history folder
+        # You can use a database or a file system to store the images
+        # For simplicity, let's assume we're using a file system
+        history_folder = "history"
+        if not os.path.exists(history_folder):
+            os.makedirs(history_folder)
+        for i, result_image in enumerate(result_images):
+            filename = f"result_{i}.jpg"
+            result_image.save(os.path.join(history_folder, filename))
+        return result_images
+
+    gr.Button("Save to History").click(save_history, inputs=result_images, outputs=history_images)
 
 app.launch()