32 lines
1.1 KiB
Python
32 lines
1.1 KiB
Python
from typing import List, Optional
|
|
|
|
import gradio
|
|
|
|
from facefusion import state_manager, wording
|
|
from facefusion.choices import download_provider_set
|
|
from facefusion.core import clear_model_sets
|
|
from facefusion.typing import DownloadProviderKey
|
|
|
|
DOWNLOAD_PROVIDERS_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None
|
|
|
|
|
|
def render() -> None:
|
|
global DOWNLOAD_PROVIDERS_CHECKBOX_GROUP
|
|
|
|
DOWNLOAD_PROVIDERS_CHECKBOX_GROUP = gradio.CheckboxGroup(
|
|
label = wording.get('uis.download_providers_checkbox_group'),
|
|
choices = list(download_provider_set.keys()),
|
|
value = state_manager.get_item('download_providers')
|
|
)
|
|
|
|
|
|
def listen() -> None:
|
|
DOWNLOAD_PROVIDERS_CHECKBOX_GROUP.change(update_download_providers, inputs = DOWNLOAD_PROVIDERS_CHECKBOX_GROUP, outputs = DOWNLOAD_PROVIDERS_CHECKBOX_GROUP)
|
|
|
|
|
|
def update_download_providers(download_providers : List[DownloadProviderKey]) -> gradio.CheckboxGroup:
|
|
clear_model_sets()
|
|
download_providers = download_providers or list(download_provider_set.keys())
|
|
state_manager.set_item('download_providers', download_providers)
|
|
return gradio.CheckboxGroup(value = state_manager.get_item('download_providers'))
|