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, ) # 人脸对齐 face_alignment_options = gr.CheckboxGroup( label=LOCALES["face_alignment"][DEFAULT_LANG]["label"], choices=LOCALES["face_alignment"][DEFAULT_LANG]["choices"], interactive=True, ) # 尺寸列表 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], ) # 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"] ), face_alignment_options: gr.update( label=LOCALES["face_alignment"][language]["label"], choices=LOCALES["face_alignment"][language]["choices"], ), 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"] ), } 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), face_alignment_options: gr.update(visible=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), face_alignment_options: gr.update(visible=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), face_alignment_options: gr.update(visible=False), } # 如果选择预设尺寸，则隐藏自定义尺寸组件 else: return { custom_size_px: gr.update(visible=False), custom_size_mm: gr.update(visible=False), size_list_row: gr.update(visible=True), face_alignment_options: gr.update(visible=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, face_alignment_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, face_alignment_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, face_alignment_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