7a09479fb5
* Replace audio whenever set via source * add H264_qsv&HEVC_qsv (#768) * Update ffmpeg.py * Update choices.py * Update typing.py * Fix spaces and newlines * Fix return type * Introduce hififace swapper * Disable stream for expression restorer * Webcam polishing part1 (#796) * Cosmetics on ignore comments * Testing for replace audio * Testing for restore audio * Testing for restore audio * Fix replace_audio() * Remove shortest and use fixed video duration * Remove shortest and use fixed video duration * Prevent duplicate entries to local PATH * Do hard exit on invalid args * Need for Python 3.10 * Fix state of face selector * Fix OpenVINO by aliasing GPU.0 to GPU * Fix OpenVINO by aliasing GPU.0 to GPU * Fix/age modifier styleganex 512 (#798) * fix * styleganex template * changes * changes * fix occlusion mask * add age modifier scale * change * change * hardcode * Cleanup * Use model_sizes and model_templates variables * No need for prepare when just 2 lines of code * Someone used spaces over tabs * Revert back [0][0] --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com> * Feat/update gradio5 (#799) * Update to Gradio 5 * Remove overrides for Gradio * Fix dark mode for Gradio * Polish errors * More styles for tabs and co * Make slider inputs and reset like a unit * Make slider inputs and reset like a unit * Adjust naming * Improved color matching (#800) * aura fix * fix import * move to vision.py * changes * changes * changes * changes * further reduction * add test * better test * change name * Minor cleanup * Minor cleanup * Minor cleanup * changes (#801) * Switch to official assets repo * Add __pycache__ to gitignore * Gradio pinned python-multipart to 0.0.12 * Update dependencies * Feat/temp path second try (#802) * Terminate base directory from temp helper * Partial adjust program codebase * Move arguments around * Make `-j` absolete * Resolve args * Fix job register keys * Adjust date test * Finalize temp path * Update onnxruntime * Update dependencies * Adjust color for checkboxes * Revert due terrible performance * Fix/enforce vp9 for webm (#805) * Simple fix to enforce vp9 for webm * Remove suggest methods from program helper * Cleanup ffmpeg.py a bit * Update onnxruntime (second try) * Update onnxruntime (second try) * Remove cudnn_conv_algo_search tweaks * Remove cudnn_conv_algo_search tweaks * changes * add both mask instead of multiply * adaptive color correction * changes * remove model size requirement * changes * add to facefusion.ini * changes * changes * changes * Add namespace for dfm creators * Release five frame enhancer models * Remove vendor from model name * Remove vendor from model name * changes * changes * changes * changes * Feat/download providers (#809) * Introduce download providers * update processors download method * add ui * Fix CI * Adjust UI component order, Use download resolver for benchmark * Remove is_download_done() * Introduce download provider set, Remove choices method from execution, cast all dict keys() via list() * Fix spacing --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com> * Fix model paths for 3.1.0 * Introduce bulk-run (#810) * Introduce bulk-run * Make bulk run bullet proof * Integration test for bulk-run * new alignment * Add safer global named resolve_file_pattern() (#811) * Allow bulk runner with target pattern only * changes * changes * Update Python to 3.12 for CI (#813) * changes * Improve NVIDIA device lookups * Rename template key to deepfacelive * Fix name * Improve resolve download * Rename bulk-run to batch-run * Make deep swapper inputs universal * Add more deepfacelive models * Use different morph value * Feat/simplify hashes sources download (#814) * Extract download directory path from assets path * Fix lint * Fix force-download command, Fix urls in frame enhancer * changes * fix warp_face_by_bounding_box dtype error * DFM Morph (#816) * changes * Improve wording, Replace [None], SideQuest: clean forward() of age modifier * SideQuest: clean forward() of face enhancer --------- Co-authored-by: henryruhs <info@henryruhs.com> * Fix preview refresh after slide * Add more deepfacelive models (#817) * Add more deepfacelive models * Add more deepfacelive models * Fix deep swapper sizes * Kill accent colors, Number input styles for Chrome * Simplify thumbnail-item looks * Fix first black screen * Introduce model helper * ci.yml: Add macOS on ARM64 to the testing (#818) * ci.yml: Add macOS on ARM64 to the testing * ci.yml: uses: AnimMouse/setup-ffmpeg@v1 * ci.yml: strategy: matrix: os: macos-latest, * - name: Set up FFmpeg * Update .github/workflows/ci.yml * Update ci.yml --------- Co-authored-by: Henry Ruhs <info@henryruhs.com> * Show/hide morph slider for deep swapper (#822) * remove dfl_head and update dfl_whole_face template * Add deep swapper models by Mats * Add deep swapper models by Druuzil * Add deep swapper models by Rumateus * Implement face enhancer weight for codeformer, Side Quest: has proces… (#823) * Implement face enhancer weight for codeformer, Side Quest: has processor checks * Fix typo * Fix face enhancer blend in UI * Use static model set creation * Add deep swapper models by Jen * Introduce create_static_model_set() everywhere (#824) * Move clear over to the UI (#825) * Fix model key * Undo restore_audio() * Switch to latest XSeg * Switch to latest XSeg * Switch to latest XSeg * Use resolve_download_url() everywhere, Vanish --skip-download flag * Fix resolve_download_url * Fix space * Kill resolve_execution_provider_keys() and move fallbacks where they belong * Kill resolve_execution_provider_keys() and move fallbacks where they belong * Remove as this does not work * Change TempFrameFormat order * Fix CoreML partially * Remove duplicates (Rumateus is the creator) * Add deep swapper models by Edel * Introduce download scopes (#826) * Introduce download scopes * Limit download scopes to force-download command * Change source-paths behaviour * Fix space * Update README * Rename create_log_level_program to create_misc_program * Fix wording * Fix wording * Update dependencies * Use tolerant for video_memory_strategy in benchmark * Feat/ffmpeg with progress (#827) * FFmpeg with progress bar * Fix typing * FFmpeg with progress bar part2 * Restore streaming wording * Change order in choices and typing * Introduce File using list_directory() (#830) * Feat/local deep swapper models (#832) * Local model support for deep swapper * Local model support for deep swapper part2 * Local model support for deep swapper part3 * Update yet another dfm by Druuzil * Refactor/choices and naming (#833) * Refactor choices, imports and naming * Refactor choices, imports and naming * Fix styles for tabs, Restore toast * Update yet another dfm by Druuzil * Feat/face masker models (#834) * Introduce face masker models * Introduce face masker models * Introduce face masker models * Register needed step keys * Provide different XSeg models * Simplify model context * Fix out of range for trim frame, Fix ffmpeg extraction count (#836) * Fix out of range for trim frame, Fix ffmpeg extraction count * Move restrict of trim frame to the core, Make sure all values are within the range * Fix and merge testing * Fix typing * Add region mask for deep swapper * Adjust wording * Move FACE_MASK_REGIONS to choices * Update dependencies * Feat/download provider fallback (#837) * Introduce download providers fallback, Use CURL everywhre * Fix CI * Use readlines() over readline() to avoid while * Use readlines() over readline() to avoid while * Use readlines() over readline() to avoid while * Use communicate() over wait() * Minor updates for testing * Stop webcam on source image change * Feat/webcam improvements (#838) * Detect available webcams * Fix CI, Move webcam id dropdown to the sidebar, Disable warnings * Fix CI * Remove signal on hard_exit() to prevent exceptions * Fix border color in toast timer * Prepare release * Update preview * Update preview * Hotfix progress bar --------- Co-authored-by: DDXDB <38449595+DDXDB@users.noreply.github.com> Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com> Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com> Co-authored-by: Christian Clauss <cclauss@me.com>
273 lines
16 KiB
Python
Executable File
273 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
|
|
|
|
has_face_editor = 'face_editor' in state_manager.get_item('processors')
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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 = has_face_editor
|
|
)
|
|
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)
|