diff --git a/args_manager.py b/args_manager.py index 5a2b37c9..08e4dc6e 100644 --- a/args_manager.py +++ b/args_manager.py @@ -28,8 +28,8 @@ args_parser.parser.add_argument("--disable-metadata", action='store_true', args_parser.parser.add_argument("--disable-preset-download", action='store_true', help="Disables downloading models for presets", default=False) -args_parser.parser.add_argument("--enable-describe-uov-image", action='store_true', - help="Disables automatic description of uov images when prompt is empty", default=False) +args_parser.parser.add_argument("--enable-auto-describe-image", action='store_true', + help="Enables automatic description of uov and enhance image when prompt is empty", default=False) args_parser.parser.add_argument("--always-download-new-model", action='store_true', help="Always download newer models ", default=False) diff --git a/webui.py b/webui.py index ca436057..02bb93b5 100644 --- a/webui.py +++ b/webui.py @@ -1001,15 +1001,18 @@ with shared.gradio_root: desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image], outputs=[prompt, style_selections], show_progress=True, queue=True) - if args_manager.args.enable_describe_uov_image: - def trigger_uov_describe(mode, img, prompt): + if args_manager.args.enable_auto_describe_image: + def trigger_auto_describe(mode, img, prompt): # keep prompt if not empty if prompt == '': return trigger_describe(mode, img) return gr.update(), gr.update() - uov_input_image.upload(trigger_uov_describe, inputs=[desc_method, uov_input_image, prompt], - outputs=[prompt, style_selections], show_progress=True, queue=True) + uov_input_image.upload(trigger_auto_describe, inputs=[desc_method, uov_input_image, prompt], + outputs=[prompt, style_selections], show_progress=True, queue=True) + + enhance_input_image.upload(trigger_auto_describe, inputs=[desc_method, enhance_input_image, prompt], + outputs=[prompt, style_selections], show_progress=True, queue=True) def dump_default_english_config(): from modules.localization import dump_english_config