mirror of
https://github.com/Zeyi-Lin/HivisionIDPhotos.git
synced 2026-08-28 19:45:12 +08:00
Ze-Yi LIN
753 lines
32 KiB
Python
753 lines
32 KiB
Python
import gradio as gr
|
|
import os
|
|
import pathlib
|
|
from demo.locales import LOCALES
|
|
from demo.processor import IDPhotoProcessor
|
|
|
|
"""
|
|
只裁切模式:
|
|
1. 如果重新上传了照片，然后点击按钮，第一次会调用不裁切的模式，第二次会调用裁切的模式
|
|
"""
|
|
|
|
|
|
def load_description(fp):
|
|
with open(fp, "r", encoding="utf-8") as f:
|
|
content = f.read()
|
|
return content
|
|
|
|
|
|
def create_ui(
|
|
processor: IDPhotoProcessor,
|
|
root_dir: str,
|
|
human_matting_models: list,
|
|
face_detect_models: list,
|
|
language: list,
|
|
):
|
|
|
|
# 加载环境变量DEFAULT_LANG, 如果有且在language中，则将DEFAULT_LANG设置为环境变量
|
|
if "DEFAULT_LANG" in os.environ and os.environ["DEFAULT_LANG"] in language:
|
|
DEFAULT_LANG = os.environ["DEFAULT_LANG"]
|
|
else:
|
|
DEFAULT_LANG = language[0]
|
|
|
|
DEFAULT_HUMAN_MATTING_MODEL = "modnet_photographic_portrait_matting"
|
|
DEFAULT_FACE_DETECT_MODEL = "retinaface-resnet50"
|
|
|
|
if DEFAULT_HUMAN_MATTING_MODEL in human_matting_models:
|
|
human_matting_models.remove(DEFAULT_HUMAN_MATTING_MODEL)
|
|
human_matting_models.insert(0, DEFAULT_HUMAN_MATTING_MODEL)
|
|
|
|
if DEFAULT_FACE_DETECT_MODEL not in face_detect_models:
|
|
DEFAULT_FACE_DETECT_MODEL = "mtcnn"
|
|
|
|
demo = gr.Blocks(title="HivisionIDPhotos")
|
|
|
|
with demo:
|
|
gr.HTML(load_description(os.path.join(root_dir, "demo/assets/title.md")))
|
|
with gr.Row():
|
|
# ------------------------ 左半边 UI ------------------------
|
|
with gr.Column():
|
|
img_input = gr.Image(height=400)
|
|
|
|
with gr.Row():
|
|
# 语言选择器
|
|
language_options = gr.Dropdown(
|
|
choices=language,
|
|
label="Language",
|
|
value=DEFAULT_LANG,
|
|
)
|
|
|
|
face_detect_model_options = gr.Dropdown(
|
|
choices=face_detect_models,
|
|
label=LOCALES["face_model"][DEFAULT_LANG]["label"],
|
|
value=DEFAULT_FACE_DETECT_MODEL,
|
|
)
|
|
|
|
matting_model_options = gr.Dropdown(
|
|
choices=human_matting_models,
|
|
label=LOCALES["matting_model"][DEFAULT_LANG]["label"],
|
|
value=human_matting_models[0],
|
|
)
|
|
|
|
# TAB1 - 关键参数 ------------------------------------------------
|
|
with gr.Tab(
|
|
LOCALES["key_param"][DEFAULT_LANG]["label"]
|
|
) as key_parameter_tab:
|
|
# 尺寸模式
|
|
with gr.Row():
|
|
mode_options = gr.Radio(
|
|
choices=LOCALES["size_mode"][DEFAULT_LANG]["choices"],
|
|
label=LOCALES["size_mode"][DEFAULT_LANG]["label"],
|
|
value=LOCALES["size_mode"][DEFAULT_LANG]["choices"][0],
|
|
min_width=520,
|
|
)
|
|
|
|
# 尺寸列表
|
|
with gr.Row(visible=True) as size_list_row:
|
|
size_list_options = gr.Dropdown(
|
|
choices=LOCALES["size_list"][DEFAULT_LANG]["choices"],
|
|
label=LOCALES["size_list"][DEFAULT_LANG]["label"],
|
|
value=LOCALES["size_list"][DEFAULT_LANG]["choices"][0],
|
|
elem_id="size_list",
|
|
)
|
|
# 自定义尺寸px
|
|
with gr.Row(visible=False) as custom_size_px:
|
|
custom_size_height_px = gr.Number(
|
|
value=413,
|
|
label=LOCALES["custom_size_px"][DEFAULT_LANG]["height"],
|
|
interactive=True,
|
|
)
|
|
custom_size_width_px = gr.Number(
|
|
value=295,
|
|
label=LOCALES["custom_size_px"][DEFAULT_LANG]["width"],
|
|
interactive=True,
|
|
)
|
|
# 自定义尺寸mm
|
|
with gr.Row(visible=False) as custom_size_mm:
|
|
custom_size_height_mm = gr.Number(
|
|
value=35,
|
|
label=LOCALES["custom_size_mm"][DEFAULT_LANG]["height"],
|
|
interactive=True,
|
|
)
|
|
custom_size_width_mm = gr.Number(
|
|
value=25,
|
|
label=LOCALES["custom_size_mm"][DEFAULT_LANG]["width"],
|
|
interactive=True,
|
|
)
|
|
|
|
# 背景颜色
|
|
color_options = gr.Radio(
|
|
choices=LOCALES["bg_color"][DEFAULT_LANG]["choices"],
|
|
label=LOCALES["bg_color"][DEFAULT_LANG]["label"],
|
|
value=LOCALES["bg_color"][DEFAULT_LANG]["choices"][0],
|
|
)
|
|
|
|
# 自定义颜色RGB
|
|
with gr.Row(visible=False) as custom_color_rgb:
|
|
custom_color_R = gr.Number(value=0, label="R", minimum=0, maximum=255, interactive=True)
|
|
custom_color_G = gr.Number(value=0, label="G", minimum=0, maximum=255, interactive=True)
|
|
custom_color_B = gr.Number(value=0, label="B", minimum=0, maximum=255, interactive=True)
|
|
|
|
# 自定义颜色HEX
|
|
with gr.Row(visible=False) as custom_color_hex:
|
|
custom_color_hex_value = gr.Text(value="000000", label="Hex", interactive=True)
|
|
|
|
# 渲染模式
|
|
render_options = gr.Radio(
|
|
choices=LOCALES["render_mode"][DEFAULT_LANG]["choices"],
|
|
label=LOCALES["render_mode"][DEFAULT_LANG]["label"],
|
|
value=LOCALES["render_mode"][DEFAULT_LANG]["choices"][0],
|
|
)
|
|
|
|
with gr.Row():
|
|
# 插件模式
|
|
plugin_options = gr.CheckboxGroup(
|
|
label=LOCALES["plugin"][DEFAULT_LANG]["label"],
|
|
choices=LOCALES["plugin"][DEFAULT_LANG]["choices"],
|
|
interactive=True,
|
|
)
|
|
|
|
# TAB2 - 高级参数 ------------------------------------------------
|
|
with gr.Tab(
|
|
LOCALES["advance_param"][DEFAULT_LANG]["label"]
|
|
) as advance_parameter_tab:
|
|
head_measure_ratio_option = gr.Slider(
|
|
minimum=0.1,
|
|
maximum=0.5,
|
|
value=0.2,
|
|
step=0.01,
|
|
label=LOCALES["head_measure_ratio"][DEFAULT_LANG]["label"],
|
|
interactive=True,
|
|
)
|
|
top_distance_option = gr.Slider(
|
|
minimum=0.02,
|
|
maximum=0.5,
|
|
value=0.12,
|
|
step=0.01,
|
|
label=LOCALES["top_distance"][DEFAULT_LANG]["label"],
|
|
interactive=True,
|
|
)
|
|
|
|
image_kb_options = gr.Radio(
|
|
choices=LOCALES["image_kb"][DEFAULT_LANG]["choices"],
|
|
label=LOCALES["image_kb"][DEFAULT_LANG]["label"],
|
|
value=LOCALES["image_kb"][DEFAULT_LANG]["choices"][0],
|
|
)
|
|
|
|
custom_image_kb_size = gr.Slider(
|
|
minimum=10,
|
|
maximum=1000,
|
|
value=50,
|
|
label=LOCALES["image_kb_size"][DEFAULT_LANG]["label"],
|
|
interactive=True,
|
|
visible=False,
|
|
)
|
|
|
|
image_dpi_options = gr.Radio(
|
|
choices=LOCALES["image_dpi"][DEFAULT_LANG]["choices"],
|
|
label=LOCALES["image_dpi"][DEFAULT_LANG]["label"],
|
|
value=LOCALES["image_dpi"][DEFAULT_LANG]["choices"][0],
|
|
)
|
|
custom_image_dpi_size = gr.Slider(
|
|
minimum=72,
|
|
maximum=600,
|
|
value=300,
|
|
label=LOCALES["image_dpi_size"][DEFAULT_LANG]["label"],
|
|
interactive=True,
|
|
visible=False,
|
|
)
|
|
|
|
# TAB3 - 美颜 ------------------------------------------------
|
|
with gr.Tab(
|
|
LOCALES["beauty_tab"][DEFAULT_LANG]["label"]
|
|
) as beauty_parameter_tab:
|
|
# 美白组件
|
|
whitening_option = gr.Slider(
|
|
label=LOCALES["whitening_strength"][DEFAULT_LANG]["label"],
|
|
minimum=0,
|
|
maximum=15,
|
|
value=2,
|
|
step=1,
|
|
interactive=True,
|
|
)
|
|
|
|
with gr.Row():
|
|
# 亮度组件
|
|
brightness_option = gr.Slider(
|
|
label=LOCALES["brightness_strength"][DEFAULT_LANG]["label"],
|
|
minimum=-5,
|
|
maximum=25,
|
|
value=0,
|
|
step=1,
|
|
interactive=True,
|
|
)
|
|
# 对比度组件
|
|
contrast_option = gr.Slider(
|
|
label=LOCALES["contrast_strength"][DEFAULT_LANG]["label"],
|
|
minimum=-10,
|
|
maximum=50,
|
|
value=0,
|
|
step=1,
|
|
interactive=True,
|
|
)
|
|
# 饱和度组件
|
|
saturation_option = gr.Slider(
|
|
label=LOCALES["saturation_strength"][DEFAULT_LANG]["label"],
|
|
minimum=-10,
|
|
maximum=50,
|
|
value=0,
|
|
step=1,
|
|
interactive=True,
|
|
)
|
|
|
|
# 锐化组件
|
|
sharpen_option = gr.Slider(
|
|
label=LOCALES["sharpen_strength"][DEFAULT_LANG]["label"],
|
|
minimum=0,
|
|
maximum=5,
|
|
value=0,
|
|
step=1,
|
|
interactive=True,
|
|
)
|
|
|
|
# TAB4 - 水印 ------------------------------------------------
|
|
with gr.Tab(
|
|
LOCALES["watermark_tab"][DEFAULT_LANG]["label"]
|
|
) as watermark_parameter_tab:
|
|
watermark_options = gr.Radio(
|
|
choices=LOCALES["watermark_switch"][DEFAULT_LANG]["choices"],
|
|
label=LOCALES["watermark_switch"][DEFAULT_LANG]["label"],
|
|
value=LOCALES["watermark_switch"][DEFAULT_LANG]["choices"][0],
|
|
)
|
|
|
|
with gr.Row():
|
|
watermark_text_options = gr.Text(
|
|
max_length=20,
|
|
label=LOCALES["watermark_text"][DEFAULT_LANG]["label"],
|
|
value=LOCALES["watermark_text"][DEFAULT_LANG]["value"],
|
|
placeholder=LOCALES["watermark_text"][DEFAULT_LANG][
|
|
"placeholder"
|
|
],
|
|
interactive=False,
|
|
)
|
|
watermark_text_color = gr.ColorPicker(
|
|
label=LOCALES["watermark_color"][DEFAULT_LANG]["label"],
|
|
interactive=False,
|
|
value="#FFFFFF",
|
|
)
|
|
|
|
watermark_text_size = gr.Slider(
|
|
minimum=10,
|
|
maximum=100,
|
|
value=20,
|
|
label=LOCALES["watermark_size"][DEFAULT_LANG]["label"],
|
|
interactive=False,
|
|
step=1,
|
|
)
|
|
|
|
watermark_text_opacity = gr.Slider(
|
|
minimum=0,
|
|
maximum=1,
|
|
value=0.15,
|
|
label=LOCALES["watermark_opacity"][DEFAULT_LANG]["label"],
|
|
interactive=False,
|
|
step=0.01,
|
|
)
|
|
|
|
watermark_text_angle = gr.Slider(
|
|
minimum=0,
|
|
maximum=360,
|
|
value=30,
|
|
label=LOCALES["watermark_angle"][DEFAULT_LANG]["label"],
|
|
interactive=False,
|
|
step=1,
|
|
)
|
|
|
|
watermark_text_space = gr.Slider(
|
|
minimum=10,
|
|
maximum=200,
|
|
value=25,
|
|
label=LOCALES["watermark_space"][DEFAULT_LANG]["label"],
|
|
interactive=False,
|
|
step=1,
|
|
)
|
|
|
|
def update_watermark_text_visibility(choice, language):
|
|
return [
|
|
gr.update(
|
|
interactive=(
|
|
choice
|
|
== LOCALES["watermark_switch"][language]["choices"][
|
|
1
|
|
]
|
|
)
|
|
)
|
|
] * 6
|
|
|
|
watermark_options.change(
|
|
fn=update_watermark_text_visibility,
|
|
inputs=[watermark_options, language_options],
|
|
outputs=[
|
|
watermark_text_options,
|
|
watermark_text_color,
|
|
watermark_text_size,
|
|
watermark_text_opacity,
|
|
watermark_text_angle,
|
|
watermark_text_space,
|
|
],
|
|
)
|
|
|
|
img_but = gr.Button(
|
|
LOCALES["button"][DEFAULT_LANG]["label"], elem_id="btn"
|
|
)
|
|
|
|
example_images = gr.Examples(
|
|
inputs=[img_input],
|
|
examples=[
|
|
[path.as_posix()]
|
|
for path in sorted(
|
|
pathlib.Path(os.path.join(root_dir, "demo/images")).rglob(
|
|
"*.jpg"
|
|
)
|
|
)
|
|
],
|
|
)
|
|
|
|
# ---------------- 右半边 UI ----------------
|
|
with gr.Column():
|
|
notification = gr.Text(
|
|
label=LOCALES["notification"][DEFAULT_LANG]["label"], visible=False
|
|
)
|
|
with gr.Row():
|
|
# 标准照
|
|
img_output_standard = gr.Image(
|
|
label=LOCALES["standard_photo"][DEFAULT_LANG]["label"],
|
|
height=350,
|
|
format="png",
|
|
)
|
|
# 高清照
|
|
img_output_standard_hd = gr.Image(
|
|
label=LOCALES["hd_photo"][DEFAULT_LANG]["label"],
|
|
height=350,
|
|
format="png",
|
|
)
|
|
# 排版照
|
|
img_output_layout = gr.Image(
|
|
label=LOCALES["layout_photo"][DEFAULT_LANG]["label"],
|
|
height=350,
|
|
format="png",
|
|
)
|
|
# 模版照片
|
|
with gr.Accordion(
|
|
LOCALES["template_photo"][DEFAULT_LANG]["label"], open=False
|
|
) as template_image_accordion:
|
|
img_output_template = gr.Gallery(
|
|
label=LOCALES["template_photo"][DEFAULT_LANG]["label"],
|
|
height=350,
|
|
format="png",
|
|
)
|
|
# 抠图图像
|
|
with gr.Accordion(
|
|
LOCALES["matting_image"][DEFAULT_LANG]["label"], open=False
|
|
) as matting_image_accordion:
|
|
with gr.Row():
|
|
img_output_standard_png = gr.Image(
|
|
label=LOCALES["standard_photo_png"][DEFAULT_LANG]["label"],
|
|
height=350,
|
|
format="png",
|
|
elem_id="standard_photo_png",
|
|
)
|
|
img_output_standard_hd_png = gr.Image(
|
|
label=LOCALES["hd_photo_png"][DEFAULT_LANG]["label"],
|
|
height=350,
|
|
format="png",
|
|
elem_id="hd_photo_png",
|
|
)
|
|
|
|
# ---------------- 多语言切换函数 ----------------
|
|
def change_language(language):
|
|
return {
|
|
face_detect_model_options: gr.update(
|
|
label=LOCALES["face_model"][language]["label"]
|
|
),
|
|
matting_model_options: gr.update(
|
|
label=LOCALES["matting_model"][language]["label"]
|
|
),
|
|
size_list_options: gr.update(
|
|
label=LOCALES["size_list"][language]["label"],
|
|
choices=LOCALES["size_list"][language]["choices"],
|
|
value=LOCALES["size_list"][language]["choices"][0],
|
|
),
|
|
mode_options: gr.update(
|
|
label=LOCALES["size_mode"][language]["label"],
|
|
choices=LOCALES["size_mode"][language]["choices"],
|
|
value=LOCALES["size_mode"][language]["choices"][0],
|
|
),
|
|
color_options: gr.update(
|
|
label=LOCALES["bg_color"][language]["label"],
|
|
choices=LOCALES["bg_color"][language]["choices"],
|
|
value=LOCALES["bg_color"][language]["choices"][0],
|
|
),
|
|
img_but: gr.update(value=LOCALES["button"][language]["label"]),
|
|
render_options: gr.update(
|
|
label=LOCALES["render_mode"][language]["label"],
|
|
choices=LOCALES["render_mode"][language]["choices"],
|
|
value=LOCALES["render_mode"][language]["choices"][0],
|
|
),
|
|
image_kb_options: gr.update(
|
|
label=LOCALES["image_kb_size"][language]["label"],
|
|
choices=LOCALES["image_kb"][language]["choices"],
|
|
value=LOCALES["image_kb"][language]["choices"][0],
|
|
),
|
|
custom_image_kb_size: gr.update(
|
|
label=LOCALES["image_kb"][language]["label"]
|
|
),
|
|
notification: gr.update(
|
|
label=LOCALES["notification"][language]["label"]
|
|
),
|
|
img_output_standard: gr.update(
|
|
label=LOCALES["standard_photo"][language]["label"]
|
|
),
|
|
img_output_standard_hd: gr.update(
|
|
label=LOCALES["hd_photo"][language]["label"]
|
|
),
|
|
img_output_standard_png: gr.update(
|
|
label=LOCALES["standard_photo_png"][language]["label"]
|
|
),
|
|
img_output_standard_hd_png: gr.update(
|
|
label=LOCALES["hd_photo_png"][language]["label"]
|
|
),
|
|
img_output_layout: gr.update(
|
|
label=LOCALES["layout_photo"][language]["label"]
|
|
),
|
|
head_measure_ratio_option: gr.update(
|
|
label=LOCALES["head_measure_ratio"][language]["label"]
|
|
),
|
|
top_distance_option: gr.update(
|
|
label=LOCALES["top_distance"][language]["label"]
|
|
),
|
|
key_parameter_tab: gr.update(
|
|
label=LOCALES["key_param"][language]["label"]
|
|
),
|
|
advance_parameter_tab: gr.update(
|
|
label=LOCALES["advance_param"][language]["label"]
|
|
),
|
|
watermark_parameter_tab: gr.update(
|
|
label=LOCALES["watermark_tab"][language]["label"]
|
|
),
|
|
watermark_text_options: gr.update(
|
|
label=LOCALES["watermark_text"][language]["label"],
|
|
placeholder=LOCALES["watermark_text"][language]["placeholder"],
|
|
),
|
|
watermark_text_color: gr.update(
|
|
label=LOCALES["watermark_color"][language]["label"]
|
|
),
|
|
watermark_text_size: gr.update(
|
|
label=LOCALES["watermark_size"][language]["label"]
|
|
),
|
|
watermark_text_opacity: gr.update(
|
|
label=LOCALES["watermark_opacity"][language]["label"]
|
|
),
|
|
watermark_text_angle: gr.update(
|
|
label=LOCALES["watermark_angle"][language]["label"]
|
|
),
|
|
watermark_text_space: gr.update(
|
|
label=LOCALES["watermark_space"][language]["label"]
|
|
),
|
|
watermark_options: gr.update(
|
|
label=LOCALES["watermark_switch"][language]["label"],
|
|
choices=LOCALES["watermark_switch"][language]["choices"],
|
|
value=LOCALES["watermark_switch"][language]["choices"][0],
|
|
),
|
|
matting_image_accordion: gr.update(
|
|
label=LOCALES["matting_image"][language]["label"]
|
|
),
|
|
beauty_parameter_tab: gr.update(
|
|
label=LOCALES["beauty_tab"][language]["label"]
|
|
),
|
|
whitening_option: gr.update(
|
|
label=LOCALES["whitening_strength"][language]["label"]
|
|
),
|
|
image_dpi_options: gr.update(
|
|
label=LOCALES["image_dpi"][language]["label"],
|
|
choices=LOCALES["image_dpi"][language]["choices"],
|
|
value=LOCALES["image_dpi"][language]["choices"][0],
|
|
),
|
|
custom_image_dpi_size: gr.update(
|
|
label=LOCALES["image_dpi"][language]["label"]
|
|
),
|
|
brightness_option: gr.update(
|
|
label=LOCALES["brightness_strength"][language]["label"]
|
|
),
|
|
contrast_option: gr.update(
|
|
label=LOCALES["contrast_strength"][language]["label"]
|
|
),
|
|
sharpen_option: gr.update(
|
|
label=LOCALES["sharpen_strength"][language]["label"]
|
|
),
|
|
saturation_option: gr.update(
|
|
label=LOCALES["saturation_strength"][language]["label"]
|
|
),
|
|
custom_size_width_px: gr.update(
|
|
label=LOCALES["custom_size_px"][language]["width"]
|
|
),
|
|
custom_size_height_px: gr.update(
|
|
label=LOCALES["custom_size_px"][language]["height"]
|
|
),
|
|
custom_size_width_mm: gr.update(
|
|
label=LOCALES["custom_size_mm"][language]["width"]
|
|
),
|
|
custom_size_height_mm: gr.update(
|
|
label=LOCALES["custom_size_mm"][language]["height"]
|
|
),
|
|
img_output_template: gr.update(
|
|
label=LOCALES["template_photo"][language]["label"]
|
|
),
|
|
template_image_accordion: gr.update(
|
|
label=LOCALES["template_photo"][language]["label"]
|
|
),
|
|
plugin_options: gr.update(
|
|
label=LOCALES["plugin"][language]["label"],
|
|
choices=LOCALES["plugin"][language]["choices"],
|
|
value=LOCALES["plugin"][language]["choices"][0],
|
|
),
|
|
}
|
|
|
|
def change_visibility(option, lang, locales_key, custom_component):
|
|
return {
|
|
custom_component: gr.update(
|
|
visible=option == LOCALES[locales_key][lang]["choices"][-1]
|
|
)
|
|
}
|
|
|
|
def change_color(colors, lang):
|
|
return {
|
|
custom_color_rgb: gr.update(visible = colors == LOCALES["bg_color"][lang]["choices"][-2]),
|
|
custom_color_hex: gr.update(visible = colors == LOCALES["bg_color"][lang]["choices"][-1]),
|
|
}
|
|
|
|
|
|
def change_size_mode(size_option_item, lang):
|
|
choices = LOCALES["size_mode"][lang]["choices"]
|
|
# 如果选择自定义尺寸mm
|
|
if size_option_item == choices[3]:
|
|
return {
|
|
custom_size_px: gr.update(visible=False),
|
|
custom_size_mm: gr.update(visible=True),
|
|
size_list_row: gr.update(visible=False),
|
|
plugin_options: gr.update(interactive=True),
|
|
}
|
|
# 如果选择自定义尺寸px
|
|
elif size_option_item == choices[2]:
|
|
return {
|
|
custom_size_px: gr.update(visible=True),
|
|
custom_size_mm: gr.update(visible=False),
|
|
size_list_row: gr.update(visible=False),
|
|
plugin_options: gr.update(interactive=True),
|
|
}
|
|
# 如果选择只换底，则隐藏所有尺寸组件
|
|
elif size_option_item == choices[1]:
|
|
return {
|
|
custom_size_px: gr.update(visible=False),
|
|
custom_size_mm: gr.update(visible=False),
|
|
size_list_row: gr.update(visible=False),
|
|
plugin_options: gr.update(interactive=False),
|
|
}
|
|
# 如果选择预设尺寸，则隐藏自定义尺寸组件
|
|
else:
|
|
return {
|
|
custom_size_px: gr.update(visible=False),
|
|
custom_size_mm: gr.update(visible=False),
|
|
size_list_row: gr.update(visible=True),
|
|
plugin_options: gr.update(interactive=True),
|
|
}
|
|
|
|
def change_image_kb(image_kb_option, lang):
|
|
return change_visibility(
|
|
image_kb_option, lang, "image_kb", custom_image_kb_size
|
|
)
|
|
|
|
def change_image_dpi(image_dpi_option, lang):
|
|
return change_visibility(
|
|
image_dpi_option, lang, "image_dpi", custom_image_dpi_size
|
|
)
|
|
|
|
# ---------------- 绑定事件 ----------------
|
|
# 语言切换
|
|
language_options.input(
|
|
change_language,
|
|
inputs=[language_options],
|
|
outputs=[
|
|
size_list_options,
|
|
mode_options,
|
|
color_options,
|
|
img_but,
|
|
render_options,
|
|
image_kb_options,
|
|
matting_model_options,
|
|
face_detect_model_options,
|
|
custom_image_kb_size,
|
|
notification,
|
|
img_output_standard,
|
|
img_output_standard_hd,
|
|
img_output_standard_png,
|
|
img_output_standard_hd_png,
|
|
img_output_layout,
|
|
head_measure_ratio_option,
|
|
top_distance_option,
|
|
key_parameter_tab,
|
|
advance_parameter_tab,
|
|
watermark_parameter_tab,
|
|
watermark_text_options,
|
|
watermark_text_color,
|
|
watermark_text_size,
|
|
watermark_text_opacity,
|
|
watermark_text_angle,
|
|
watermark_text_space,
|
|
watermark_options,
|
|
matting_image_accordion,
|
|
beauty_parameter_tab,
|
|
whitening_option,
|
|
image_dpi_options,
|
|
custom_image_dpi_size,
|
|
brightness_option,
|
|
contrast_option,
|
|
sharpen_option,
|
|
saturation_option,
|
|
plugin_options,
|
|
custom_size_width_px,
|
|
custom_size_height_px,
|
|
custom_size_width_mm,
|
|
custom_size_height_mm,
|
|
img_output_template,
|
|
template_image_accordion,
|
|
],
|
|
)
|
|
|
|
# ---------------- 设置隐藏/显示交互效果 ----------------
|
|
# 尺寸模式
|
|
mode_options.input(
|
|
change_size_mode,
|
|
inputs=[mode_options, language_options],
|
|
outputs=[
|
|
custom_size_px,
|
|
custom_size_mm,
|
|
size_list_row,
|
|
plugin_options,
|
|
],
|
|
)
|
|
|
|
# 颜色
|
|
color_options.input(
|
|
change_color,
|
|
inputs=[color_options, language_options],
|
|
outputs=[custom_color_rgb, custom_color_hex],
|
|
)
|
|
|
|
# 图片kb
|
|
image_kb_options.input(
|
|
change_image_kb,
|
|
inputs=[image_kb_options, language_options],
|
|
outputs=[custom_image_kb_size],
|
|
)
|
|
|
|
# 图片dpi
|
|
image_dpi_options.input(
|
|
change_image_dpi,
|
|
inputs=[image_dpi_options, language_options],
|
|
outputs=[custom_image_dpi_size],
|
|
)
|
|
|
|
img_but.click(
|
|
processor.process,
|
|
inputs=[
|
|
img_input,
|
|
mode_options,
|
|
size_list_options,
|
|
color_options,
|
|
render_options,
|
|
image_kb_options,
|
|
custom_color_R,
|
|
custom_color_G,
|
|
custom_color_B,
|
|
custom_color_hex_value,
|
|
custom_size_height_px,
|
|
custom_size_width_px,
|
|
custom_size_height_mm,
|
|
custom_size_width_mm,
|
|
custom_image_kb_size,
|
|
language_options,
|
|
matting_model_options,
|
|
watermark_options,
|
|
watermark_text_options,
|
|
watermark_text_color,
|
|
watermark_text_size,
|
|
watermark_text_opacity,
|
|
watermark_text_angle,
|
|
watermark_text_space,
|
|
face_detect_model_options,
|
|
head_measure_ratio_option,
|
|
top_distance_option,
|
|
whitening_option,
|
|
image_dpi_options,
|
|
custom_image_dpi_size,
|
|
brightness_option,
|
|
contrast_option,
|
|
sharpen_option,
|
|
saturation_option,
|
|
plugin_options,
|
|
],
|
|
outputs=[
|
|
img_output_standard,
|
|
img_output_standard_hd,
|
|
img_output_standard_png,
|
|
img_output_standard_hd_png,
|
|
img_output_layout,
|
|
img_output_template,
|
|
template_image_accordion,
|
|
notification,
|
|
],
|
|
)
|
|
|
|
return demo
|