Spaces:
Runtime error
Runtime error
| 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, *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" | |
| 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 = rest_args[0] | |
| selected_frame_processors = ' '.join(frame_processor) | |
| face_analyser_direction = rest_args[1] | |
| face_recognition = rest_args[2] | |
| face_analyser_gender = rest_args[3] | |
| cmd = ( | |
| f"python run.py --execution-providers cpu -s {source} -t {target} -o {output} " | |
| f"--frame-processors {selected_frame_processors} " | |
| f"--face-analyser-direction {face_analyser_direction} " | |
| ) | |
| if face_recognition != 'none': | |
| cmd += f"--face-recognition {face_recognition} " | |
| 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) | |
| 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_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="https://codegenius.me">DeepFakeAI 1.0.1</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 RECOGNITION', | |
| choices = ['none'] + ['reference', 'many'], | |
| value = 'reference' | |
| ) | |
| face_analyser_gender_dropdown = gr.Dropdown( | |
| label = 'FACE ANALYSER GENDER', | |
| choices = ['none'] + ['male', 'female'], | |
| value = 'none' | |
| ) | |
| unique_id = gr.Textbox(value=str(uuid.uuid4()), visible=False) | |
| 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.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_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown], | |
| 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") | |
| 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.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_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown, skip_audio, keep_fps, keep_temp], | |
| outputs=video_output | |
| ) | |
| clear_video_button.click(fn=clear_output, inputs=unique_id) | |
| ui.launch(debug=True) |