| from typing import List, Optional |
|
|
| import gradio |
|
|
| import facefusion.choices |
| from facefusion import state_manager, translator |
| from facefusion.common_helper import calculate_int_step |
| from facefusion.types import BenchmarkMode, BenchmarkResolution |
|
|
| BENCHMARK_MODE_DROPDOWN : Optional[gradio.Dropdown] = None |
| BENCHMARK_RESOLUTIONS_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None |
| BENCHMARK_CYCLE_COUNT_SLIDER : Optional[gradio.Button] = None |
|
|
|
|
| def render() -> None: |
| global BENCHMARK_MODE_DROPDOWN |
| global BENCHMARK_RESOLUTIONS_CHECKBOX_GROUP |
| global BENCHMARK_CYCLE_COUNT_SLIDER |
|
|
| BENCHMARK_MODE_DROPDOWN = gradio.Dropdown( |
| label = translator.get('uis.benchmark_mode_dropdown'), |
| choices = facefusion.choices.benchmark_modes, |
| value = state_manager.get_item('benchmark_mode') |
| ) |
| BENCHMARK_RESOLUTIONS_CHECKBOX_GROUP = gradio.CheckboxGroup( |
| label = translator.get('uis.benchmark_resolutions_checkbox_group'), |
| choices = facefusion.choices.benchmark_resolutions, |
| value = state_manager.get_item('benchmark_resolutions') |
| ) |
| BENCHMARK_CYCLE_COUNT_SLIDER = gradio.Slider( |
| label = translator.get('uis.benchmark_cycle_count_slider'), |
| value = state_manager.get_item('benchmark_cycle_count'), |
| step = calculate_int_step(facefusion.choices.benchmark_cycle_count_range), |
| minimum = facefusion.choices.benchmark_cycle_count_range[0], |
| maximum = facefusion.choices.benchmark_cycle_count_range[-1] |
| ) |
|
|
|
|
| def listen() -> None: |
| BENCHMARK_MODE_DROPDOWN.change(update_benchmark_mode, inputs = BENCHMARK_MODE_DROPDOWN) |
| BENCHMARK_RESOLUTIONS_CHECKBOX_GROUP.change(update_benchmark_resolutions, inputs = BENCHMARK_RESOLUTIONS_CHECKBOX_GROUP) |
| BENCHMARK_CYCLE_COUNT_SLIDER.release(update_benchmark_cycle_count, inputs = BENCHMARK_CYCLE_COUNT_SLIDER) |
|
|
|
|
| def update_benchmark_mode(benchmark_mode : BenchmarkMode) -> None: |
| state_manager.set_item('benchmark_mode', benchmark_mode) |
|
|
|
|
| def update_benchmark_resolutions(benchmark_resolutions : List[BenchmarkResolution]) -> None: |
| state_manager.set_item('benchmark_resolutions', benchmark_resolutions) |
|
|
|
|
| def update_benchmark_cycle_count(benchmark_cycle_count : int) -> None: |
| state_manager.set_item('benchmark_cycle_count', benchmark_cycle_count) |
|
|
