Spaces:
Paused
Paused
| from typing import Optional, Tuple, List | |
| import tempfile | |
| import gradio | |
| import facefusion.globals | |
| import facefusion.choices | |
| from facefusion import wording | |
| from facefusion.typing import OutputVideoEncoder, OutputVideoPreset, Fps | |
| from facefusion.filesystem import is_image, is_video | |
| from facefusion.uis.typing import ComponentName | |
| from facefusion.uis.core import get_ui_component, register_ui_component | |
| from facefusion.vision import detect_video_fps, create_video_resolutions, detect_video_resolution, pack_resolution | |
| OUTPUT_PATH_TEXTBOX : Optional[gradio.Textbox] = None | |
| OUTPUT_IMAGE_QUALITY_SLIDER : Optional[gradio.Slider] = None | |
| OUTPUT_VIDEO_ENCODER_DROPDOWN : Optional[gradio.Dropdown] = None | |
| OUTPUT_VIDEO_PRESET_DROPDOWN : Optional[gradio.Dropdown] = None | |
| OUTPUT_VIDEO_RESOLUTION_DROPDOWN : Optional[gradio.Dropdown] = None | |
| OUTPUT_VIDEO_QUALITY_SLIDER : Optional[gradio.Slider] = None | |
| OUTPUT_VIDEO_FPS_SLIDER : Optional[gradio.Slider] = None | |
| def render() -> None: | |
| global OUTPUT_PATH_TEXTBOX | |
| global OUTPUT_IMAGE_QUALITY_SLIDER | |
| global OUTPUT_VIDEO_ENCODER_DROPDOWN | |
| global OUTPUT_VIDEO_PRESET_DROPDOWN | |
| global OUTPUT_VIDEO_RESOLUTION_DROPDOWN | |
| global OUTPUT_VIDEO_QUALITY_SLIDER | |
| global OUTPUT_VIDEO_FPS_SLIDER | |
| OUTPUT_PATH_TEXTBOX = gradio.Textbox( | |
| label = wording.get('output_path_textbox_label'), | |
| value = facefusion.globals.output_path or tempfile.gettempdir(), | |
| max_lines = 1 | |
| ) | |
| OUTPUT_IMAGE_QUALITY_SLIDER = gradio.Slider( | |
| label = wording.get('output_image_quality_slider_label'), | |
| value = facefusion.globals.output_image_quality, | |
| step = facefusion.choices.output_image_quality_range[1] - facefusion.choices.output_image_quality_range[0], | |
| minimum = facefusion.choices.output_image_quality_range[0], | |
| maximum = facefusion.choices.output_image_quality_range[-1], | |
| visible = is_image(facefusion.globals.target_path) | |
| ) | |
| OUTPUT_VIDEO_ENCODER_DROPDOWN = gradio.Dropdown( | |
| label = wording.get('output_video_encoder_dropdown_label'), | |
| choices = facefusion.choices.output_video_encoders, | |
| value = facefusion.globals.output_video_encoder, | |
| visible = is_video(facefusion.globals.target_path) | |
| ) | |
| OUTPUT_VIDEO_PRESET_DROPDOWN = gradio.Dropdown( | |
| label = wording.get('output_video_preset_dropdown_label'), | |
| choices = facefusion.choices.output_video_presets, | |
| value = facefusion.globals.output_video_preset, | |
| visible = is_video(facefusion.globals.target_path) | |
| ) | |
| OUTPUT_VIDEO_QUALITY_SLIDER = gradio.Slider( | |
| label = wording.get('output_video_quality_slider_label'), | |
| value = facefusion.globals.output_video_quality, | |
| step = facefusion.choices.output_video_quality_range[1] - facefusion.choices.output_video_quality_range[0], | |
| minimum = facefusion.choices.output_video_quality_range[0], | |
| maximum = facefusion.choices.output_video_quality_range[-1], | |
| visible = is_video(facefusion.globals.target_path) | |
| ) | |
| OUTPUT_VIDEO_RESOLUTION_DROPDOWN = gradio.Dropdown( | |
| label = wording.get('output_video_resolution_dropdown_label'), | |
| choices = create_video_resolutions(facefusion.globals.target_path), | |
| value = facefusion.globals.output_video_resolution, | |
| visible = is_video(facefusion.globals.target_path) | |
| ) | |
| OUTPUT_VIDEO_FPS_SLIDER = gradio.Slider( | |
| label = wording.get('output_video_fps_slider_label'), | |
| value = facefusion.globals.output_video_fps, | |
| step = 0.01, | |
| minimum = 1, | |
| maximum = 60, | |
| visible = is_video(facefusion.globals.target_path) | |
| ) | |
| register_ui_component('output_path_textbox', OUTPUT_PATH_TEXTBOX) | |
| def listen() -> None: | |
| OUTPUT_PATH_TEXTBOX.change(update_output_path, inputs = OUTPUT_PATH_TEXTBOX) | |
| OUTPUT_IMAGE_QUALITY_SLIDER.change(update_output_image_quality, inputs = OUTPUT_IMAGE_QUALITY_SLIDER) | |
| OUTPUT_VIDEO_ENCODER_DROPDOWN.change(update_output_video_encoder, inputs = OUTPUT_VIDEO_ENCODER_DROPDOWN) | |
| OUTPUT_VIDEO_PRESET_DROPDOWN.change(update_output_video_preset, inputs = OUTPUT_VIDEO_PRESET_DROPDOWN) | |
| OUTPUT_VIDEO_QUALITY_SLIDER.change(update_output_video_quality, inputs = OUTPUT_VIDEO_QUALITY_SLIDER) | |
| OUTPUT_VIDEO_RESOLUTION_DROPDOWN.change(update_output_video_resolution, inputs = OUTPUT_VIDEO_RESOLUTION_DROPDOWN) | |
| OUTPUT_VIDEO_FPS_SLIDER.change(update_output_video_fps, inputs = OUTPUT_VIDEO_FPS_SLIDER) | |
| multi_component_names : List[ComponentName] =\ | |
| [ | |
| 'source_image', | |
| 'target_image', | |
| 'target_video' | |
| ] | |
| for component_name in multi_component_names: | |
| component = get_ui_component(component_name) | |
| if component: | |
| for method in [ 'upload', 'change', 'clear' ]: | |
| getattr(component, method)(remote_update, outputs = [ OUTPUT_IMAGE_QUALITY_SLIDER, OUTPUT_VIDEO_ENCODER_DROPDOWN, OUTPUT_VIDEO_PRESET_DROPDOWN, OUTPUT_VIDEO_QUALITY_SLIDER, OUTPUT_VIDEO_RESOLUTION_DROPDOWN, OUTPUT_VIDEO_FPS_SLIDER ]) | |
| def remote_update() -> Tuple[gradio.Slider, gradio.Dropdown, gradio.Dropdown, gradio.Slider, gradio.Dropdown, gradio.Slider]: | |
| if is_image(facefusion.globals.target_path): | |
| return gradio.Slider(visible = True), gradio.Dropdown(visible = False), gradio.Dropdown(visible = False), gradio.Slider(visible = False), gradio.Dropdown(visible = False, value = None, choices = None), gradio.Slider(visible = False, value = None) | |
| if is_video(facefusion.globals.target_path): | |
| target_video_resolution = detect_video_resolution(facefusion.globals.target_path) | |
| output_video_resolution = pack_resolution(target_video_resolution) | |
| output_video_resolutions = create_video_resolutions(facefusion.globals.target_path) | |
| output_video_fps = detect_video_fps(facefusion.globals.target_path) | |
| return gradio.Slider(visible = False), gradio.Dropdown(visible = True), gradio.Dropdown(visible = True), gradio.Slider(visible = True), gradio.Dropdown(visible = True, value = output_video_resolution, choices = output_video_resolutions), gradio.Slider(visible = True, value = output_video_fps) | |
| return gradio.Slider(visible = False), gradio.Dropdown(visible = False), gradio.Dropdown(visible = False), gradio.Slider(visible = False), gradio.Dropdown(visible = False, value = None, choices = None), gradio.Slider(visible = False, value = None) | |
| def update_output_path(output_path : str) -> None: | |
| facefusion.globals.output_path = output_path | |
| def update_output_image_quality(output_image_quality : int) -> None: | |
| facefusion.globals.output_image_quality = output_image_quality | |
| def update_output_video_encoder(output_video_encoder: OutputVideoEncoder) -> None: | |
| facefusion.globals.output_video_encoder = output_video_encoder | |
| def update_output_video_preset(output_video_preset : OutputVideoPreset) -> None: | |
| facefusion.globals.output_video_preset = output_video_preset | |
| def update_output_video_quality(output_video_quality : int) -> None: | |
| facefusion.globals.output_video_quality = output_video_quality | |
| def update_output_video_resolution(output_video_resolution : str) -> None: | |
| facefusion.globals.output_video_resolution = output_video_resolution | |
| def update_output_video_fps(output_video_fps : Fps) -> None: | |
| facefusion.globals.output_video_fps = output_video_fps | |