File size: 768 Bytes
edcba35 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | from typing import Optional
import gradio
import facefusion.choices
from facefusion import state_manager, translator
from facefusion.types import WorkflowStrategy
WORKFLOW_STRATEGY_DROPDOWN : Optional[gradio.Dropdown] = None
def render() -> None:
global WORKFLOW_STRATEGY_DROPDOWN
WORKFLOW_STRATEGY_DROPDOWN = gradio.Dropdown(
label = translator.get('uis.workflow_strategy_dropdown'),
choices = facefusion.choices.workflow_strategies,
value = state_manager.get_item('workflow_strategy')
)
def listen() -> None:
WORKFLOW_STRATEGY_DROPDOWN.change(update_workflow_strategy, inputs = WORKFLOW_STRATEGY_DROPDOWN)
def update_workflow_strategy(workflow_strategy : WorkflowStrategy) -> None:
state_manager.set_item('workflow_strategy', workflow_strategy)
|