From 8f7802f6d81b2d4e4f46cbb057418cfbc309afef Mon Sep 17 00:00:00 2001 From: AUTOMATIC <16777216c@gmail.com> Date: Fri, 2 Sep 2022 02:12:50 +0300 Subject: saving text files along with images for >>88400870 --- webui.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/webui.py b/webui.py index ad38323e..0d335d92 100644 --- a/webui.py +++ b/webui.py @@ -231,7 +231,9 @@ class Options: "enable_pnginfo": OptionInfo(True, "Save text information about generation parameters as chunks to png files"), "font": OptionInfo("arial.ttf", "Font for image grids that have text"), "prompt_matrix_add_to_start": OptionInfo(True, "In prompt matrix, add the variable combination of text to the start of the prompt, rather than the end"), - "enable_emphasis": OptionInfo(True, "Use (text) to make model pay more attention to text text and [text] to make it pay less attention") + "enable_emphasis": OptionInfo(True, "Use (text) to make model pay more attention to text text and [text] to make it pay less attention"), + "save_txt": OptionInfo(False, "Create a text file next to every image with generation parameters."), + } def __init__(self): @@ -420,6 +422,11 @@ def save_image(image, path, basename, seed=None, prompt=None, extension='png', i image.save(f"{fullfn_without_extension}.jpg", quality=opts.jpeg_quality, pnginfo=pnginfo) + if opts.save_txt and info is not None: + with open(f"{fullfn_without_extension}.txt", "w", encoding="utf8") as file: + file.write(info + "\n") + + def sanitize_filename_part(text): return text.replace(' ', '_').translate({ord(x): '' for x in invalid_filename_chars})[:128] -- cgit v1.2.1