| from typing import Optional, Tuple |
|
|
| from gradio_rangeslider import RangeSlider |
|
|
| from facefusion import state_manager, translator |
| from facefusion.face_store import clear_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': translator.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_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) |
|
|
