Merge branch 'AUTOMATIC1111:master' into master

1 files changed, 202 insertions, 0 deletions

diff --git a/modules/ui_common.py b/modules/ui_common.py
new file mode 100644
index 00000000..9405ac1f
--- /dev/null
+++ b/modules/ui_common.py
@@ -0,0 +1,202 @@
+import json

+import html

+import os

+import platform

+import sys

+

+import gradio as gr

+import subprocess as sp

+

+from modules import call_queue, shared

+from modules.generation_parameters_copypaste import image_from_url_text

+import modules.images

+

+folder_symbol = '\U0001f4c2'  # 📂

+

+

+def update_generation_info(generation_info, html_info, img_index):

+    try:

+        generation_info = json.loads(generation_info)

+        if img_index < 0 or img_index >= len(generation_info["infotexts"]):

+            return html_info, gr.update()

+        return plaintext_to_html(generation_info["infotexts"][img_index]), gr.update()

+    except Exception:

+        pass

+    # if the json parse or anything else fails, just return the old html_info

+    return html_info, gr.update()

+

+

+def plaintext_to_html(text):

+    text = "<p>" + "<br>\n".join([f"{html.escape(x)}" for x in text.split('\n')]) + "</p>"

+    return text

+

+

+def save_files(js_data, images, do_make_zip, index):

+    import csv

+    filenames = []

+    fullfns = []

+

+    #quick dictionary to class object conversion. Its necessary due apply_filename_pattern requiring it

+    class MyObject:

+        def __init__(self, d=None):

+            if d is not None:

+                for key, value in d.items():

+                    setattr(self, key, value)

+

+    data = json.loads(js_data)

+

+    p = MyObject(data)

+    path = shared.opts.outdir_save

+    save_to_dirs = shared.opts.use_save_to_dirs_for_ui

+    extension: str = shared.opts.samples_format

+    start_index = 0

+

+    if index > -1 and shared.opts.save_selected_only and (index >= data["index_of_first_image"]):  # ensures we are looking at a specific non-grid picture, and we have save_selected_only

+

+        images = [images[index]]

+        start_index = index

+

+    os.makedirs(shared.opts.outdir_save, exist_ok=True)

+

+    with open(os.path.join(shared.opts.outdir_save, "log.csv"), "a", encoding="utf8", newline='') as file:

+        at_start = file.tell() == 0

+        writer = csv.writer(file)

+        if at_start:

+            writer.writerow(["prompt", "seed", "width", "height", "sampler", "cfgs", "steps", "filename", "negative_prompt"])

+

+        for image_index, filedata in enumerate(images, start_index):

+            image = image_from_url_text(filedata)

+

+            is_grid = image_index < p.index_of_first_image

+            i = 0 if is_grid else (image_index - p.index_of_first_image)

+

+            fullfn, txt_fullfn = modules.images.save_image(image, path, "", seed=p.all_seeds[i], prompt=p.all_prompts[i], extension=extension, info=p.infotexts[image_index], grid=is_grid, p=p, save_to_dirs=save_to_dirs)

+

+            filename = os.path.relpath(fullfn, path)

+            filenames.append(filename)

+            fullfns.append(fullfn)

+            if txt_fullfn:

+                filenames.append(os.path.basename(txt_fullfn))

+                fullfns.append(txt_fullfn)

+

+        writer.writerow([data["prompt"], data["seed"], data["width"], data["height"], data["sampler_name"], data["cfg_scale"], data["steps"], filenames[0], data["negative_prompt"]])

+

+    # Make Zip

+    if do_make_zip:

+        zip_filepath = os.path.join(path, "images.zip")

+

+        from zipfile import ZipFile

+        with ZipFile(zip_filepath, "w") as zip_file:

+            for i in range(len(fullfns)):

+                with open(fullfns[i], mode="rb") as f:

+                    zip_file.writestr(filenames[i], f.read())

+        fullfns.insert(0, zip_filepath)

+

+    return gr.File.update(value=fullfns, visible=True), plaintext_to_html(f"Saved: {filenames[0]}")

+

+

+def create_output_panel(tabname, outdir):

+    from modules import shared

+    import modules.generation_parameters_copypaste as parameters_copypaste

+

+    def open_folder(f):

+        if not os.path.exists(f):

+            print(f'Folder "{f}" does not exist. After you create an image, the folder will be created.')

+            return

+        elif not os.path.isdir(f):

+            print(f"""

+WARNING

+An open_folder request was made with an argument that is not a folder.

+This could be an error or a malicious attempt to run code on your computer.

+Requested path was: {f}

+""", file=sys.stderr)

+            return

+

+        if not shared.cmd_opts.hide_ui_dir_config:

+            path = os.path.normpath(f)

+            if platform.system() == "Windows":

+                os.startfile(path)

+            elif platform.system() == "Darwin":

+                sp.Popen(["open", path])

+            elif "microsoft-standard-WSL2" in platform.uname().release:

+                sp.Popen(["wsl-open", path])

+            else:

+                sp.Popen(["xdg-open", path])

+

+    with gr.Column(variant='panel', elem_id=f"{tabname}_results"):

+        with gr.Group(elem_id=f"{tabname}_gallery_container"):

+            result_gallery = gr.Gallery(label='Output', show_label=False, elem_id=f"{tabname}_gallery").style(grid=4)

+

+        generation_info = None

+        with gr.Column():

+            with gr.Row(elem_id=f"image_buttons_{tabname}"):

+                open_folder_button = gr.Button(folder_symbol, elem_id="hidden_element" if shared.cmd_opts.hide_ui_dir_config else f'open_folder_{tabname}')

+

+                if tabname != "extras":

+                    save = gr.Button('Save', elem_id=f'save_{tabname}')

+                    save_zip = gr.Button('Zip', elem_id=f'save_zip_{tabname}')

+

+                buttons = parameters_copypaste.create_buttons(["img2img", "inpaint", "extras"])

+

+            open_folder_button.click(

+                fn=lambda: open_folder(shared.opts.outdir_samples or outdir),

+                inputs=[],

+                outputs=[],

+            )

+

+            if tabname != "extras":

+                with gr.Row():

+                    download_files = gr.File(None, file_count="multiple", interactive=False, show_label=False, visible=False, elem_id=f'download_files_{tabname}')

+

+                with gr.Group():

+                    html_info = gr.HTML(elem_id=f'html_info_{tabname}')

+                    html_log = gr.HTML(elem_id=f'html_log_{tabname}')

+

+                    generation_info = gr.Textbox(visible=False, elem_id=f'generation_info_{tabname}')

+                    if tabname == 'txt2img' or tabname == 'img2img':

+                        generation_info_button = gr.Button(visible=False, elem_id=f"{tabname}_generation_info_button")

+                        generation_info_button.click(

+                            fn=update_generation_info,

+                            _js="function(x, y, z){ return [x, y, selected_gallery_index()] }",

+                            inputs=[generation_info, html_info, html_info],

+                            outputs=[html_info, html_info],

+                        )

+

+                    save.click(

+                        fn=call_queue.wrap_gradio_call(save_files),

+                        _js="(x, y, z, w) => [x, y, false, selected_gallery_index()]",

+                        inputs=[

+                            generation_info,

+                            result_gallery,

+                            html_info,

+                            html_info,

+                        ],

+                        outputs=[

+                            download_files,

+                            html_log,

+                        ],

+                        show_progress=False,

+                    )

+

+                    save_zip.click(

+                        fn=call_queue.wrap_gradio_call(save_files),

+                        _js="(x, y, z, w) => [x, y, true, selected_gallery_index()]",

+                        inputs=[

+                            generation_info,

+                            result_gallery,

+                            html_info,

+                            html_info,

+                        ],

+                        outputs=[

+                            download_files,

+                            html_log,

+                        ]

+                    )

+

+            else:

+                html_info_x = gr.HTML(elem_id=f'html_info_x_{tabname}')

+                html_info = gr.HTML(elem_id=f'html_info_{tabname}')

+                html_log = gr.HTML(elem_id=f'html_log_{tabname}')

+

+            parameters_copypaste.bind_buttons(buttons, result_gallery, "txt2img" if tabname == "txt2img" else None)

+            return result_gallery, generation_info if tabname != "extras" else html_info_x, html_info, html_log