OPENGIT/stable-diffusion-webui
1
0
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-04-29 00:09:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Copy img2img layers

Browse Source
This commit is contained in:
missionfloyd 2024-04-25 18:36:29 -06:00
parent 43e893ce2a
commit ca59516fa1
1 changed files with 3 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
modules/ui.py
View File

@ -542,7 +542,7 @@ def create_ui():
add_copy_image_controls('img2img', init_img) add_copy_image_controls('img2img', init_img)
with gr.TabItem('Sketch', id='img2img_sketch', elem_id="img2img_img2img_sketch_tab") as tab_sketch: with gr.TabItem('Sketch', id='img2img_sketch', elem_id="img2img_img2img_sketch_tab") as tab_sketch:
sketch = gr.ImageEditor(label="Image for img2img", elem_id="img2img_sketch", show_label=False, interactive=True, type="pil", image_mode="RGB", brush=Brush(default_color=opts.img2img_sketch_default_brush_color)) sketch = gr.ImageEditor(label="Image for img2img", elem_id="img2img_sketch", show_label=False, interactive=True, type="pil", image_mode="RGBA", brush=Brush(default_color=opts.img2img_sketch_default_brush_color))
add_copy_image_controls('sketch', sketch) add_copy_image_controls('sketch', sketch)
with gr.TabItem('Inpaint', id='inpaint', elem_id="img2img_inpaint_tab") as tab_inpaint: with gr.TabItem('Inpaint', id='inpaint', elem_id="img2img_inpaint_tab") as tab_inpaint:
@ -578,23 +578,14 @@ def create_ui():
for i, tab in enumerate(img2img_tabs): for i, tab in enumerate(img2img_tabs):
tab.select(fn=lambda tabnum=i: tabnum, inputs=[], outputs=[img2img_selected_tab]) tab.select(fn=lambda tabnum=i: tabnum, inputs=[], outputs=[img2img_selected_tab])
def copy_image(img):
if isinstance(img, dict) and 'image' in img:
return img['image']
if isinstance(img, dict) and 'composite' in img:
return img['composite']
return img
for button, name, elem in copy_image_buttons: for button, name, elem in copy_image_buttons:
button.click( button.click(
fn=copy_image, fn=lambda img: img,
inputs=[elem], inputs=[elem],
outputs=[copy_image_destinations[name]], outputs=[copy_image_destinations[name]],
) )
button.click( button.click(
fn=lambda: None, fn=None,
_js=f"switch_to_{name.replace(' ', '_')}", _js=f"switch_to_{name.replace(' ', '_')}",
inputs=[], inputs=[],
outputs=[], outputs=[],