| from typing import Optional, Tuple | |
| from gradio_rangeslider import RangeSlider | |
| from facefusion import state_manager, wording | |
| from facefusion.face_store import clear_static_faces | |
| from facefusion.filesystem import is_video | |
| from facefusion.uis.core import get_ui_components | |
| from facefusion.uis.types import ComponentOptions | |
| from facefusion.vision import count_video_frame_total | |
| TRIM_FRAME_RANGE_SLIDER : Optional[RangeSlider] = None | |
| def render() -> None: | |
| global TRIM_FRAME_RANGE_SLIDER | |
| trim_frame_range_slider_options : ComponentOptions =\ | |
| { | |
| 'label': wording.get('uis.trim_frame_slider'), | |
| 'minimum': 0, | |
| 'step': 1, | |
| 'visible': False | |
| } | |
| if is_video(state_manager.get_item('target_path')): | |
| video_frame_total = count_video_frame_total(state_manager.get_item('target_path')) | |
| trim_frame_start = state_manager.get_item('trim_frame_start') or 0 | |
| trim_frame_end = state_manager.get_item('trim_frame_end') or video_frame_total | |
| trim_frame_range_slider_options['maximum'] = video_frame_total | |
| trim_frame_range_slider_options['value'] = (trim_frame_start, trim_frame_end) | |
| trim_frame_range_slider_options['visible'] = True | |
| TRIM_FRAME_RANGE_SLIDER = RangeSlider(**trim_frame_range_slider_options) | |
| def listen() -> None: | |
| TRIM_FRAME_RANGE_SLIDER.release(update_trim_frame, inputs = TRIM_FRAME_RANGE_SLIDER) | |
| for ui_component in get_ui_components( | |
| [ | |
| 'target_image', | |
| 'target_video' | |
| ]): | |
| for method in [ 'change', 'clear' ]: | |
| getattr(ui_component, method)(remote_update, outputs = [ TRIM_FRAME_RANGE_SLIDER ]) | |
| def remote_update() -> RangeSlider: | |
| if is_video(state_manager.get_item('target_path')): | |
| video_frame_total = count_video_frame_total(state_manager.get_item('target_path')) | |
| state_manager.clear_item('trim_frame_start') | |
| state_manager.clear_item('trim_frame_end') | |
| return RangeSlider(value = (0, video_frame_total), maximum = video_frame_total, visible = True) | |
| return RangeSlider(visible = False) | |
| def update_trim_frame(trim_frame : Tuple[float, float]) -> None: | |
| clear_static_faces() | |
| trim_frame_start, trim_frame_end = trim_frame | |
| video_frame_total = count_video_frame_total(state_manager.get_item('target_path')) | |
| trim_frame_start = int(trim_frame_start) if trim_frame_start > 0 else None | |
| trim_frame_end = int(trim_frame_end) if trim_frame_end < video_frame_total else None | |
| state_manager.set_item('trim_frame_start', trim_frame_start) | |
| state_manager.set_item('trim_frame_end', trim_frame_end) | |