From 5c1d2e81723187411d47a0fb92584ad4cf905ce9 Mon Sep 17 00:00:00 2001 From: Manuel Schmid Date: Mon, 15 Jan 2024 13:25:51 +0100 Subject: [PATCH] feat: automatically describe image on uov image upload if prompt is empty --- webui.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/webui.py b/webui.py index 581e3101..bf520e98 100644 --- a/webui.py +++ b/webui.py @@ -605,6 +605,14 @@ with shared.gradio_root: desc_btn.click(trigger_describe, inputs=[desc_method, desc_input_image], outputs=[prompt, style_selections], show_progress=True, queue=True) + def trigger_uov_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) def dump_default_english_config(): from modules.localization import dump_english_config