Spaces:
Running
Running
File size: 4,829 Bytes
359a784 6a28782 359a784 6a28782 8e78d10 6a28782 359a784 8e78d10 359a784 6a28782 359a784 6a28782 359a784 6a28782 359a784 6a28782 359a784 6a28782 359a784 6a28782 359a784 6a28782 359a784 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 |
import gradio as gr
import subprocess as sp
import os
os.makedirs("./output", exist_ok=True)
def run(*args):
source, target, *rest_args = args
if not os.path.exists(source):
return "Source file does not exist"
if not os.path.exists(target):
return "Target file does not exist"
filename = os.path.basename(target)
output = f"./output/{filename}"
frame_processor = rest_args[0]
selected_frame_processors = ' '.join(frame_processor)
face_analyser_direction = rest_args[1]
face_analyser_age = rest_args[2]
face_analyser_gender = rest_args[3]
cmd = (
f"python run.py --execution-providers cuda -s {source} -t {target} -o {output} "
f"--frame-processors {selected_frame_processors} "
f"--face-analyser-direction {face_analyser_direction} "
)
if face_analyser_age != 'none':
cmd += f" --face-analyser-age {face_analyser_age}"
if face_analyser_gender != 'none':
cmd += f" --face-analyser-gender {face_analyser_gender}"
if len(rest_args) > 4:
skip_audio = rest_args[4]
keep_fps = rest_args[5]
keep_temp = rest_args[6]
if skip_audio:
cmd += " --skip-audio"
if keep_fps:
cmd += " --keep-fps"
if keep_temp:
cmd += " --keep-temp"
try:
print("Started...", cmd)
sp.run(cmd, shell=True, capture_output=True, text=True).stdout
return output
except Exception as e:
return f"An error occurred: {str(e)}"
def clear_output(output):
if os.path.exists(output):
os.remove(output)
return "Output file deleted"
else:
return "Output file does not exist"
def get_theme() -> gr.Theme:
return gr.themes.Soft(
primary_hue = gr.themes.colors.red,
secondary_hue = gr.themes.colors.gray,
font = gr.themes.GoogleFont('Inter')
).set(
background_fill_primary = '*neutral_50',
block_label_text_size = '*text_sm',
block_title_text_size = '*text_sm'
)
with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
with gr.Box():
gr.HTML('<center><a href="codegenius.me">DeepFakeAI 1.0.0</a></center>')
with gr.Box():
with gr.Column(scale=3):
frame_processor_checkbox = gr.CheckboxGroup(
choices = ['face_swapper', 'face_enhancer', 'frame_enhancer'],
label = 'FRAME PROCESSORS',
value = ['face_swapper'] # Default value
)
with gr.Box():
with gr.Column(scale=3):
face_analyser_direction_dropdown = gr.Dropdown(
label = 'FACE ANALYSER DIRECTION',
choices = ['left-right', 'right-left', 'top-bottom', 'bottom-top', 'small-large', 'large-small'],
value = 'left-right'
)
face_analyser_age_dropdown = gr.Dropdown(
label = 'FACE ANALYSER AGE',
choices = ['none'] + ['child', 'young', 'adult', 'old'],
value = 'none'
)
face_analyser_gender_dropdown = gr.Dropdown(
label = 'FACE ANALYSER GENDER',
choices = ['none'] + ['male', 'female'],
value = 'none'
)
with gr.Tab("Image: "):
source_image = gr.Image(type="filepath", label="SOURCE IMAGE")
target_image = gr.Image(type="filepath", label="TARGET IMAGE")
image_button = gr.Button("START")
clear_button = gr.Button("CLEAR")
image_output = gr.Image(label="OUTPUT")
image_button.click(
run,
inputs=[source_image, target_image, frame_processor_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown],
outputs=image_output
)
clear_button.click(clear_output, inputs=[image_output])
with gr.Tab("Video: "):
source_image_video = gr.Image(type="filepath", label="SOURCE IMAGE")
target_video = gr.Video(label="TARGET VIDEO")
with gr.Box():
skip_audio = gr.Checkbox(label="SKIP AUDIO")
keep_fps = gr.Checkbox(label="KEEP FPS")
keep_temp = gr.Checkbox(label="KEEP TEMP")
video_button = gr.Button("START")
clear_video_button = gr.Button("CLEAR")
video_output = gr.Video(label="OUTPUT")
video_button.click(
run,
inputs=[source_image_video, target_video, frame_processor_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown, skip_audio, keep_fps, keep_temp],
outputs=video_output
)
clear_video_button.click(clear_output, inputs=[video_output])
ui.launch(debug=True) |