From 017587a2fb89e1582321d2de6b1f86b0658f8674 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Sat, 27 Jul 2024 20:28:15 +0200 Subject: [PATCH] feat: add config for default_inpaint_advanced_masking_checkbox --- modules/config.py | 6 ++++++ webui.py | 4 ++-- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/modules/config.py b/modules/config.py index 1beb2f49..329f69c2 100644 --- a/modules/config.py +++ b/modules/config.py @@ -427,6 +427,12 @@ default_image_prompt_advanced_checkbox = get_config_item_or_set_default( validator=lambda x: isinstance(x, bool), expected_type=bool ) +default_inpaint_advanced_masking_checkbox = get_config_item_or_set_default( + key='default_inpaint_advanced_masking_checkbox', + default_value=False, + validator=lambda x: isinstance(x, bool), + expected_type=bool +) default_max_image_number = get_config_item_or_set_default( key='default_max_image_number', default_value=32, diff --git a/webui.py b/webui.py index 60af7905..c363ddb1 100644 --- a/webui.py +++ b/webui.py @@ -260,7 +260,7 @@ with shared.gradio_root: with gr.Row(): with gr.Column(): inpaint_input_image = grh.Image(label='Image', source='upload', type='numpy', tool='sketch', height=500, brush_color="#FFFFFF", elem_id='inpaint_canvas', show_label=False) - inpaint_advanced_masking_checkbox = gr.Checkbox(label='Enable Advanced Masking Features', value=False) + inpaint_advanced_masking_checkbox = gr.Checkbox(label='Enable Advanced Masking Features', value=modules.config.default_inpaint_advanced_masking_checkbox) inpaint_mode = gr.Dropdown(choices=modules.flags.inpaint_options, value=modules.config.default_inpaint_method, label='Method') inpaint_additional_prompt = gr.Textbox(placeholder="Describe what you want to inpaint.", elem_id='inpaint_additional_prompt', label='Inpaint Additional Prompt', visible=False) outpaint_selections = gr.CheckboxGroup(choices=['Left', 'Right', 'Top', 'Bottom'], value=[], label='Outpaint Direction') @@ -271,7 +271,7 @@ with shared.gradio_root: gr.HTML('* Powered by Fooocus Inpaint Engine \U0001F4D4 Documentation') example_inpaint_prompts.click(lambda x: x[0], inputs=example_inpaint_prompts, outputs=inpaint_additional_prompt, show_progress=False, queue=False) - with gr.Column(visible=False) as inpaint_mask_generation_col: + with gr.Column(visible=modules.config.default_inpaint_advanced_masking_checkbox) as inpaint_mask_generation_col: inpaint_mask_image = grh.Image(label='Mask Upload', source='upload', type='numpy', tool='sketch', height=500, brush_color="#FFFFFF", mask_opacity=1, elem_id='inpaint_mask_canvas') invert_mask_checkbox = gr.Checkbox(label='Invert Mask When Generating', value=False) inpaint_mask_model = gr.Dropdown(label='Mask generation model',