Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| import subprocess as sp | |
| import os | |
| import uuid | |
| import time | |
| import shutil | |
| os.makedirs("./output", exist_ok=True) | |
| def run(*args): | |
| source, target, unique_id, frame_processor_checkbox, face_swapper_model_dropdown, face_enhancer_model_dropdown, frame_enhancer_model_dropdown, face_debugger_items_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown, face_mask_type_checkbox, *video_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" | |
| remove_old_directories("./output", num_minutes=60) | |
| filename = os.path.basename(target) | |
| os.makedirs(f"./output/{unique_id}",exist_ok=True) | |
| output = f"./output/{unique_id}/{filename}" | |
| frame_processor = frame_processor_checkbox | |
| selected_frame_processors = ' '.join(frame_processor) | |
| face_analyser_direction = face_analyser_direction_dropdown | |
| face_recognition = face_analyser_age_dropdown | |
| face_analyser_gender = face_analyser_gender_dropdown | |
| cmd = ( | |
| f"python run.py --execution-providers cpu --headless -s {source} -t {target} -o {output} " | |
| f"--frame-processors {selected_frame_processors} " | |
| f"--face-analyser-order {face_analyser_direction} " | |
| ) | |
| if "face_swapper" in selected_frame_processors: | |
| cmd += f"--face-swapper-model {face_swapper_model_dropdown} " | |
| if "face_enhancer" in selected_frame_processors: | |
| cmd += f"--face-enhancer-model {face_enhancer_model_dropdown} " | |
| if "frame_enhancer" in selected_frame_processors: | |
| cmd += f"--frame-enhancer-model {frame_enhancer_model_dropdown} " | |
| if "face_debugger" in selected_frame_processors: | |
| selected_face_debugger_items = ' '.join(face_debugger_items_checkbox) | |
| cmd += f"--face-debugger-items {selected_face_debugger_items} " | |
| if face_recognition != 'none': | |
| cmd += f"--face-selector-mode {face_recognition} " | |
| if face_analyser_gender != 'none': | |
| cmd += f"--face-analyser-gender {face_analyser_gender} " | |
| if video_args: | |
| skip_audio, keep_fps, keep_temp = video_args | |
| if skip_audio: | |
| cmd += "--skip-audio " | |
| if keep_fps: | |
| cmd += "--keep-fps " | |
| if keep_temp: | |
| cmd += "--keep-temp " | |
| try: | |
| print("Started...", cmd) | |
| output_text = sp.run(cmd, shell=True, capture_output=True, text=True).stdout | |
| print(output_text) | |
| return output | |
| except Exception as e: | |
| return f"An error occurred: {str(e)}" | |
| def clear_output(unique_id): | |
| try: | |
| output_path = f"./output/{unique_id}" | |
| if os.path.exists(output_path): | |
| print("Trying to delete ") | |
| for filename in os.listdir(output_path): | |
| file_path = os.path.join(output_path, filename) | |
| if os.path.isfile(file_path): | |
| os.remove(file_path) | |
| print(f"Output files in {output_path} are deleted") | |
| return "Output files for unique_id deleted" | |
| else: | |
| print(f"Output files in {output_path} does not exist") | |
| return "Output directory for (output_path} does not exist" | |
| except Exception as e: | |
| return f"An error occurred: {str(e)}" | |
| def remove_old_directories(directory, num_minutes=60): | |
| now = time.time() | |
| for r, d, f in os.walk(directory): | |
| for dir_name in d: | |
| dir_path = os.path.join(r, dir_name) | |
| timestamp = os.path.getmtime(dir_path) | |
| age_minutes = (now - timestamp) / 60 # Convert to minutes | |
| if age_minutes >= num_minutes: | |
| try: | |
| print("Removing", dir_path) | |
| shutil.rmtree(dir_path) | |
| print("Directory removed:", dir_path) | |
| except Exception as e: | |
| print(e) | |
| pass | |
| def get_css() -> str: | |
| fixes_css = """ | |
| :root:root:root button:not([class]) | |
| { | |
| border-radius: 0.375rem; | |
| float: left; | |
| overflow: hidden; | |
| width: 100%; | |
| } | |
| """ | |
| overrides_css = """ | |
| :root:root:root input[type="number"] | |
| { | |
| max-width: 6rem; | |
| } | |
| :root:root:root [type="checkbox"], | |
| :root:root:root [type="radio"] | |
| { | |
| border-radius: 50%; | |
| height: 1.125rem; | |
| width: 1.125rem; | |
| } | |
| :root:root:root input[type="range"] | |
| { | |
| height: 0.5rem; | |
| } | |
| :root:root:root input[type="range"]::-moz-range-thumb, | |
| :root:root:root input[type="range"]::-webkit-slider-thumb | |
| { | |
| background: var(--neutral-300); | |
| border: unset; | |
| border-radius: 50%; | |
| height: 1.125rem; | |
| width: 1.125rem; | |
| } | |
| :root:root:root input[type="range"]::-webkit-slider-thumb | |
| { | |
| margin-top: 0.375rem; | |
| } | |
| :root:root:root .grid-wrap.fixed-height | |
| { | |
| min-height: unset; | |
| } | |
| :root:root:root .grid-container | |
| { | |
| grid-auto-rows: minmax(5em, 1fr); | |
| grid-template-columns: repeat(var(--grid-cols), minmax(5em, 1fr)); | |
| grid-template-rows: repeat(var(--grid-rows), minmax(5em, 1fr)); | |
| } | |
| """ | |
| return fixes_css + overrides_css | |
| def get_html(): | |
| with open("DeepFakeAI.html", "r") as f: | |
| html = f.read() | |
| return html | |
| def get_theme() -> gr.Theme: | |
| return gr.themes.Base( | |
| primary_hue = gr.themes.colors.red, | |
| secondary_hue = gr.themes.colors.neutral, | |
| font = gr.themes.GoogleFont('Open Sans') | |
| ).set( | |
| background_fill_primary = '*neutral_100', | |
| block_background_fill = 'white', | |
| block_border_width = '0', | |
| block_label_background_fill = '*primary_100', | |
| block_label_background_fill_dark = '*primary_600', | |
| block_label_border_width = 'none', | |
| block_label_margin = '0.5rem', | |
| block_label_radius = '*radius_md', | |
| block_label_text_color = '*primary_500', | |
| block_label_text_color_dark = 'white', | |
| block_label_text_weight = '600', | |
| block_title_background_fill = '*primary_100', | |
| block_title_background_fill_dark = '*primary_600', | |
| block_title_padding = '*block_label_padding', | |
| block_title_radius = '*block_label_radius', | |
| block_title_text_color = '*primary_500', | |
| block_title_text_size = '*text_sm', | |
| block_title_text_weight = '600', | |
| block_padding = '0.5rem', | |
| border_color_primary = 'transparent', | |
| border_color_primary_dark = 'transparent', | |
| #button_large_padding = '2rem 0.5rem', | |
| #button_large_text_weight = 'normal', | |
| #button_primary_background_fill = '*primary_500', | |
| #button_primary_text_color = 'white', | |
| #button_secondary_background_fill = 'white', | |
| #button_secondary_border_color = 'transparent', | |
| #button_secondary_border_color_dark = 'transparent', | |
| #button_secondary_border_color_hover = 'transparent', | |
| #button_secondary_border_color_hover_dark = 'transparent', | |
| #button_secondary_text_color = '*neutral_800', | |
| #button_small_padding = '0.75rem', | |
| checkbox_background_color = '*neutral_200', | |
| checkbox_background_color_selected = '*primary_600', | |
| checkbox_background_color_selected_dark = '*primary_700', | |
| checkbox_border_color_focus = '*primary_500', | |
| checkbox_border_color_focus_dark = '*primary_600', | |
| checkbox_border_color_selected = '*primary_600', | |
| checkbox_border_color_selected_dark = '*primary_700', | |
| checkbox_label_background_fill = '*neutral_50', | |
| checkbox_label_background_fill_hover = '*neutral_50', | |
| checkbox_label_background_fill_selected = '*primary_500', | |
| checkbox_label_background_fill_selected_dark = '*primary_600', | |
| checkbox_label_text_color_selected = 'white', | |
| input_background_fill = '*neutral_50', | |
| shadow_drop = 'none', | |
| slider_color = '*primary_500', | |
| slider_color_dark = '*primary_600' | |
| ) | |
| with gr.Blocks(theme=get_theme(), css=get_css(), title="DeepFakeAI 2.0.1") as ui: | |
| with gr.Row(): | |
| gr.HTML(get_html()) | |
| with gr.Row(): | |
| with gr.Column(scale=3): | |
| frame_processor_checkbox = gr.CheckboxGroup( | |
| choices = ['face_swapper', 'face_enhancer', 'frame_enhancer', 'face_debugger'], | |
| label = 'FRAME PROCESSORS', | |
| value = ['face_swapper'] # Default value | |
| ) | |
| with gr.Row(): | |
| with gr.Column(scale=3): | |
| face_swapper_model_dropdown = gr.Dropdown( | |
| choices = ['blendswap_256', 'inswapper_128', 'inswapper_128_fp16', 'simswap_256', 'simswap_512_unofficial'], | |
| label = 'FACE SWAPPER MODEL', | |
| value = 'inswapper_128', | |
| interactive=True, | |
| visible = True #'face_swapper' in frame_processor_checkbox.value | |
| ) | |
| face_enhancer_model_dropdown = gr.Dropdown( | |
| choices = ['codeformer', 'gfpgan_1.2', 'gfpgan_1.3', 'gfpgan_1.4', 'gpen_bfr_256', 'gpen_bfr_512', 'restoreformer'], | |
| label = 'FACE ENHANCER MODEL', | |
| value = 'gfpgan_1.2', | |
| visible = True, #'face_enhancer' in frame_processor_checkbox.value | |
| interactive=True | |
| ) | |
| frame_enhancer_model_dropdown = gr.Dropdown( | |
| choices = ['real_esrgan_x2plus', 'real_esrgan_x4plus', 'real_esrnet_x4plus'], | |
| label = 'FRAME ENHANCER MODEL', | |
| value = 'real_esrgan_x4plus', | |
| interactive=True, | |
| visible = True #'frame-enhancer' in frame_processor_checkbox.value | |
| ) | |
| face_debugger_items_checkbox = gr.CheckboxGroup( | |
| choices = ['face-mask', 'bbox', 'kps', 'score'], | |
| label = 'FACE DEBUGGER ITEMS', | |
| value = 'face-mask', | |
| interactive=True, | |
| visible = True | |
| ) | |
| with gr.Row(): | |
| 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', 'best-worst', 'worst-best'], | |
| value = 'left-right' | |
| ) | |
| face_analyser_age_dropdown = gr.Dropdown( | |
| label = 'FACE RECOGNITION', | |
| choices = ['none'] + ['reference', 'many'], | |
| value = 'reference' | |
| ) | |
| face_analyser_gender_dropdown = gr.Dropdown( | |
| label = 'FACE ANALYSER GENDER', | |
| choices = ['none'] + ['male', 'female'], | |
| value = 'none' | |
| ) | |
| with gr.Row(): | |
| with gr.Column(scale=3): | |
| face_mask_type_checkbox = gr.CheckboxGroup( | |
| choices = ['box', 'occlusion', 'region'], | |
| label = 'FACE MASK TYPES', | |
| value = 'box', | |
| interactive=True ) | |
| unique_id = gr.Textbox(value=str(uuid.uuid4()), visible=False) | |
| with gr.Tab("Image: "): | |
| source_image = gr.Image(type="filepath", label="SOURCE IMAGE",)# sources=["upload", "webcam", "clipboard"]) | |
| target_image = gr.Image(type="filepath", label="TARGET IMAGE",)# sources=["upload", "webcam", "clipboard"]) | |
| image_button = gr.Button("START") | |
| clear_button = gr.ClearButton(value="CLEAR") | |
| image_output = gr.Image(label="OUTPUT") | |
| clear_button.add(image_output) | |
| image_button.click( | |
| run, | |
| inputs=[source_image, target_image, unique_id, frame_processor_checkbox, face_swapper_model_dropdown, face_enhancer_model_dropdown, frame_enhancer_model_dropdown, face_debugger_items_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown, face_mask_type_checkbox], | |
| outputs=image_output | |
| ) | |
| clear_button.click(fn=clear_output, inputs=unique_id) | |
| with gr.Tab("Video: "): | |
| source_image_video = gr.Image(type="filepath", label="SOURCE IMAGE",)# sources=["upload", "webcam", "clipboard"]) | |
| target_video = gr.Video(label="TARGET VIDEO", sources=["upload"]) | |
| with gr.Row(): | |
| 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.ClearButton(value="CLEAR") | |
| video_output = gr.Video(label="OUTPUT") | |
| clear_video_button.add(video_output) | |
| video_button.click( | |
| run, | |
| inputs=[source_image_video, target_video, unique_id, frame_processor_checkbox, face_swapper_model_dropdown, face_enhancer_model_dropdown, frame_enhancer_model_dropdown, face_debugger_items_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown, face_mask_type_checkbox, skip_audio, keep_fps, keep_temp], | |
| outputs=video_output | |
| ) | |
| clear_video_button.click(fn=clear_output, inputs=unique_id) | |
| with gr.Row(): | |
| gr.HTML("""<center>Made with ❤️ by <a href='https://codegenius.me' style="font-weight: bold; font-style: italic; color: red;">Ashiq Hussain</a></center>""") | |
| ui.launch(debug=True) |