vnyash
120 lines
6.1 KiB
Python
Executable File
120 lines
6.1 KiB
Python
Executable File
from typing import Optional, Tuple, List
|
|
import gradio
|
|
|
|
import deepfuze.globals
|
|
import deepfuze.choices
|
|
from deepfuze import wording
|
|
from deepfuze.typing import FaceMaskType, FaceMaskRegion
|
|
from deepfuze.uis.core import register_ui_component
|
|
|
|
FACE_MASK_TYPES_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None
|
|
FACE_MASK_BLUR_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_MASK_BOX_GROUP : Optional[gradio.Group] = None
|
|
FACE_MASK_REGION_GROUP : Optional[gradio.Group] = None
|
|
FACE_MASK_PADDING_TOP_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_MASK_PADDING_RIGHT_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_MASK_PADDING_BOTTOM_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_MASK_PADDING_LEFT_SLIDER : Optional[gradio.Slider] = None
|
|
FACE_MASK_REGION_CHECKBOX_GROUP : Optional[gradio.CheckboxGroup] = None
|
|
|
|
|
|
def render() -> None:
|
|
global FACE_MASK_TYPES_CHECKBOX_GROUP
|
|
global FACE_MASK_BLUR_SLIDER
|
|
global FACE_MASK_BOX_GROUP
|
|
global FACE_MASK_REGION_GROUP
|
|
global FACE_MASK_PADDING_TOP_SLIDER
|
|
global FACE_MASK_PADDING_RIGHT_SLIDER
|
|
global FACE_MASK_PADDING_BOTTOM_SLIDER
|
|
global FACE_MASK_PADDING_LEFT_SLIDER
|
|
global FACE_MASK_REGION_CHECKBOX_GROUP
|
|
|
|
has_box_mask = 'box' in deepfuze.globals.face_mask_types
|
|
has_region_mask = 'region' in deepfuze.globals.face_mask_types
|
|
FACE_MASK_TYPES_CHECKBOX_GROUP = gradio.CheckboxGroup(
|
|
label = wording.get('uis.face_mask_types_checkbox_group'),
|
|
choices = deepfuze.choices.face_mask_types,
|
|
value = deepfuze.globals.face_mask_types
|
|
)
|
|
with gradio.Group(visible = has_box_mask) as FACE_MASK_BOX_GROUP:
|
|
FACE_MASK_BLUR_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_mask_blur_slider'),
|
|
step = deepfuze.choices.face_mask_blur_range[1] - deepfuze.choices.face_mask_blur_range[0],
|
|
minimum = deepfuze.choices.face_mask_blur_range[0],
|
|
maximum = deepfuze.choices.face_mask_blur_range[-1],
|
|
value = deepfuze.globals.face_mask_blur
|
|
)
|
|
with gradio.Row():
|
|
FACE_MASK_PADDING_TOP_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_mask_padding_top_slider'),
|
|
step = deepfuze.choices.face_mask_padding_range[1] - deepfuze.choices.face_mask_padding_range[0],
|
|
minimum = deepfuze.choices.face_mask_padding_range[0],
|
|
maximum = deepfuze.choices.face_mask_padding_range[-1],
|
|
value = deepfuze.globals.face_mask_padding[0]
|
|
)
|
|
FACE_MASK_PADDING_RIGHT_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_mask_padding_right_slider'),
|
|
step = deepfuze.choices.face_mask_padding_range[1] - deepfuze.choices.face_mask_padding_range[0],
|
|
minimum = deepfuze.choices.face_mask_padding_range[0],
|
|
maximum = deepfuze.choices.face_mask_padding_range[-1],
|
|
value = deepfuze.globals.face_mask_padding[1]
|
|
)
|
|
with gradio.Row():
|
|
FACE_MASK_PADDING_BOTTOM_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_mask_padding_bottom_slider'),
|
|
step = deepfuze.choices.face_mask_padding_range[1] - deepfuze.choices.face_mask_padding_range[0],
|
|
minimum = deepfuze.choices.face_mask_padding_range[0],
|
|
maximum = deepfuze.choices.face_mask_padding_range[-1],
|
|
value = deepfuze.globals.face_mask_padding[2]
|
|
)
|
|
FACE_MASK_PADDING_LEFT_SLIDER = gradio.Slider(
|
|
label = wording.get('uis.face_mask_padding_left_slider'),
|
|
step = deepfuze.choices.face_mask_padding_range[1] - deepfuze.choices.face_mask_padding_range[0],
|
|
minimum = deepfuze.choices.face_mask_padding_range[0],
|
|
maximum = deepfuze.choices.face_mask_padding_range[-1],
|
|
value = deepfuze.globals.face_mask_padding[3]
|
|
)
|
|
with gradio.Row():
|
|
FACE_MASK_REGION_CHECKBOX_GROUP = gradio.CheckboxGroup(
|
|
label = wording.get('uis.face_mask_region_checkbox_group'),
|
|
choices = deepfuze.choices.face_mask_regions,
|
|
value = deepfuze.globals.face_mask_regions,
|
|
visible = has_region_mask
|
|
)
|
|
register_ui_component('face_mask_types_checkbox_group', FACE_MASK_TYPES_CHECKBOX_GROUP)
|
|
register_ui_component('face_mask_blur_slider', FACE_MASK_BLUR_SLIDER)
|
|
register_ui_component('face_mask_padding_top_slider', FACE_MASK_PADDING_TOP_SLIDER)
|
|
register_ui_component('face_mask_padding_right_slider', FACE_MASK_PADDING_RIGHT_SLIDER)
|
|
register_ui_component('face_mask_padding_bottom_slider', FACE_MASK_PADDING_BOTTOM_SLIDER)
|
|
register_ui_component('face_mask_padding_left_slider', FACE_MASK_PADDING_LEFT_SLIDER)
|
|
register_ui_component('face_mask_region_checkbox_group', FACE_MASK_REGION_CHECKBOX_GROUP)
|
|
|
|
|
|
def listen() -> None:
|
|
FACE_MASK_TYPES_CHECKBOX_GROUP.change(update_face_mask_type, inputs = FACE_MASK_TYPES_CHECKBOX_GROUP, outputs = [ FACE_MASK_TYPES_CHECKBOX_GROUP, FACE_MASK_BOX_GROUP, FACE_MASK_REGION_CHECKBOX_GROUP ])
|
|
FACE_MASK_BLUR_SLIDER.release(update_face_mask_blur, inputs = FACE_MASK_BLUR_SLIDER)
|
|
FACE_MASK_REGION_CHECKBOX_GROUP.change(update_face_mask_regions, inputs = FACE_MASK_REGION_CHECKBOX_GROUP, outputs = FACE_MASK_REGION_CHECKBOX_GROUP)
|
|
face_mask_padding_sliders = [ FACE_MASK_PADDING_TOP_SLIDER, FACE_MASK_PADDING_RIGHT_SLIDER, FACE_MASK_PADDING_BOTTOM_SLIDER, FACE_MASK_PADDING_LEFT_SLIDER ]
|
|
for face_mask_padding_slider in face_mask_padding_sliders:
|
|
face_mask_padding_slider.release(update_face_mask_padding, inputs = face_mask_padding_sliders)
|
|
|
|
|
|
def update_face_mask_type(face_mask_types : List[FaceMaskType]) -> Tuple[gradio.CheckboxGroup, gradio.Group, gradio.CheckboxGroup]:
|
|
deepfuze.globals.face_mask_types = face_mask_types or deepfuze.choices.face_mask_types
|
|
has_box_mask = 'box' in face_mask_types
|
|
has_region_mask = 'region' in face_mask_types
|
|
return gradio.CheckboxGroup(value = deepfuze.globals.face_mask_types), gradio.Group(visible = has_box_mask), gradio.CheckboxGroup(visible = has_region_mask)
|
|
|
|
|
|
def update_face_mask_blur(face_mask_blur : float) -> None:
|
|
deepfuze.globals.face_mask_blur = face_mask_blur
|
|
|
|
|
|
def update_face_mask_padding(face_mask_padding_top : int, face_mask_padding_right : int, face_mask_padding_bottom : int, face_mask_padding_left : int) -> None:
|
|
deepfuze.globals.face_mask_padding = (face_mask_padding_top, face_mask_padding_right, face_mask_padding_bottom, face_mask_padding_left)
|
|
|
|
|
|
def update_face_mask_regions(face_mask_regions : List[FaceMaskRegion]) -> gradio.CheckboxGroup:
|
|
deepfuze.globals.face_mask_regions = face_mask_regions or deepfuze.choices.face_mask_regions
|
|
return gradio.CheckboxGroup(value = deepfuze.globals.face_mask_regions)
|