| from typing import List, Optional | |
| import gradio | |
| from facefusion import state_manager, wording | |
| from facefusion.filesystem import get_file_name, resolve_file_paths | |
| from facefusion.processors.core import get_processors_modules | |
| from facefusion.uis.core import register_ui_component | |
| PROCESSORS_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None | |
| def render() -> None: | |
| global PROCESSORS_CHECKBOX_GROUP | |
| PROCESSORS_CHECKBOX_GROUP = gradio.CheckboxGroup( | |
| label = wording.get('uis.processors_checkbox_group'), | |
| choices = sort_processors(state_manager.get_item('processors')), | |
| value = state_manager.get_item('processors') | |
| ) | |
| register_ui_component('processors_checkbox_group', PROCESSORS_CHECKBOX_GROUP) | |
| def listen() -> None: | |
| PROCESSORS_CHECKBOX_GROUP.change(update_processors, inputs = PROCESSORS_CHECKBOX_GROUP, outputs = PROCESSORS_CHECKBOX_GROUP) | |
| def update_processors(processors : List[str]) -> gradio.CheckboxGroup: | |
| for processor_module in get_processors_modules(state_manager.get_item('processors')): | |
| if hasattr(processor_module, 'clear_inference_pool'): | |
| processor_module.clear_inference_pool() | |
| for processor_module in get_processors_modules(processors): | |
| if not processor_module.pre_check(): | |
| return gradio.CheckboxGroup() | |
| state_manager.set_item('processors', processors) | |
| return gradio.CheckboxGroup(value = state_manager.get_item('processors'), choices = sort_processors(state_manager.get_item('processors'))) | |
| def sort_processors(processors : List[str]) -> List[str]: | |
| available_processors = [ get_file_name(file_path) for file_path in resolve_file_paths('facefusion/processors/modules') ] | |
| return sorted(available_processors, key = lambda processor : processors.index(processor) if processor in processors else len(processors)) | |