13761af044
* Cleanup after age modifier PR * Cleanup after age modifier PR * Use OpenVino 2024.2.0 for installer * Prepare 3.0.0 for installer * Fix benchmark suite, Introduce sync_item() for state manager * Fix lint * Render slide preview also in lower res * Lower thread and queue count to avoid false usage * Fix spacing * Feat/jobs UI (#627) * Jobs UI part1 * Change naming * Jobs UI part2 * Jobs UI part3 * Jobs UI part4 * Jobs UI part4 * Jobs UI part5 * Jobs UI part6 * Jobs UI part7 * Jobs UI part8 * Jobs UI part9 * Jobs UI part10 * Jobs UI part11 * Jobs UI part12 * Fix rebase * Jobs UI part13 * Jobs UI part14 * Jobs UI part15 * changes (#626) * Remove useless ui registration * Remove useless ui registration * move job_list.py replace [0] with get_first() * optimize imports * fix date None problem add test job list * Jobs UI part16 * Jobs UI part17 * Jobs UI part18 * Jobs UI part19 * Jobs UI part20 * Jobs UI part21 * Jobs UI part22 * move job_list_options * Add label to job status checkbox group * changes * changes --------- Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> * Update some dependencies * UI helper to convert 'none' * validate job (#628) * changes * changes * add test * changes * changes * Minor adjustments * Replace is_json with is_file * Handle empty and invalid json in job_list * Handle empty and invalid json in job_list * Handle empty and invalid json in job_list * Work on the job manager UI * Cosmetic changes on common helper * Just make it work for now * Just make it work for now * Just make it work for now * Streamline the step index lookups * Hide footer * Simplify instant runner * Simplify instant runner UI and job manager UI * Fix empty step choices * Fix empty step choices * Fix none values in UI * Rework on benchmark (add warmup) and job list * Improve ValueAndUnit * Add step 1 of x output * Cosmetic changes on the UI * Fix invalid job file names * Update preview * Introducing has_step() and sorting out insert behaviour * Introducing has_step() and sorting out insert behaviour * Add [ none ] to some job id dropdowns * Make updated dropdown values kinda perfect * Make updated dropdown values kinda perfect * Fix testing * Minor improvement on UI * Fix false config lookup * Remove TensorRT as our models are not made for it * Feat/cli commands second try rev2 (#640) * Refactor CLI to commands * Refactor CLI to commands part2 * Refactor CLI to commands part3 * Refactor CLI to commands part4 * Rename everything to facefusion.py * Refactor CLI to commands part5 * Refactor CLI to commands part6 * Adjust testing * Fix lint * Fix lint * Fix lint * Refactor CLI to commands part7 * Extend State typing * Fix false config lookup, adjust logical orders * Move away from passing program part1 * Move away from passing program part2 * Move away from passing program part3 * Fix lint * Move away from passing program part4 * ui-args update * ui-args update * ui-args update * temporary type fix * Move away from passing program part5 * remove unused * creates args.py * Move away from passing program part6 * Move away from passing program part7 --------- Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> * Minor optimizations * Update commands in README * Fix job-retry command * Fix multi runs via UI * add more job keys * Cleanup codebase * One method to create inference session (#641) * One method to create inference session * Remove warnings, as there are none * Remember job id during processing * Fix face masker config block * Change wording * Prevent age modifier from using CoreML * add expression restorer (#642) * add expression restorer * fix import * fix lint * changes * changes * changes * Host the final model for expression restorer * Insert step on the given index * UI workover (#644) * UI workover part1 * Introduce ComponentOptions * Only set Media components to None when visibility changes * Clear static faces and reference faces between step processing * Minor changes * Minor changes * Fix testing * Enable test_sanitize_path_for_windows (#646) * Dynamic download during job processing (#647) * Fix face masker UI * Rename run-headless to headless-run * Feat/split frame processor UI (#649) * Split frame processor UI * Split frame processor UI part3, Refactor get_model_initializer * Split frame processor UI part4 * Feat/rename frame processors (#651) * Rename frame processors * Rename frame processors part2 * Fix imports Conflicts: facefusion/uis/layouts/benchmark.py facefusion/uis/layouts/default.py * Fix imports * Cosmetic changes * Fix multi threading for ROCm * Change temp frames pattern * Adjust terminal help * remove expression restorer (#653) * Expression restorer as processor (#655) * add expression restorer * changes * Cleanup code * Add TensorRT support back * Add TensorRT support back * Add TensorRT support back * changes (#656) * Change minor wording * Fix face enhancer slider * Add more typing * Fix expression-restorer when using trim (#659) * changes * changes * Rework/model and inference pool part2 (#660) * Rework on model and inference pool * Introduce inference sources and pools part1 * Introduce inference sources and pools part2 * Introduce inference sources and pools part3 * Introduce inference sources and pools part4 * Introduce inference sources and pools part5 * Introduce inference sources and pools part6 * Introduce inference sources and pools part6 * Introduce inference sources and pools part6 * Introduce inference sources and pools part7 * Introduce inference sources and pools part7 * Introduce inference sources and pools part8 * Introduce inference sources and pools part9 * Introduce inference sources and pools part10 * Introduce inference sources and pools part11 * Introduce inference sources and pools part11 * Introduce inference sources and pools part11 * Introduce inference sources and pools part12 * Reorganize the face masker UI * Fix trim in UI * Feat/hashed sources (#668) * Introduce source helper * Remove post_check() and just use process_manager * Remove post_check() part2 * Add hash based downloads * Add hash based downloads part2 * Add hash based downloads part3 * Add hash based downloads part4 * Add hash based downloads part5 * Add hash based downloads part6 * Add hash based downloads part7 * Add hash based downloads part7 * Add hash based downloads part8 * Remove print * Prepare 3.0.0 release * Fix UI * Release the check when really done * Update inputs for live portrait * Update to 3.0.0 releases, extend download postfix * Move files to the right place * Logging for the hash and source validation * Changing logic to handle corrupt sources * Fix typo * Use names over get_inputs(), Remove set_options() call * Age modifier now works for CoreML too * Update age_modifier.py * Add video encoder h264_videotoolbox and hevc_videotoolbox * Face editor add eye gaze & remove open factor sliders (#670) * changes * add eye gaze * changes * cleanup * add eyebrow control * changes * changes * Feat/terminal UI (#671) * Introduce terminal to the UI * Introduce terminal to the UI part2 * Introduce terminal to the UI part2 * Introduce terminal to the UI part2 * Calc range step to avoid weird values * Use Sequence for ranges * Use Sequence for ranges * changes (#673) * Use Sequence for ranges * Finalize terminal UI * Finalize terminal UI * Webcam cosmetics, Fix normalize fps to accept int * Cosmetic changes * Finalize terminal UI * Rename leftover typings * Fix wording * Fix rounding in metavar * Fix rounding in metavar * Rename to face classifier * Face editor lip moves (#677) * changes * changes * changes * Fix rounding in metavar * Rename to face classifier * changes * changes * update naming --------- Co-authored-by: henryruhs <info@henryruhs.com> * Fix wording * Feat/many landmarker + face analyser breakdown (#678) * Basic multi landmarker integration * Simplify some method names * Break into face_detector and face_landmarker * Fix cosmetics * Fix testing * Break into face_attributor and face_recognizer * Clear them all * Clear them all * Rename to face classifier * Rename to face classifier * Fix testing * Fix stuff * Add face landmarker model to UI * Add face landmarker model to UI part2 * Split the config * Split the UI * Improvement from code review * Improvement from code review * Validate args also for sub parsers * Remove clear of processors in process step * Allow finder control for the face editor * Fix lint * Improve testing performance * Remove unused file, Clear processors from the UI before job runs * Update the installer * Uniform set handler for swapper and detector in the UI * Fix example urls * Feat/inference manager (#684) * Introduce inference manager * Migrate all to inference manager * clean ini * Introduce app context based inference pools * Fix lint * Fix typing * Adjust layout * Less border radius * Rename app context names * Fix/live portrait directml (#691) * changes (#690) * Adjust naming * Use our assets release * Adjust naming --------- Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> * Add caches to gitignore * Update dependencies and drop CUDA 11.8 support (#693) * Update dependencies and drop CUDA 11.8 support * Play save and keep numpy 1.x.x * Improve TensorRT optimization * changes * changes * changes * changes * changes * changes * changes * changes * changes * Reuse inference sessions (#696) * Fix force-download command * Refactor processors to forward() (#698) * Install tensorrt when selecting cuda * Minor changes * Use latest numpy * Fix limit system memory * Implement forward() for every inference (#699) * Implement forward() for every inference * Implement forward() for every inference * Implement forward() for every inference * Implement forward() for every inference * changes * changes * changes * changes * Feat/fairface (#710) * Replace gender_age model with fair face (#709) * changes * changes * changes * age dropdown to range-slider * Cleanup code * Cleanup code --------- Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> * Extend installer to set library paths for cuda and tensorrt (#707) * Extend installer to set library paths for cuda and tensorrt * Add refresh of conda env * Remove invalid commands * Set the conda env according to operating system * Update for ROCm 6.2 * fix installer * Aktualisieren von installer.py * Add missing face selector keys * Try to keep original LD_LIBRARY_PATH * windows support installer * Final touch to the installer * Remove spaces * Simplidy collect_model_downloads() * Fix force download for once and forever * Housekeeping (#715) * changes * changes * changes * Fix performance part1 * Fix mixed states (#689) * Fix mixed states * Add missing sync for job args * Move UnionStateXXX to base typing * Undo * Remove UnionStateXXX * Fix app context performance lookup (#717) * Restore performance for inswapper * Mover upper() to the logger * Undo debugging * Move TensorRT installation to docs * Sort out log level typing, Add log level UI dropdown (#719) * Fix inference pool part1 * Validate conda library paths existence * Default face selector order to large-small * Fix inference pool context according to execution provider (#720) * Fix app context under Windows * CUDA and TensorRT update for the installer * Remove concept of static processor modules * Revert false commit * Change event order makes a difference * Fix multi model context in inference pool (#721) * Fix multi model context in inference pool * Fix multi model context in inference pool part2 * Use latest gradio to avoid fastapi bug * Rework on the Windows Installer * Use embedding converter (#724) * changes (#723) * Upload models to official assets repo --------- Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> * Rework on the Windows Installer part2 * Resolve subprocess calls (#726) * Experiment * Resolve subprocess calls to cover edge cases like broken PATH * Adjust wording * Simplify code * Rework on the Windows Installer part3 * Rework on the Windows Installer part4 * Numpy fix for older onnxruntime * changes (#729) * Add space * Add MacOS installer * Use favicon * Fix disabled logger * Layout polishing (#731) * Update dependencies, Adjust many face landmarker logic * Cosmetics changes * Should be button * Introduce randomized action button * Fix update of lip syncer and expression restorer * Stop sharing inference session this prevents flushing VRAM * Fix test * Fix urls * Prepare release * Vanish inquirer * Sticky preview does not work on portrait images * Sticky preview only for landscape images and videos * remove gradio tunnel env * Change wording and deeplinks * increase peppa landmark score offset * Change wording * Graceful exit install.py * Just adding a required * Cannot use the exit_helper * Rename our model * Change color of face-landmark-68/5 * Limit liveportrait (#739) * changes * changes * changes * Cleanup * Cleanup --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com> Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> * limit expression restorer * change expression restorer 0-100 range * Use 256x icon * changes * changes * changes * changes * Limit face editor rotation (#745) * changes (#743) * Finish euler methods --------- Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> * Use different coveralls badge * Move about wording * Shorten scope in the logger * changes * changes * Shorten scope in the logger * fix typo * Simplify the arcface converter names * Update preview --------- Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>
272 lines
16 KiB
Python
Executable File
272 lines
16 KiB
Python
Executable File
from typing import List, Optional, Tuple
|
|
|
|
import gradio
|
|
|
|
from facefusion import state_manager, wording
|
|
from facefusion.common_helper import calc_float_step
|
|
from facefusion.processors import choices as processors_choices
|
|
from facefusion.processors.core import load_processor_module
|
|
from facefusion.processors.typing import FaceEditorModel
|
|
from facefusion.uis.core import get_ui_component, register_ui_component
|
|
|
|
FACE_EDITOR_MODEL_DROPDOWN : Optional[gradio.Dropdown] = None
|
|
FACE_EDITOR_EYEBROW_DIRECTION_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_EYE_GAZE_HORIZONTAL_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_EYE_GAZE_VERTICAL_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_EYE_OPEN_RATIO_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_LIP_OPEN_RATIO_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_MOUTH_GRIM_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_MOUTH_POUT_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_MOUTH_PURSE_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_MOUTH_SMILE_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_MOUTH_POSITION_HORIZONTAL_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_MOUTH_POSITION_VERTICAL_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_HEAD_PITCH_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_HEAD_YAW_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_EDITOR_HEAD_ROLL_SLIDER : Optional[gradio.Slider] = None
|
|
|
|
|
|
def render() -> None:
|
|
global FACE_EDITOR_MODEL_DROPDOWN
|
|
global FACE_EDITOR_EYEBROW_DIRECTION_SLIDER
|
|
global FACE_EDITOR_EYE_GAZE_HORIZONTAL_SLIDER
|
|
global FACE_EDITOR_EYE_GAZE_VERTICAL_SLIDER
|
|
global FACE_EDITOR_EYE_OPEN_RATIO_SLIDER
|
|
global FACE_EDITOR_LIP_OPEN_RATIO_SLIDER
|
|
global FACE_EDITOR_MOUTH_GRIM_SLIDER
|
|
global FACE_EDITOR_MOUTH_POUT_SLIDER
|
|
global FACE_EDITOR_MOUTH_PURSE_SLIDER
|
|
global FACE_EDITOR_MOUTH_SMILE_SLIDER
|
|
global FACE_EDITOR_MOUTH_POSITION_HORIZONTAL_SLIDER
|
|
global FACE_EDITOR_MOUTH_POSITION_VERTICAL_SLIDER
|
|
global FACE_EDITOR_HEAD_PITCH_SLIDER
|
|
global FACE_EDITOR_HEAD_YAW_SLIDER
|
|
global FACE_EDITOR_HEAD_ROLL_SLIDER
|
|
|
|
FACE_EDITOR_MODEL_DROPDOWN = gradio.Dropdown(
|
|
label = wording.get('uis.face_editor_model_dropdown'),
|
|
choices = processors_choices.face_editor_models,
|
|
value = state_manager.get_item('face_editor_model'),
|
|
visible = 'face_editor' in state_manager.get_item('processors')
|
|
)
|
|
FACE_EDITOR_EYEBROW_DIRECTION_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_eyebrow_direction_slider'),
|
|
value = state_manager.get_item('face_editor_eyebrow_direction'),
|
|
step = calc_float_step(processors_choices.face_editor_eyebrow_direction_range),
|
|
minimum = processors_choices.face_editor_eyebrow_direction_range[0],
|
|
maximum = processors_choices.face_editor_eyebrow_direction_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_EYE_GAZE_HORIZONTAL_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_eye_gaze_horizontal_slider'),
|
|
value = state_manager.get_item('face_editor_eye_gaze_horizontal'),
|
|
step = calc_float_step(processors_choices.face_editor_eye_gaze_horizontal_range),
|
|
minimum = processors_choices.face_editor_eye_gaze_horizontal_range[0],
|
|
maximum = processors_choices.face_editor_eye_gaze_horizontal_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_EYE_GAZE_VERTICAL_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_eye_gaze_vertical_slider'),
|
|
value = state_manager.get_item('face_editor_eye_gaze_vertical'),
|
|
step = calc_float_step(processors_choices.face_editor_eye_gaze_vertical_range),
|
|
minimum = processors_choices.face_editor_eye_gaze_vertical_range[0],
|
|
maximum = processors_choices.face_editor_eye_gaze_vertical_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_EYE_OPEN_RATIO_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_eye_open_ratio_slider'),
|
|
value = state_manager.get_item('face_editor_eye_open_ratio'),
|
|
step = calc_float_step(processors_choices.face_editor_eye_open_ratio_range),
|
|
minimum = processors_choices.face_editor_eye_open_ratio_range[0],
|
|
maximum = processors_choices.face_editor_eye_open_ratio_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_LIP_OPEN_RATIO_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_lip_open_ratio_slider'),
|
|
value = state_manager.get_item('face_editor_lip_open_ratio'),
|
|
step = calc_float_step(processors_choices.face_editor_lip_open_ratio_range),
|
|
minimum = processors_choices.face_editor_lip_open_ratio_range[0],
|
|
maximum = processors_choices.face_editor_lip_open_ratio_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_MOUTH_GRIM_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_mouth_grim_slider'),
|
|
value = state_manager.get_item('face_editor_mouth_grim'),
|
|
step = calc_float_step(processors_choices.face_editor_mouth_grim_range),
|
|
minimum = processors_choices.face_editor_mouth_grim_range[0],
|
|
maximum = processors_choices.face_editor_mouth_grim_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_MOUTH_POUT_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_mouth_pout_slider'),
|
|
value = state_manager.get_item('face_editor_mouth_pout'),
|
|
step = calc_float_step(processors_choices.face_editor_mouth_pout_range),
|
|
minimum = processors_choices.face_editor_mouth_pout_range[0],
|
|
maximum = processors_choices.face_editor_mouth_pout_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_MOUTH_PURSE_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_mouth_purse_slider'),
|
|
value = state_manager.get_item('face_editor_mouth_purse'),
|
|
step = calc_float_step(processors_choices.face_editor_mouth_purse_range),
|
|
minimum = processors_choices.face_editor_mouth_purse_range[0],
|
|
maximum = processors_choices.face_editor_mouth_purse_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_MOUTH_SMILE_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_mouth_smile_slider'),
|
|
value = state_manager.get_item('face_editor_mouth_smile'),
|
|
step = calc_float_step(processors_choices.face_editor_mouth_smile_range),
|
|
minimum = processors_choices.face_editor_mouth_smile_range[0],
|
|
maximum = processors_choices.face_editor_mouth_smile_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_MOUTH_POSITION_HORIZONTAL_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_mouth_position_horizontal_slider'),
|
|
value = state_manager.get_item('face_editor_mouth_position_horizontal'),
|
|
step = calc_float_step(processors_choices.face_editor_mouth_position_horizontal_range),
|
|
minimum = processors_choices.face_editor_mouth_position_horizontal_range[0],
|
|
maximum = processors_choices.face_editor_mouth_position_horizontal_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_MOUTH_POSITION_VERTICAL_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_mouth_position_vertical_slider'),
|
|
value = state_manager.get_item('face_editor_mouth_position_vertical'),
|
|
step = calc_float_step(processors_choices.face_editor_mouth_position_vertical_range),
|
|
minimum = processors_choices.face_editor_mouth_position_vertical_range[0],
|
|
maximum = processors_choices.face_editor_mouth_position_vertical_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_HEAD_PITCH_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_head_pitch_slider'),
|
|
value = state_manager.get_item('face_editor_head_pitch'),
|
|
step = calc_float_step(processors_choices.face_editor_head_pitch_range),
|
|
minimum = processors_choices.face_editor_head_pitch_range[0],
|
|
maximum = processors_choices.face_editor_head_pitch_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_HEAD_YAW_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_head_yaw_slider'),
|
|
value = state_manager.get_item('face_editor_head_yaw'),
|
|
step = calc_float_step(processors_choices.face_editor_head_yaw_range),
|
|
minimum = processors_choices.face_editor_head_yaw_range[0],
|
|
maximum = processors_choices.face_editor_head_yaw_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
FACE_EDITOR_HEAD_ROLL_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_editor_head_roll_slider'),
|
|
value = state_manager.get_item('face_editor_head_roll'),
|
|
step = calc_float_step(processors_choices.face_editor_head_roll_range),
|
|
minimum = processors_choices.face_editor_head_roll_range[0],
|
|
maximum = processors_choices.face_editor_head_roll_range[-1],
|
|
visible = 'face_editor' in state_manager.get_item('processors'),
|
|
)
|
|
register_ui_component('face_editor_model_dropdown', FACE_EDITOR_MODEL_DROPDOWN)
|
|
register_ui_component('face_editor_eyebrow_direction_slider', FACE_EDITOR_EYEBROW_DIRECTION_SLIDER)
|
|
register_ui_component('face_editor_eye_gaze_horizontal_slider', FACE_EDITOR_EYE_GAZE_HORIZONTAL_SLIDER)
|
|
register_ui_component('face_editor_eye_gaze_vertical_slider', FACE_EDITOR_EYE_GAZE_VERTICAL_SLIDER)
|
|
register_ui_component('face_editor_eye_open_ratio_slider', FACE_EDITOR_EYE_OPEN_RATIO_SLIDER)
|
|
register_ui_component('face_editor_lip_open_ratio_slider', FACE_EDITOR_LIP_OPEN_RATIO_SLIDER)
|
|
register_ui_component('face_editor_mouth_grim_slider', FACE_EDITOR_MOUTH_GRIM_SLIDER)
|
|
register_ui_component('face_editor_mouth_pout_slider', FACE_EDITOR_MOUTH_POUT_SLIDER)
|
|
register_ui_component('face_editor_mouth_purse_slider', FACE_EDITOR_MOUTH_PURSE_SLIDER)
|
|
register_ui_component('face_editor_mouth_smile_slider', FACE_EDITOR_MOUTH_SMILE_SLIDER)
|
|
register_ui_component('face_editor_mouth_position_horizontal_slider', FACE_EDITOR_MOUTH_POSITION_HORIZONTAL_SLIDER)
|
|
register_ui_component('face_editor_mouth_position_vertical_slider', FACE_EDITOR_MOUTH_POSITION_VERTICAL_SLIDER)
|
|
register_ui_component('face_editor_head_pitch_slider', FACE_EDITOR_HEAD_PITCH_SLIDER)
|
|
register_ui_component('face_editor_head_yaw_slider', FACE_EDITOR_HEAD_YAW_SLIDER)
|
|
register_ui_component('face_editor_head_roll_slider', FACE_EDITOR_HEAD_ROLL_SLIDER)
|
|
|
|
|
|
def listen() -> None:
|
|
FACE_EDITOR_MODEL_DROPDOWN.change(update_face_editor_model, inputs = FACE_EDITOR_MODEL_DROPDOWN, outputs = FACE_EDITOR_MODEL_DROPDOWN)
|
|
FACE_EDITOR_EYEBROW_DIRECTION_SLIDER.release(update_face_editor_eyebrow_direction, inputs = FACE_EDITOR_EYEBROW_DIRECTION_SLIDER)
|
|
FACE_EDITOR_EYE_GAZE_HORIZONTAL_SLIDER.release(update_face_editor_eye_gaze_horizontal, inputs = FACE_EDITOR_EYE_GAZE_HORIZONTAL_SLIDER)
|
|
FACE_EDITOR_EYE_GAZE_VERTICAL_SLIDER.release(update_face_editor_eye_gaze_vertical, inputs = FACE_EDITOR_EYE_GAZE_VERTICAL_SLIDER)
|
|
FACE_EDITOR_EYE_OPEN_RATIO_SLIDER.release(update_face_editor_eye_open_ratio, inputs = FACE_EDITOR_EYE_OPEN_RATIO_SLIDER)
|
|
FACE_EDITOR_LIP_OPEN_RATIO_SLIDER.release(update_face_editor_lip_open_ratio, inputs = FACE_EDITOR_LIP_OPEN_RATIO_SLIDER)
|
|
FACE_EDITOR_MOUTH_GRIM_SLIDER.release(update_face_editor_mouth_grim, inputs = FACE_EDITOR_MOUTH_GRIM_SLIDER)
|
|
FACE_EDITOR_MOUTH_POUT_SLIDER.release(update_face_editor_mouth_pout, inputs = FACE_EDITOR_MOUTH_POUT_SLIDER)
|
|
FACE_EDITOR_MOUTH_PURSE_SLIDER.release(update_face_editor_mouth_purse, inputs = FACE_EDITOR_MOUTH_PURSE_SLIDER)
|
|
FACE_EDITOR_MOUTH_SMILE_SLIDER.release(update_face_editor_mouth_smile, inputs = FACE_EDITOR_MOUTH_SMILE_SLIDER)
|
|
FACE_EDITOR_MOUTH_POSITION_HORIZONTAL_SLIDER.release(update_face_editor_mouth_position_horizontal, inputs = FACE_EDITOR_MOUTH_POSITION_HORIZONTAL_SLIDER)
|
|
FACE_EDITOR_MOUTH_POSITION_VERTICAL_SLIDER.release(update_face_editor_mouth_position_vertical, inputs = FACE_EDITOR_MOUTH_POSITION_VERTICAL_SLIDER)
|
|
FACE_EDITOR_HEAD_PITCH_SLIDER.release(update_face_editor_head_pitch, inputs = FACE_EDITOR_HEAD_PITCH_SLIDER)
|
|
FACE_EDITOR_HEAD_YAW_SLIDER.release(update_face_editor_head_yaw, inputs = FACE_EDITOR_HEAD_YAW_SLIDER)
|
|
FACE_EDITOR_HEAD_ROLL_SLIDER.release(update_face_editor_head_roll, inputs = FACE_EDITOR_HEAD_ROLL_SLIDER)
|
|
|
|
processors_checkbox_group = get_ui_component('processors_checkbox_group')
|
|
if processors_checkbox_group:
|
|
processors_checkbox_group.change(remote_update, inputs = processors_checkbox_group, outputs = [FACE_EDITOR_MODEL_DROPDOWN, FACE_EDITOR_EYEBROW_DIRECTION_SLIDER, FACE_EDITOR_EYE_GAZE_HORIZONTAL_SLIDER, FACE_EDITOR_EYE_GAZE_VERTICAL_SLIDER, FACE_EDITOR_EYE_OPEN_RATIO_SLIDER, FACE_EDITOR_LIP_OPEN_RATIO_SLIDER, FACE_EDITOR_MOUTH_GRIM_SLIDER, FACE_EDITOR_MOUTH_POUT_SLIDER, FACE_EDITOR_MOUTH_PURSE_SLIDER, FACE_EDITOR_MOUTH_SMILE_SLIDER, FACE_EDITOR_MOUTH_POSITION_HORIZONTAL_SLIDER, FACE_EDITOR_MOUTH_POSITION_VERTICAL_SLIDER, FACE_EDITOR_HEAD_PITCH_SLIDER, FACE_EDITOR_HEAD_YAW_SLIDER, FACE_EDITOR_HEAD_ROLL_SLIDER])
|
|
|
|
|
|
def remote_update(processors : List[str]) -> Tuple[gradio.Dropdown, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider, gradio.Slider]:
|
|
has_face_editor = 'face_editor' in processors
|
|
return gradio.Dropdown(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor), gradio.Slider(visible = has_face_editor)
|
|
|
|
|
|
def update_face_editor_model(face_editor_model : FaceEditorModel) -> gradio.Dropdown:
|
|
face_editor_module = load_processor_module('face_editor')
|
|
face_editor_module.clear_inference_pool()
|
|
state_manager.set_item('face_editor_model', face_editor_model)
|
|
|
|
if face_editor_module.pre_check():
|
|
return gradio.Dropdown(value = state_manager.get_item('face_editor_model'))
|
|
return gradio.Dropdown()
|
|
|
|
|
|
def update_face_editor_eyebrow_direction(face_editor_eyebrow_direction : float) -> None:
|
|
state_manager.set_item('face_editor_eyebrow_direction', face_editor_eyebrow_direction)
|
|
|
|
|
|
def update_face_editor_eye_gaze_horizontal(face_editor_eye_gaze_horizontal : float) -> None:
|
|
state_manager.set_item('face_editor_eye_gaze_horizontal', face_editor_eye_gaze_horizontal)
|
|
|
|
|
|
def update_face_editor_eye_gaze_vertical(face_editor_eye_gaze_vertical : float) -> None:
|
|
state_manager.set_item('face_editor_eye_gaze_vertical', face_editor_eye_gaze_vertical)
|
|
|
|
|
|
def update_face_editor_eye_open_ratio(face_editor_eye_open_ratio : float) -> None:
|
|
state_manager.set_item('face_editor_eye_open_ratio', face_editor_eye_open_ratio)
|
|
|
|
|
|
def update_face_editor_lip_open_ratio(face_editor_lip_open_ratio : float) -> None:
|
|
state_manager.set_item('face_editor_lip_open_ratio', face_editor_lip_open_ratio)
|
|
|
|
|
|
def update_face_editor_mouth_grim(face_editor_mouth_grim : float) -> None:
|
|
state_manager.set_item('face_editor_mouth_grim', face_editor_mouth_grim)
|
|
|
|
|
|
def update_face_editor_mouth_pout(face_editor_mouth_pout : float) -> None:
|
|
state_manager.set_item('face_editor_mouth_pout', face_editor_mouth_pout)
|
|
|
|
|
|
def update_face_editor_mouth_purse(face_editor_mouth_purse : float) -> None:
|
|
state_manager.set_item('face_editor_mouth_purse', face_editor_mouth_purse)
|
|
|
|
|
|
def update_face_editor_mouth_smile(face_editor_mouth_smile : float) -> None:
|
|
state_manager.set_item('face_editor_mouth_smile', face_editor_mouth_smile)
|
|
|
|
|
|
def update_face_editor_mouth_position_horizontal(face_editor_mouth_position_horizontal : float) -> None:
|
|
state_manager.set_item('face_editor_mouth_position_horizontal', face_editor_mouth_position_horizontal)
|
|
|
|
|
|
def update_face_editor_mouth_position_vertical(face_editor_mouth_position_vertical : float) -> None:
|
|
state_manager.set_item('face_editor_mouth_position_vertical', face_editor_mouth_position_vertical)
|
|
|
|
|
|
def update_face_editor_head_pitch(face_editor_head_pitch : float) -> None:
|
|
state_manager.set_item('face_editor_head_pitch', face_editor_head_pitch)
|
|
|
|
|
|
def update_face_editor_head_yaw(face_editor_head_yaw : float) -> None:
|
|
state_manager.set_item('face_editor_head_yaw', face_editor_head_yaw)
|
|
|
|
|
|
def update_face_editor_head_roll(face_editor_head_roll : float) -> None:
|
|
state_manager.set_item('face_editor_head_roll', face_editor_head_roll)
|