|
import numpy as np |
|
import gradio as gr |
|
import roop.globals |
|
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 |
|
|
|
|
|
def swap_face(source_file, target_file, 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() |
|
|
|
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 |
|
|
|
start() |
|
return output_path |
|
|
|
|
|
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.</p></div>" |
|
html_section_3 = "<div><p>After processing, the swapped image will be displayed below.</p></div>" |
|
|
|
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" |
|
) |
|
gr.HTML(html_section_3) |
|
|
|
app.launch() |
|
|