Spaces:
Running
Running
| from typing import Optional, Tuple | |
| import gradio | |
| import facefusion.globals | |
| import facefusion.choices | |
| from facefusion import wording | |
| from facefusion.typing import TempFrameFormat | |
| from facefusion.utilities import is_video | |
| from facefusion.uis.core import get_ui_component | |
| TEMP_FRAME_FORMAT_DROPDOWN : Optional[gradio.Dropdown] = None | |
| TEMP_FRAME_QUALITY_SLIDER : Optional[gradio.Slider] = None | |
| def render() -> None: | |
| global TEMP_FRAME_FORMAT_DROPDOWN | |
| global TEMP_FRAME_QUALITY_SLIDER | |
| TEMP_FRAME_FORMAT_DROPDOWN = gradio.Dropdown( | |
| label = wording.get('temp_frame_format_dropdown_label'), | |
| choices = facefusion.choices.temp_frame_formats, | |
| value = facefusion.globals.temp_frame_format, | |
| visible = is_video(facefusion.globals.target_path) | |
| ) | |
| TEMP_FRAME_QUALITY_SLIDER = gradio.Slider( | |
| label = wording.get('temp_frame_quality_slider_label'), | |
| value = facefusion.globals.temp_frame_quality, | |
| step = facefusion.choices.temp_frame_quality_range[1] - facefusion.choices.temp_frame_quality_range[0], | |
| minimum = facefusion.choices.temp_frame_quality_range[0], | |
| maximum = facefusion.choices.temp_frame_quality_range[-1], | |
| visible = is_video(facefusion.globals.target_path) | |
| ) | |
| def listen() -> None: | |
| TEMP_FRAME_FORMAT_DROPDOWN.select(update_temp_frame_format, inputs = TEMP_FRAME_FORMAT_DROPDOWN) | |
| TEMP_FRAME_QUALITY_SLIDER.change(update_temp_frame_quality, inputs = TEMP_FRAME_QUALITY_SLIDER) | |
| target_video = get_ui_component('target_video') | |
| if target_video: | |
| for method in [ 'upload', 'change', 'clear' ]: | |
| getattr(target_video, method)(remote_update, outputs = [ TEMP_FRAME_FORMAT_DROPDOWN, TEMP_FRAME_QUALITY_SLIDER ]) | |
| def remote_update() -> Tuple[gradio.Dropdown, gradio.Slider]: | |
| if is_video(facefusion.globals.target_path): | |
| return gradio.Dropdown(visible = True), gradio.Slider(visible = True) | |
| return gradio.Dropdown(visible = False), gradio.Slider(visible = False) | |
| def update_temp_frame_format(temp_frame_format : TempFrameFormat) -> None: | |
| facefusion.globals.temp_frame_format = temp_frame_format | |
| def update_temp_frame_quality(temp_frame_quality : int) -> None: | |
| facefusion.globals.temp_frame_quality = temp_frame_quality | |