Add files via upload

Fixed image saver so it does not force the user to connect "prompt_list"

image_saver/endless_image_saver.py

@@ -1,3 +1,7 @@
+# =============================
+# 🤓 Module-Level Code
+# =============================
+
 import os
 import json
 import re
@@ -11,19 +15,23 @@ import folder_paths
 from PIL.PngImagePlugin import PngInfo
 import platform
 
+# =============================
+# 🛠️ Workflow helper
+# =============================
+
 def get_workflow(prompt=None, extra_pnginfo=None):
     from server import PromptServer
 
-    # ✅ First: try directly from prompt
+    # First: try directly from prompt
     if isinstance(prompt, dict) and "workflow" in prompt:
         return prompt["workflow"]
 
-    # ✅ Second: try from extra_pnginfo
+    # Second: try from extra_pnginfo
     if isinstance(extra_pnginfo, dict) and "workflow" in extra_pnginfo:
         print("[INFO] Workflow recovered from extra_pnginfo.")
         return extra_pnginfo["workflow"]
 
-    # ✅ Third: fallback from PromptServer
+    # Third: fallback from PromptServer
     last = getattr(PromptServer.instance, "last_prompt", {})
     workflow = last.get("workflow")
     if workflow:
@@ -32,6 +40,16 @@ def get_workflow(prompt=None, extra_pnginfo=None):
 
     raise ValueError("🚫 No workflow found in prompt, extra_pnginfo, or PromptServer context.")
 
+# =============================
+# 🚧 Node Construction
+# =============================
+
+CATEGORY = "Endless 🌊✨/IO"
+
+# =============================
+# 💾 Image Saver
+# =============================
+
         
 class EndlessNode_Imagesaver:
     """
@@ -57,14 +75,10 @@ class EndlessNode_Imagesaver:
         else:
             return 200  # Conservative fallback
 
-
-
-
     @classmethod
     def INPUT_TYPES(s):
         return {"required": 
                     {"images": ("IMAGE", ),
-                     "prompt_list": ("STRING", {"forceInput": True}),
                      "include_timestamp": ("BOOLEAN", {"default": True}),
                      "timestamp_format": ("STRING", {"default": "%Y-%m-%d_%H-%M-%S", "description": "Use Python strftime format.\nExample: %Y-%m-%d %H-%M-%S\nSee: strftime.org for full options."}),
                      "image_format": (["PNG", "JPEG", "WEBP"], {"default": "PNG"}),
@@ -83,6 +97,7 @@ class EndlessNode_Imagesaver:
                      },
                 "optional":
                     {"output_path": ("STRING", {"default": ""}),
+                     "prompt_list": ("STRING", {"forceInput": True, "default": ""}),
                      "filename_prefix": ("STRING", {"default": "Batch"}),
                      "negative_prompt_list": ("STRING", {"default": ""}),
                      "json_folder": ("STRING", {"default": ""}),
@@ -97,7 +112,7 @@ class EndlessNode_Imagesaver:
     RETURN_NAMES = ("saved_paths",)
     FUNCTION = "save_batch_images"
     OUTPUT_NODE = True
-    CATEGORY = "Endless 🌊✨/IO"
+    CATEGORY = CATEGORY
 
     def encode_emoji(self, obj):
         """Properly encode emojis and special characters"""
@@ -236,7 +251,7 @@ class EndlessNode_Imagesaver:
         try:
             workflow = get_workflow(prompt)
 
-            # ✅ Ensure core fields
+            # Ensure core fields
             workflow.setdefault("version", 1)
             workflow.setdefault("nodes", [])
             workflow.setdefault("links", [])
@@ -244,10 +259,10 @@ class EndlessNode_Imagesaver:
                 max_id = max((n.get("id", 0) for n in workflow["nodes"]), default=0)
                 workflow["state"] = {"idCounter": max_id + 1}
 
-            # ✅ Remove sidecar metadata if present
+            # Remove sidecar metadata if present
             workflow.pop("extra_pnginfo", None)
 
-            # ✅ Embed custom metadata
+            # Embed custom metadata
             workflow["extra"] = workflow.get("extra", {})
             workflow["extra"].update({
                 "prompt": prompt_text,
@@ -304,7 +319,7 @@ class EndlessNode_Imagesaver:
         
         return filename
 
-    def save_batch_images(self, images, prompt_list, include_timestamp=True, 
+    def save_batch_images(self, images, prompt_list="", include_timestamp=True, 
                           timestamp_format="%Y-%m-%d_%H-%M-%S", image_format="PNG", 
                           jpeg_quality=95, delimiter="_", 
                           prompt_words_limit=8, embed_workflow=True, save_json_metadata=False,
@@ -317,7 +332,7 @@ class EndlessNode_Imagesaver:
         print(f"DEBUG: Images tensor shape: {images.shape}")
         print(f"DEBUG: Images tensor type: {type(images)}")
 
-        # ✅ Fallback: repair prompt if missing or partial
+        # Fallback: repair prompt if missing or partial
         if prompt is None or not isinstance(prompt, dict) or "workflow" not in prompt:
             if extra_pnginfo and "workflow" in extra_pnginfo:
                 print("[INFO] Workflow recovered from extra_pnginfo.")

pyproject.toml

@@ -1,15 +1,15 @@
-[project]
-name = "endless-nodes"
-description = "A small set of nodes I created for myself. Features multiple simultaneous prompts in batches, an image saver with ability to have JSON saved to separate folder, image analysis nodes, switches for text and numbers, and more."
-version = "1.2.4"
-license = { file = "LICENSE" }
-dependencies = ""
-
-[project.urls]
-Repository = "https://github.com/tusharbhutt/Endless-Nodes"
-#  Used by Comfy Registry https://comfyregistry.org
-
-[tool.comfy]
-PublisherId = "tusharbhutt"
-DisplayName = "Endless-Nodes"
-Icon = "https://raw.githubusercontent.com/tusharbhutt/Endless-Nodes/main/img/icon.png"
+[project]
+name = "endless-nodes"
+version = "1.3.0"
+description = "Modular, prompt-aware nodes for ComfyUI—analyze images, rank prompts, embed with CLIP, batch with control, and explore aesthetic/structural depth. 🌊✨"
+license = { file = "LICENSE" }
+readme = "README.md"
+dependencies = ""
+
+[project.urls]
+Repository = "https://github.com/tusharbhutt/Endless-Nodes"
+
+[tool.comfy]
+PublisherId = "tusharbhutt"
+DisplayName = "Endless 🌊✨ Nodes"
+Icon = "https://raw.githubusercontent.com/tusharbhutt/Endless-Nodes/main/img/icon.png"