add img2img

app.py

@@ -5,6 +5,7 @@ import base64
 import requests
 import json 
 from PIL import Image
+import ui_functions as uifn
 
 def read_content(file_path: str) -> str:
     """read the content of target file
@@ -79,59 +80,299 @@ if __name__ == "__main__":
     with block:
         gr.HTML(read_content("header.html"))
 
-        with gr.Group():
-            with gr.Box():
-                with gr.Row().style(mobile_collapse=False, equal_height=True):
-                    text = gr.Textbox(
-                        label="Prompt",
-                        show_label=False,
-                        max_lines=1,
-                        placeholder="Input text(输入文字)",
-                        interactive=True,
-                    ).style(
-                        border=(True, False, True, True),
-                        rounded=(True, False, False, True),
-                        container=False,
-                    )
-                    
-                    btn = gr.Button("Generate image").style(
-                        margin=False,
-                        rounded=(True, True, True, True),
+        with gr.Tabs():
+
+            with gr.TabItem("文生图(Text-to-img)"):
+
+                with gr.Group():
+                    with gr.Box():
+                        with gr.Row().style(mobile_collapse=False, equal_height=True):
+                            text = gr.Textbox(
+                                label="Prompt",
+                                show_label=False,
+                                max_lines=1,
+                                placeholder="Input text(输入文字)",
+                                interactive=True,
+                            ).style(
+                                border=(True, False, True, True),
+                                rounded=(True, False, False, True),
+                                container=False,
+                            )
+                            
+                            btn = gr.Button("Generate image").style(
+                                margin=False,
+                                rounded=(True, True, True, True),
+                            )
+                        with gr.Row().style(mobile_collapse=False, equal_height=True):
+                            class_draw = gr.Dropdown(["通用(general)", "国画(traditional Chinese painting)", 
+                                                        "照片，摄影(picture photography)", "油画(oil painting)", 
+                                                        "铅笔素描(pencil sketch)", "CG", 
+                                                        "水彩画(watercolor painting)", "水墨画(ink and wash)",
+                                                        "插画(illustrations)", "3D", "图生图(img2img)"],
+                                                    label="生成类型(type)",
+                                                    show_label=True,
+                                                    value="通用(general)")
+                        with gr.Row().style(mobile_collapse=False, equal_height=True):
+                            style_draw = gr.CheckboxGroup(["蒸汽朋克(steampunk)", "电影摄影风格(film photography)", 
+                                                        "概念艺术(concept art)", "Warming lighting", 
+                                                        "Dramatic lighting", "Natural lighting", 
+                                                        "虚幻引擎(unreal engine)", "4k", "8k",
+                                                        "充满细节(full details)"],
+                                                    label="画面风格(style)",
+                                                    show_label=True,
+                                                    )
+                        with gr.Row().style(mobile_collapse=False, equal_height=True):
+                            sample_size = gr.Slider(minimum=1,
+                                                    maximum=4,
+                                                    step=1,
+                                                    label="生成数量(number)",
+                                                    show_label=True,
+                                                    interactive=True,
+                                                    )
+
+                    gallery = gr.Gallery(
+                        label="Generated images", show_label=False, elem_id="gallery"
+                    ).style(grid=[2], height="auto")
+
+                    gr.Examples(examples=examples, fn=request_images, inputs=text, outputs=gallery, examples_per_page=100)
+                    text.submit(request_images, inputs=[text, class_draw, style_draw, sample_size], outputs=gallery)
+                    btn.click(request_images, inputs=[text, class_draw, style_draw, sample_size], outputs=gallery)
+
+            with gr.TabItem("图生图(Img-to-Img)", id="img2img_tab"):
+                with gr.Row(elem_id="prompt_row"):
+                    img2img_prompt = gr.Textbox(label="Prompt",
+                                                elem_id='img2img_prompt_input',
+                                                placeholder="神奇的森林，流淌的河流.",
+                                                lines=1,
+                                                max_lines=1 if txt2img_defaults['submit_on_enter'] == 'Yes' else 25,
+                                                value=img2img_defaults['prompt'],
+                                                show_label=False).style()
+
+                    img2img_btn_mask = gr.Button("Generate", variant="primary", visible=False,
+                                                 elem_id="img2img_mask_btn")
+                    img2img_btn_editor = gr.Button("Generate", variant="primary", elem_id="img2img_edit_btn")
+                with gr.Row().style(equal_height=False):
+                    with gr.Column():
+                        gr.Markdown('#### 输入图像')
+                        img2img_image_mask = gr.Image(
+                            value=sample_img2img,
+                            source="upload",
+                            interactive=True,
+                            type="pil", tool="sketch",
+                            elem_id="img2img_mask",
+                            image_mode="RGBA"
+                        )
+                        img2img_image_editor = gr.Image(
+                            value=sample_img2img,
+                            source="upload",
+                            interactive=True,
+                            type="pil",
+                            tool="select",
+                            visible=False,
+                            image_mode="RGBA",
+                            elem_id="img2img_editor"
+                        )
+
+                        with gr.Tabs():
+                            with gr.TabItem("编辑设置"):
+                                with gr.Row():
+                                    # disable Uncrop for now
+                                    choices=["Mask", "Crop", "Uncrop"]
+                                    img2img_image_editor_mode = gr.Radio(choices=["Mask"],
+                                                                         label="编辑模式",
+                                                                         value="Mask", elem_id='edit_mode_select',
+                                                                         visible=True)
+                                    img2img_mask = gr.Radio(choices=["保留mask区域", "生成mask区域"],
+                                                            label="Mask 方式", type="index",
+                                                            #value=img2img_mask_modes[img2img_defaults['mask_mode']],
+                                                            value = "生成mask区域",
+                                                            visible=True)
+
+                                    img2img_mask_blur_strength = gr.Slider(minimum=1, maximum=10, step=1,
+                                                                           label="How much blurry should the mask be? (to avoid hard edges)",
+                                                                           value=3, visible=False)
+
+                                    img2img_resize = gr.Radio(label="Resize mode",
+                                                              choices=["Just resize", "Crop and resize",
+                                                                       "Resize and fill"],
+                                                              type="index",
+                                                              value=img2img_resize_modes[
+                                                                  img2img_defaults['resize_mode']], visible=False)
+
+                                img2img_painterro_btn = gr.Button("Advanced Editor",visible=False)
+                            # with gr.TabItem("Hints",visible=False):
+                            #     img2img_help = gr.Markdown(visible=False, value=uifn.help_text)
+
+                    with gr.Column():
+                        gr.Markdown('#### 编辑后的图片')
+                        output_img2img_gallery = gr.Gallery(label="Images", elem_id="img2img_gallery_output").style(
+                            grid=[4, 4, 4])
+                        img2img_job_ui = job_manager.draw_gradio_ui() if job_manager else None
+                        with gr.Column(visible=False):
+                            with gr.Tabs(visible=False):
+                                with gr.TabItem("", id="img2img_actions_tab",visible=False):
+                                    gr.Markdown("Select an image, then press one of the buttons below")
+                                    with gr.Row():
+                                        output_img2img_copy_to_clipboard_btn = gr.Button("Copy to clipboard")
+                                        output_img2img_copy_to_input_btn = gr.Button("Push to img2img input")
+                                        output_img2img_copy_to_mask_btn = gr.Button("Push to img2img input mask")
+
+                                    gr.Markdown("Warning: This will clear your current image and mask settings!")
+                                with gr.TabItem("", id="img2img_output_info_tab",visible=False):
+                                    output_img2img_params = gr.Textbox(label="Generation parameters")
+                                    with gr.Row():
+                                        output_img2img_copy_params = gr.Button("Copy full parameters").click(
+                                            inputs=output_img2img_params, outputs=[],
+                                            _js='(x) => {navigator.clipboard.writeText(x.replace(": ",":"))}', fn=None,
+                                            show_progress=False)
+                                        output_img2img_seed = gr.Number(label='Seed', interactive=False, visible=False)
+                                        output_img2img_copy_seed = gr.Button("Copy only seed").click(
+                                            inputs=output_img2img_seed, outputs=[],
+                                            _js=call_JS("gradioInputToClipboard"), fn=None, show_progress=False)
+                                    output_img2img_stats = gr.HTML(label='Stats')
+
+                gr.Markdown('# 编辑设置')
+
+                with gr.Row():
+                    with gr.Column():
+                        img2img_width = gr.Slider(minimum=64, maximum=2048, step=64, label="图片宽度",
+                                                  value=img2img_defaults["width"])
+                        img2img_height = gr.Slider(minimum=64, maximum=2048, step=64, label="图片高度",
+                                                   value=img2img_defaults["height"])
+                        img2img_cfg = gr.Slider(minimum=-40.0, maximum=30.0, step=0.5,
+                                                label='文本引导强度',
+                                                value=img2img_defaults['cfg_scale'], elem_id='cfg_slider')
+                        img2img_seed = gr.Textbox(label="随机种子", lines=1, max_lines=1,
+                                                  value=img2img_defaults["seed"])
+                        img2img_batch_count = gr.Slider(minimum=1, maximum=50, step=1,
+                                                        label='生成数量',
+                                                        value=img2img_defaults['n_iter'])
+                        img2img_dimensions_info_text_box = gr.Textbox(
+                            label="长宽比设置")
+                    with gr.Column():
+                        img2img_steps = gr.Slider(minimum=1, maximum=250, step=1, label="采样步数",
+                                                  value=img2img_defaults['ddim_steps'])
+
+                        img2img_sampling = gr.Dropdown(label='采样方式',
+                                                       choices=["DDIM", 'k_dpm_2_a', 'k_dpm_2', 'k_euler_a', 'k_euler',
+                                                                'k_heun', 'k_lms'],
+                                                       value=img2img_defaults['sampler_name'])
+
+                        img2img_denoising = gr.Slider(minimum=0.0, maximum=1.0, step=0.01, label='Denoising Strength',
+                                                      value=img2img_defaults['denoising_strength'],visible=False)
+
+                        img2img_toggles = gr.CheckboxGroup(label='', choices=img2img_toggles,
+                                                           value=img2img_toggle_defaults, type="index",visible=False)
+
+                        img2img_realesrgan_model_name = gr.Dropdown(label='RealESRGAN model',
+                                                                    choices=['RealESRGAN_x4plus',
+                                                                             'RealESRGAN_x4plus_anime_6B'],
+                                                                    value='RealESRGAN_x4plus',
+                                                                    visible=RealESRGAN is not None)  # TODO: Feels like I shouldnt slot it in here.
+
+                        img2img_embeddings = gr.File(label="Embeddings file for textual inversion",
+                                                     visible=show_embeddings)
+
+                img2img_image_editor_mode.change(
+                    uifn.change_image_editor_mode,
+                    [img2img_image_editor_mode,
+                     img2img_image_editor,
+                     img2img_image_mask,
+                     img2img_resize,
+                     img2img_width,
+                     img2img_height
+                     ],
+                    [img2img_image_editor, img2img_image_mask, img2img_btn_editor, img2img_btn_mask,
+                     img2img_painterro_btn, img2img_mask, img2img_mask_blur_strength]
+                )
+
+                # img2img_image_editor_mode.change(
+                #     uifn.update_image_mask,
+                #     [img2img_image_editor, img2img_resize, img2img_width, img2img_height],
+                #     img2img_image_mask
+                # )
+
+                output_txt2img_copy_to_input_btn.click(
+                    uifn.copy_img_to_input,
+                    [output_txt2img_gallery],
+                    [img2img_image_editor, img2img_image_mask, tabs],
+                    _js=call_JS("moveImageFromGallery",
+                                fromId="txt2img_gallery_output",
+                                toId="img2img_editor")
+                )
+
+                output_img2img_copy_to_input_btn.click(
+                    uifn.copy_img_to_edit,
+                    [output_img2img_gallery],
+                    [img2img_image_editor, tabs, img2img_image_editor_mode],
+                    _js=call_JS("moveImageFromGallery",
+                                fromId="img2img_gallery_output",
+                                toId="img2img_editor")
+                )
+                output_img2img_copy_to_mask_btn.click(
+                    uifn.copy_img_to_mask,
+                    [output_img2img_gallery],
+                    [img2img_image_mask, tabs, img2img_image_editor_mode],
+                    _js=call_JS("moveImageFromGallery",
+                                fromId="img2img_gallery_output",
+                                toId="img2img_editor")
+                )
+
+                output_img2img_copy_to_clipboard_btn.click(fn=None, inputs=output_img2img_gallery, outputs=[],
+                                                           _js=call_JS("copyImageFromGalleryToClipboard",
+                                                                       fromId="img2img_gallery_output")
+                                                           )
+
+                img2img_func = img2img
+                img2img_inputs = [img2img_prompt, img2img_image_editor_mode, img2img_mask,
+                                  img2img_mask_blur_strength, img2img_steps, img2img_sampling, img2img_toggles,
+                                  img2img_realesrgan_model_name, img2img_batch_count, img2img_cfg,
+                                  img2img_denoising, img2img_seed, img2img_height, img2img_width, img2img_resize,
+                                  img2img_image_editor, img2img_image_mask, img2img_embeddings]
+                img2img_outputs = [output_img2img_gallery, output_img2img_seed, output_img2img_params,
+                                   output_img2img_stats]
+
+                # If a JobManager was passed in then wrap the Generate functions
+                if img2img_job_ui:
+                    img2img_func, img2img_inputs, img2img_outputs = img2img_job_ui.wrap_func(
+                        func=img2img_func,
+                        inputs=img2img_inputs,
+                        outputs=img2img_outputs,
                     )
-                with gr.Row().style(mobile_collapse=False, equal_height=True):
-                    class_draw = gr.Dropdown(["通用(general)", "国画(traditional Chinese painting)", 
-                                                "照片，摄影(picture photography)", "油画(oil painting)", 
-                                                "铅笔素描(pencil sketch)", "CG", 
-                                                "水彩画(watercolor painting)", "水墨画(ink and wash)",
-                                                "插画(illustrations)", "3D", "图生图(img2img)"],
-                                            label="生成类型(type)",
-                                            show_label=True,
-                                            value="通用(general)")
-                with gr.Row().style(mobile_collapse=False, equal_height=True):
-                    style_draw = gr.CheckboxGroup(["蒸汽朋克(steampunk)", "电影摄影风格(film photography)", 
-                                                "概念艺术(concept art)", "Warming lighting", 
-                                                "Dramatic lighting", "Natural lighting", 
-                                                "虚幻引擎(unreal engine)", "4k", "8k",
-                                                "充满细节(full details)"],
-                                            label="画面风格(style)",
-                                            show_label=True,
-                                            )
-                with gr.Row().style(mobile_collapse=False, equal_height=True):
-                     sample_size = gr.Slider(minimum=1,
-                                            maximum=4,
-                                            step=1,
-                                            label="生成数量(number)",
-                                            show_label=True,
-                                            interactive=True,
-                                            )
 
-            gallery = gr.Gallery(
-                label="Generated images", show_label=False, elem_id="gallery"
-            ).style(grid=[2], height="auto")
+                img2img_btn_mask.click(
+                    img2img_func,
+                    img2img_inputs,
+                    img2img_outputs
+                )
+
+                def img2img_submit_params():
+                    # print([img2img_prompt, img2img_image_editor_mode, img2img_mask,
+                    #              img2img_mask_blur_strength, img2img_steps, img2img_sampling, img2img_toggles,
+                    #              img2img_realesrgan_model_name, img2img_batch_count, img2img_cfg,
+                    #              img2img_denoising, img2img_seed, img2img_height, img2img_width, img2img_resize,
+                    #              img2img_image_editor, img2img_image_mask, img2img_embeddings])
+                    return (img2img_func,
+                            img2img_inputs,
+                            img2img_outputs)
+
+                img2img_btn_editor.click(*img2img_submit_params())
+
+                # GENERATE ON ENTER
+                img2img_prompt.submit(None, None, None,
+                                      _js=call_JS("clickFirstVisibleButton",
+                                                  rowId="prompt_row"))
+
+                img2img_painterro_btn.click(None,
+                                            [img2img_image_editor, img2img_image_mask, img2img_image_editor_mode],
+                                            [img2img_image_editor, img2img_image_mask],
+                                            _js=call_JS("Painterro.init", toId="img2img_editor")
+                                            )
 
-            gr.Examples(examples=examples, fn=request_images, inputs=text, outputs=gallery, examples_per_page=100)
-            text.submit(request_images, inputs=[text, class_draw, style_draw, sample_size], outputs=gallery)
-            btn.click(request_images, inputs=[text, class_draw, style_draw, sample_size], outputs=gallery)
+                img2img_width.change(fn=uifn.update_dimensions_info, inputs=[img2img_width, img2img_height],
+                                     outputs=img2img_dimensions_info_text_box)
+                img2img_height.change(fn=uifn.update_dimensions_info, inputs=[img2img_width, img2img_height],
+                                      outputs=img2img_dimensions_info_text_box)
 
         gr.HTML(read_content("footer.html"))
         # gr.Image('./contributors.png')

ui_functions.py

@@ -0,0 +1,219 @@
+import re
+import gradio as gr
+from PIL import Image, ImageFont, ImageDraw, ImageFilter, ImageOps
+from io import BytesIO
+import base64
+import re
+
+
+def change_image_editor_mode(choice, cropped_image, masked_image, resize_mode, width, height):
+    if choice == "Mask":
+        update_image_result = update_image_mask(cropped_image, resize_mode, width, height)
+        return [gr.update(visible=False), update_image_result, gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=True)]
+
+    update_image_result = update_image_mask(masked_image["image"] if masked_image is not None else None, resize_mode, width, height)
+    return [update_image_result, gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False)]
+
+def update_image_mask(cropped_image, resize_mode, width, height):
+    resized_cropped_image = resize_image(resize_mode, cropped_image, width, height) if cropped_image else None
+    return gr.update(value=resized_cropped_image, visible=True)
+
+def toggle_options_gfpgan(selection):
+    if 0 in selection:
+        return gr.update(visible=True)
+    else:
+        return gr.update(visible=False)
+
+def toggle_options_upscalers(selection):
+    if 1 in selection:
+        return gr.update(visible=True)
+    else:
+        return gr.update(visible=False)
+
+def toggle_options_realesrgan(selection):
+    if selection == 0 or selection == 1 or selection == 3:
+        return gr.update(visible=True)
+    else:
+        return gr.update(visible=False)
+
+def toggle_options_gobig(selection):
+    if selection == 1:
+        #print(selection)
+        return gr.update(visible=True)
+    if selection == 3:
+        return gr.update(visible=True)
+    else:
+        return gr.update(visible=False)
+
+def toggle_options_ldsr(selection):
+    if selection == 2 or selection == 3:
+        return gr.update(visible=True)
+    else:
+        return gr.update(visible=False)
+
+def increment_down(value):
+    return value - 1
+
+def increment_up(value):
+    return value + 1
+
+def copy_img_to_lab(img):
+    try:
+        image_data = re.sub('^data:image/.+;base64,', '', img)
+        processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
+        tab_update = gr.update(selected='imgproc_tab')
+        img_update = gr.update(value=processed_image)
+        return processed_image, tab_update,
+    except IndexError:
+        return [None, None]
+def copy_img_params_to_lab(params):
+    try:
+        prompt = params[0][0].replace('\n', ' ').replace('\r', '')
+        seed = int(params[1][1])
+        steps = int(params[7][1])
+        cfg_scale = float(params[9][1])
+        sampler = params[11][1]
+        return prompt,seed,steps,cfg_scale,sampler
+    except IndexError:
+        return [None, None]
+def copy_img_to_input(img):
+    try:
+        image_data = re.sub('^data:image/.+;base64,', '', img)
+        processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
+        tab_update = gr.update(selected='img2img_tab')
+        img_update = gr.update(value=processed_image)
+        return processed_image, processed_image , tab_update
+    except IndexError:
+        return [None, None]
+
+def copy_img_to_edit(img):
+    try:
+        image_data = re.sub('^data:image/.+;base64,', '', img)
+        processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
+        tab_update = gr.update(selected='img2img_tab')
+        img_update = gr.update(value=processed_image)
+        mode_update = gr.update(value='Crop')
+        return processed_image, tab_update, mode_update
+    except IndexError:
+        return [None, None]
+
+def copy_img_to_mask(img):
+    try:
+        image_data = re.sub('^data:image/.+;base64,', '', img)
+        processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
+        tab_update = gr.update(selected='img2img_tab')
+        img_update = gr.update(value=processed_image)
+        mode_update = gr.update(value='Mask')
+        return processed_image, tab_update, mode_update
+    except IndexError:
+        return [None, None]
+
+
+
+def copy_img_to_upscale_esrgan(img):
+    tabs_update = gr.update(selected='realesrgan_tab')
+    image_data = re.sub('^data:image/.+;base64,', '', img)
+    processed_image = Image.open(BytesIO(base64.b64decode(image_data)))
+    return processed_image, tabs_update
+
+
+help_text = """
+    ## Mask/Crop
+    * Masking is not inpainting. You will probably get better results manually masking your images in photoshop instead.
+    * Built-in masking/cropping is very temperamental.
+    * It may take some time for the image to show when switching from Crop to Mask.
+    * If the image doesn't appear after switching to Mask, switch back to Crop and then back again to Mask
+    * If the mask appears distorted (the brush is weirdly shaped instead of round), switch back to Crop and then back again to Mask.
+
+    ## Advanced Editor
+    * Click 💾 Save to send your editor changes to the img2img workflow
+    * Click ❌ Clear to discard your editor changes
+
+    If anything breaks, try switching modes again, switch tabs, clear the image, or reload.
+"""
+
+def resize_image(resize_mode, im, width, height):
+    LANCZOS = (Image.Resampling.LANCZOS if hasattr(Image, 'Resampling') else Image.LANCZOS)
+    if resize_mode == 0:
+        res = im.resize((width, height), resample=LANCZOS)
+    elif resize_mode == 1:
+        ratio = width / height
+        src_ratio = im.width / im.height
+
+        src_w = width if ratio > src_ratio else im.width * height // im.height
+        src_h = height if ratio <= src_ratio else im.height * width // im.width
+
+        resized = im.resize((src_w, src_h), resample=LANCZOS)
+        res = Image.new("RGBA", (width, height))
+        res.paste(resized, box=(width // 2 - src_w // 2, height // 2 - src_h // 2))
+    else:
+        ratio = width / height
+        src_ratio = im.width / im.height
+
+        src_w = width if ratio < src_ratio else im.width * height // im.height
+        src_h = height if ratio >= src_ratio else im.height * width // im.width
+
+        resized = im.resize((src_w, src_h), resample=LANCZOS)
+        res = Image.new("RGBA", (width, height))
+        res.paste(resized, box=(width // 2 - src_w // 2, height // 2 - src_h // 2))
+
+        if ratio < src_ratio:
+            fill_height = height // 2 - src_h // 2
+            res.paste(resized.resize((width, fill_height), box=(0, 0, width, 0)), box=(0, 0))
+            res.paste(resized.resize((width, fill_height), box=(0, resized.height, width, resized.height)), box=(0, fill_height + src_h))
+        elif ratio > src_ratio:
+            fill_width = width // 2 - src_w // 2
+            res.paste(resized.resize((fill_width, height), box=(0, 0, 0, height)), box=(0, 0))
+            res.paste(resized.resize((fill_width, height), box=(resized.width, 0, resized.width, height)), box=(fill_width + src_w, 0))
+
+    return res
+
+def update_dimensions_info(width, height):
+    pixel_count_formated = "{:,.0f}".format(width * height)
+    return f"Aspect ratio: {round(width / height, 5)}\nTotal pixel count: {pixel_count_formated}"
+
+def get_png_nfo( image: Image ):
+    info_text = ""
+    visible = bool(image and any(image.info))
+    if visible:
+        for key,value in image.info.items():
+            info_text += f"{key}: {value}\n"
+        info_text = info_text.rstrip('\n')
+    return gr.Textbox.update(value=info_text, visible=visible)
+
+def load_settings(*values):
+    new_settings, key_names, checkboxgroup_info = values[-3:]
+    values = list(values[:-3])
+
+    if new_settings:
+        if type(new_settings) is str:
+            if os.path.exists(new_settings):
+                with open(new_settings, "r", encoding="utf8") as f:
+                    new_settings = yaml.safe_load(f)
+            elif new_settings.startswith("file://") and os.path.exists(new_settings[7:]):
+                with open(new_settings[7:], "r", encoding="utf8") as f:
+                    new_settings = yaml.safe_load(f)
+            else:
+                new_settings = yaml.safe_load(new_settings)
+        if type(new_settings) is not dict:
+            new_settings = {"prompt": new_settings}
+        if "txt2img" in new_settings:
+            new_settings = new_settings["txt2img"]
+        target = new_settings.pop("target", "txt2img")
+        if target != "txt2img":
+            print(f"Warning: applying settings to txt2img even though {target} is specified as target.", file=sys.stderr)
+
+        skipped_settings = {}
+        for key in new_settings.keys():
+            if key in key_names:
+                values[key_names.index(key)] = new_settings[key]
+            else:
+                skipped_settings[key] = new_settings[key]
+        if skipped_settings:
+            print(f"Settings could not be applied: {skipped_settings}", file=sys.stderr)
+
+    # Convert lists of checkbox indices to lists of checkbox labels:
+    for (cbg_index, cbg_choices) in checkboxgroup_info:
+        values[cbg_index] = [cbg_choices[i] for i in values[cbg_index]]
+
+    return values