Skip to content

Commit

Permalink
saving text files along with images for >>88400870
Browse files Browse the repository at this point in the history
  • Loading branch information
AUTOMATIC1111 committed Sep 1, 2022
1 parent cf726cc commit 8f7802f
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion webui.py
Original file line number Diff line number Diff line change
Expand Up @@ -231,7 +231,9 @@ def __init__(self, default=None, label="", component=None, component_args=None):
"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):
Expand Down Expand Up @@ -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]
Expand Down

0 comments on commit 8f7802f

Please sign in to comment.