ComfyUI/doc-nodes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update DOC_SaveImageAndAddToHistory to save and update history file.

Browse Source
This commit is contained in:
Francesco Bellini 2025-10-05 11:57:09 +02:00
parent 738839b038
commit 2a9890638b
1 changed files with 24 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
nodes/prompts/DOC_SaveImageAndAddToHistory.py
View File

@ -8,12 +8,23 @@ from comfy.cli_args import args
import numpy as np import numpy as np
class DOC_SaveImageAndAddToHistory: class DOC_SaveImageAndAddToHistory:
def __init__(self): def __init__(self):
self.output_dir = folder_paths.get_output_directory() self.output_dir = folder_paths.get_output_directory()
self.type = "output" self.type = "output"
self.prefix_append = "" self.prefix_append = ""
self.compress_level = 4 self.compress_level = 4
self.history_filename = "generation_history.jsonl"
def _get_history_path(self):
return os.path.join(self.output_dir, self.history_filename)
def _append_history(self, entry):
history_path = self._get_history_path()
# Crea il file se non esiste
with open(history_path, "a", encoding="utf-8") as f:
f.write(json.dumps(entry, ensure_ascii=False) + "\n")
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
@ -39,7 +50,7 @@ class DOC_SaveImageAndAddToHistory:
CATEGORY = "image" CATEGORY = "image"
DESCRIPTION = "Saves the input images to your ComfyUI output directory and store prompt-image mapping in a global history file." DESCRIPTION = "Saves the input images to your ComfyUI output directory and store prompt-image mapping in a global history file."
def save_images(self, images, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=None): def save_images(self, images, filename_prefix="ComfyUI", llm_prompt=None, final_prompt=None, steps=None, cfg=None, prompt=None, extra_pnginfo=None):
filename_prefix += self.prefix_append filename_prefix += self.prefix_append
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir, images[0].shape[1], images[0].shape[0]) full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir, images[0].shape[1], images[0].shape[0])
results = list() results = list()
@ -57,12 +68,23 @@ class DOC_SaveImageAndAddToHistory:
filename_with_batch_num = filename.replace("%batch_num%", str(batch_number)) filename_with_batch_num = filename.replace("%batch_num%", str(batch_number))
file = f"{filename_with_batch_num}_{counter:05}_.png" file = f"{filename_with_batch_num}_{counter:05}_.png"
img.save(os.path.join(full_output_folder, file), pnginfo=metadata, compress_level=self.compress_level) file_path = os.path.join(full_output_folder, file)
img.save(file_path, pnginfo=metadata, compress_level=self.compress_level)
results.append({ results.append({
"filename": file, "filename": file,
"subfolder": subfolder, "subfolder": subfolder,
"type": self.type "type": self.type
}) })
# Scrivi la riga di history
history_entry = {
"path": os.path.abspath(file_path),
"llm_prompt": llm_prompt,
"final_prompt": final_prompt,
"steps": steps,
"cfg": cfg
}
self._append_history(history_entry)
counter += 1 counter += 1
return { "ui": { "images": results } } return { "ui": { "images": results } }