diff --git a/nodes.py b/nodes.py index 18768e8..e4f05f5 100644 --- a/nodes.py +++ b/nodes.py @@ -248,6 +248,7 @@ class SDPromptSaver: "STRING", {"default": "%H%M%S", "multiline": False}, ), + "extra_info": ("STRING", {"default": "", "multiline": True}), }, "hidden": {"prompt": "PROMPT", "extra_pnginfo": "EXTRA_PNGINFO"}, } @@ -283,6 +284,7 @@ class SDPromptSaver: jpg_webp_quality: int = 100, date_format: str = "%Y-%m-%d", time_format: str = "%H%M%S", + extra_info: str = "", prompt=None, extra_pnginfo=None, ): @@ -302,6 +304,8 @@ class SDPromptSaver: sampler_name_real = sampler_name_str if sampler_name_str else sampler_name scheduler_real = scheduler_str if scheduler_str else scheduler + extra_info_real = f", Extra info: {extra_info}" if extra_info else "" + variable_map = { "%date": self.get_time(date_format), "%time": self.get_time(time_format), @@ -336,6 +340,7 @@ class SDPromptSaver: f"{model_hash}" f"Model: {Path(model_name_real).stem}, " f"Version: ComfyUI" + f"{extra_info_real}" ) subfolder = self.get_path(path, variable_map) output_folder = Path(full_output_folder) / subfolder