| from typing import Optional, Sequence, Tuple |
|
|
| import gradio |
|
|
| import facefusion.choices |
| from facefusion import face_detector, state_manager, translator |
| from facefusion.common_helper import calculate_float_step, get_last |
| from facefusion.sanitizer import sanitize_int_range |
| from facefusion.types import Angle, FaceDetectorModel, Score |
| from facefusion.uis.core import register_ui_component |
| from facefusion.uis.types import ComponentOptions |
|
|
| FACE_DETECTOR_MODEL_DROPDOWN : Optional[gradio.Dropdown] = None |
| FACE_DETECTOR_SIZE_DROPDOWN : Optional[gradio.Dropdown] = None |
| FACE_DETECTOR_MARGIN_SLIDER : Optional[gradio.Slider] = None |
| FACE_DETECTOR_ANGLES_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None |
| FACE_DETECTOR_SCORE_SLIDER : Optional[gradio.Slider] = None |
|
|
|
|
| def render() -> None: |
| global FACE_DETECTOR_MODEL_DROPDOWN |
| global FACE_DETECTOR_SIZE_DROPDOWN |
| global FACE_DETECTOR_MARGIN_SLIDER |
| global FACE_DETECTOR_ANGLES_CHECKBOX_GROUP |
| global FACE_DETECTOR_SCORE_SLIDER |
|
|
| face_detector_size_dropdown_options : ComponentOptions =\ |
| { |
| 'label': translator.get('uis.face_detector_size_dropdown'), |
| 'value': state_manager.get_item('face_detector_size') |
| } |
| if state_manager.get_item('face_detector_size') in facefusion.choices.face_detector_set[state_manager.get_item('face_detector_model')]: |
| face_detector_size_dropdown_options['choices'] = facefusion.choices.face_detector_set[state_manager.get_item('face_detector_model')] |
| with gradio.Row(): |
| FACE_DETECTOR_MODEL_DROPDOWN = gradio.Dropdown( |
| label = translator.get('uis.face_detector_model_dropdown'), |
| choices = facefusion.choices.face_detector_models, |
| value = state_manager.get_item('face_detector_model') |
| ) |
| FACE_DETECTOR_SIZE_DROPDOWN = gradio.Dropdown(**face_detector_size_dropdown_options) |
| FACE_DETECTOR_MARGIN_SLIDER = gradio.Slider( |
| label = translator.get('uis.face_detector_margin_slider'), |
| value = state_manager.get_item('face_detector_margin')[0], |
| step = calculate_float_step(facefusion.choices.face_detector_margin_range), |
| minimum = facefusion.choices.face_detector_margin_range[0], |
| maximum = facefusion.choices.face_detector_margin_range[-1] |
| ) |
| FACE_DETECTOR_ANGLES_CHECKBOX_GROUP = gradio.CheckboxGroup( |
| label = translator.get('uis.face_detector_angles_checkbox_group'), |
| choices = facefusion.choices.face_detector_angles, |
| value = state_manager.get_item('face_detector_angles') |
| ) |
| FACE_DETECTOR_SCORE_SLIDER = gradio.Slider( |
| label = translator.get('uis.face_detector_score_slider'), |
| value = state_manager.get_item('face_detector_score'), |
| step = calculate_float_step(facefusion.choices.face_detector_score_range), |
| minimum = facefusion.choices.face_detector_score_range[0], |
| maximum = facefusion.choices.face_detector_score_range[-1] |
| ) |
| register_ui_component('face_detector_model_dropdown', FACE_DETECTOR_MODEL_DROPDOWN) |
| register_ui_component('face_detector_size_dropdown', FACE_DETECTOR_SIZE_DROPDOWN) |
| register_ui_component('face_detector_margin_slider', FACE_DETECTOR_MARGIN_SLIDER) |
| register_ui_component('face_detector_angles_checkbox_group', FACE_DETECTOR_ANGLES_CHECKBOX_GROUP) |
| register_ui_component('face_detector_score_slider', FACE_DETECTOR_SCORE_SLIDER) |
|
|
|
|
| def listen() -> None: |
| FACE_DETECTOR_MODEL_DROPDOWN.change(update_face_detector_model, inputs = FACE_DETECTOR_MODEL_DROPDOWN, outputs = [ FACE_DETECTOR_MODEL_DROPDOWN, FACE_DETECTOR_SIZE_DROPDOWN ]) |
| FACE_DETECTOR_SIZE_DROPDOWN.change(update_face_detector_size, inputs = FACE_DETECTOR_SIZE_DROPDOWN) |
| FACE_DETECTOR_MARGIN_SLIDER.release(update_face_detector_margin, inputs=FACE_DETECTOR_MARGIN_SLIDER) |
| FACE_DETECTOR_ANGLES_CHECKBOX_GROUP.change(update_face_detector_angles, inputs = FACE_DETECTOR_ANGLES_CHECKBOX_GROUP, outputs = FACE_DETECTOR_ANGLES_CHECKBOX_GROUP) |
| FACE_DETECTOR_SCORE_SLIDER.release(update_face_detector_score, inputs = FACE_DETECTOR_SCORE_SLIDER) |
|
|
|
|
| def update_face_detector_model(face_detector_model : FaceDetectorModel) -> Tuple[gradio.Dropdown, gradio.Dropdown]: |
| face_detector.clear_inference_pool() |
| state_manager.set_item('face_detector_model', face_detector_model) |
|
|
| if face_detector.pre_check(): |
| face_detector_size_choices = facefusion.choices.face_detector_set.get(state_manager.get_item('face_detector_model')) |
| state_manager.set_item('face_detector_size', get_last(face_detector_size_choices)) |
| return gradio.Dropdown(value = state_manager.get_item('face_detector_model')), gradio.Dropdown(value = state_manager.get_item('face_detector_size'), choices = face_detector_size_choices) |
| return gradio.Dropdown(), gradio.Dropdown() |
|
|
|
|
| def update_face_detector_size(face_detector_size : str) -> None: |
| state_manager.set_item('face_detector_size', face_detector_size) |
|
|
|
|
| def update_face_detector_margin(face_detector_margin : int) -> None: |
| face_detector_margin = sanitize_int_range(face_detector_margin, facefusion.choices.face_detector_margin_range) |
| state_manager.set_item('face_detector_margin', (face_detector_margin, face_detector_margin, face_detector_margin, face_detector_margin)) |
|
|
|
|
| def update_face_detector_angles(face_detector_angles : Sequence[Angle]) -> gradio.CheckboxGroup: |
| face_detector_angles = face_detector_angles or facefusion.choices.face_detector_angles |
|
|
| state_manager.set_item('face_detector_angles', face_detector_angles) |
| return gradio.CheckboxGroup(value = state_manager.get_item('face_detector_angles')) |
|
|
|
|
| def update_face_detector_score(face_detector_score : Score) -> None: |
| state_manager.set_item('face_detector_score', face_detector_score) |
|
|
