diff options
author | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-14 11:55:42 +0000 |
---|---|---|
committer | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-14 11:55:42 +0000 |
commit | 7df7e4d22796fda11629463f2fcbe859b98b1d19 (patch) | |
tree | d0c2b306293a93c3233c44ac78c853d66eddecf0 /modules/ui_common.py | |
parent | 3715ece0adce7bf7c5e9c5ab3710b2fdc3848f39 (diff) | |
download | stable-diffusion-webui-gfx803-7df7e4d22796fda11629463f2fcbe859b98b1d19.tar.gz stable-diffusion-webui-gfx803-7df7e4d22796fda11629463f2fcbe859b98b1d19.tar.bz2 stable-diffusion-webui-gfx803-7df7e4d22796fda11629463f2fcbe859b98b1d19.zip |
Allow extensions to declare paste fields for "Send to X" buttons
Diffstat (limited to 'modules/ui_common.py')
-rw-r--r-- | modules/ui_common.py | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/modules/ui_common.py b/modules/ui_common.py index fd047f31..a12433d2 100644 --- a/modules/ui_common.py +++ b/modules/ui_common.py @@ -198,9 +198,16 @@ Requested path was: {f} html_info = gr.HTML(elem_id=f'html_info_{tabname}')
html_log = gr.HTML(elem_id=f'html_log_{tabname}')
+ paste_field_names = []
+ if tabname == "txt2img":
+ paste_field_names = modules.scripts.scripts_txt2img.paste_field_names
+ elif tabname == "img2img":
+ paste_field_names = modules.scripts.scripts_img2img.paste_field_names
+
for paste_tabname, paste_button in buttons.items():
parameters_copypaste.register_paste_params_button(parameters_copypaste.ParamBinding(
- paste_button=paste_button, tabname=paste_tabname, source_tabname="txt2img" if tabname == "txt2img" else None, source_image_component=result_gallery
+ paste_button=paste_button, tabname=paste_tabname, source_tabname="txt2img" if tabname == "txt2img" else None, source_image_component=result_gallery,
+ paste_field_names=paste_field_names
))
return result_gallery, generation_info if tabname != "extras" else html_info_x, html_info, html_log
|