| from typing import Tuple, Optional | |
| import gradio | |
| import facefusion.globals | |
| from facefusion import wording | |
| from facefusion.core import limit_resources, conditional_process | |
| from facefusion.uis.core import get_ui_component | |
| from facefusion.utilities import is_image, is_video, normalize_output_path, clear_temp | |
| OUTPUT_IMAGE : Optional[gradio.Image] = None | |
| OUTPUT_VIDEO : Optional[gradio.Video] = None | |
| OUTPUT_START_BUTTON : Optional[gradio.Button] = None | |
| OUTPUT_CLEAR_BUTTON : Optional[gradio.Button] = None | |
| def render() -> None: | |
| global OUTPUT_IMAGE | |
| global OUTPUT_VIDEO | |
| global OUTPUT_START_BUTTON | |
| global OUTPUT_CLEAR_BUTTON | |
| OUTPUT_IMAGE = gradio.Image( | |
| label = wording.get('output_image_or_video_label'), | |
| visible = False | |
| ) | |
| OUTPUT_VIDEO = gradio.Video( | |
| label = wording.get('output_image_or_video_label') | |
| ) | |
| OUTPUT_START_BUTTON = gradio.Button( | |
| value = wording.get('start_button_label'), | |
| variant = 'primary', | |
| size = 'sm' | |
| ) | |
| OUTPUT_CLEAR_BUTTON = gradio.Button( | |
| value = wording.get('clear_button_label'), | |
| size = 'sm' | |
| ) | |
| def listen() -> None: | |
| output_path_textbox = get_ui_component('output_path_textbox') | |
| if output_path_textbox: | |
| OUTPUT_START_BUTTON.click(start, inputs = output_path_textbox, outputs = [ OUTPUT_IMAGE, OUTPUT_VIDEO ]) | |
| OUTPUT_CLEAR_BUTTON.click(clear, outputs = [ OUTPUT_IMAGE, OUTPUT_VIDEO ]) | |
| def start(output_path : str) -> Tuple[gradio.Image, gradio.Video]: | |
| facefusion.globals.output_path = normalize_output_path(facefusion.globals.source_path, facefusion.globals.target_path, output_path) | |
| limit_resources() | |
| conditional_process() | |
| if is_image(facefusion.globals.output_path): | |
| return gradio.Image(value = facefusion.globals.output_path, visible = True), gradio.Video(value = None, visible = False) | |
| if is_video(facefusion.globals.output_path): | |
| return gradio.Image(value = None, visible = False), gradio.Video(value = facefusion.globals.output_path, visible = True) | |
| return gradio.Image(), gradio.Video() | |
| def clear() -> Tuple[gradio.Image, gradio.Video]: | |
| if facefusion.globals.target_path: | |
| clear_temp(facefusion.globals.target_path) | |
| return gradio.Image(value = None), gradio.Video(value = None) | |