diff --git a/modules/processing.py b/modules/processing.py index aaecb104..568f6098 100644 --- a/modules/processing.py +++ b/modules/processing.py @@ -65,6 +65,7 @@ class Processed: def __init__(self, p: StableDiffusionProcessing, images_list, seed, info): self.images = images_list self.prompt = p.prompt + self.negative_prompt = p.negative_prompt self.seed = seed self.info = info self.width = p.width @@ -76,6 +77,7 @@ class Processed: def js(self): obj = { "prompt": self.prompt if type(self.prompt) != list else self.prompt[0], + "negative_prompt": self.negative_prompt if type(self.negative_prompt) != list else self.negative_prompt[0], "seed": int(self.seed if type(self.seed) != list else self.seed[0]), "width": self.width, "height": self.height, diff --git a/modules/ui.py b/modules/ui.py index c32c5096..535afaeb 100644 --- a/modules/ui.py +++ b/modules/ui.py @@ -93,7 +93,7 @@ def save_files(js_data, images): at_start = file.tell() == 0 writer = csv.writer(file) if at_start: - writer.writerow(["prompt", "seed", "width", "height", "sampler", "cfgs", "steps", "filename"]) + writer.writerow(["prompt", "seed", "width", "height", "sampler", "cfgs", "steps", "filename", "negative_prompt"]) filename_base = str(int(time.time() * 1000)) for i, filedata in enumerate(images): @@ -108,7 +108,7 @@ def save_files(js_data, images): filenames.append(filename) - writer.writerow([data["prompt"], data["seed"], data["width"], data["height"], data["sampler"], data["cfg_scale"], data["steps"], filenames[0]]) + writer.writerow([data["prompt"], data["seed"], data["width"], data["height"], data["sampler"], data["cfg_scale"], data["steps"], filenames[0], data["negative_prompt"]]) return '', '', plaintext_to_html(f"Saved: {filenames[0]}")