在 pyproject.toml 中将版本号提升至 0.1.5；在 BatchImageLoopCloseSun 中添加 iteration_count 输入处理并改进图像保存逻辑。

nodes/loop_images.py

@@ -99,6 +99,7 @@ class BatchImageLoopCloseSun:
                 "current_image": ("IMAGE",),
                 "max_iterations": ("INT", {"forceInput": True}),
                 "batch_path": ("STRING", {"forceInput": True}),
+                "iteration_count": ("INT", {"forceInput": True}),
             },
             "optional": {
                 "pass_back": ("BOOLEAN", {"default": False}),
@@ -106,7 +107,6 @@ class BatchImageLoopCloseSun:
             "hidden": {
                 "dynprompt": "DYNPROMPT",
                 "unique_id": "UNIQUE_ID",
-                "iteration_count": ("INT", {"default": 0}),
             }
         }
 
@@ -122,9 +122,12 @@ class BatchImageLoopCloseSun:
         return image
 
     def save_image(self, image, batch_path, index):
+        filepath = os.path.join(batch_path, f"{index:05d}.png")
+        if os.path.exists(filepath):
+            print(f"[chen] Skip (exists): {filepath}")
+            return
         img_np = (image.squeeze(0).cpu().numpy() * 255).clip(0, 255).astype(np.uint8)
         img_pil = Image.fromarray(img_np)
-        filepath = os.path.join(batch_path, f"{index:05d}.png")
         img_pil.save(filepath)
         print(f"[chen] Saved: {filepath}")
 
@@ -139,7 +142,7 @@ class BatchImageLoopCloseSun:
         return torch.cat(results, dim=0).to(device)
 
     def while_loop_close(self, flow_control, current_image, max_iterations, batch_path,
-                         pass_back=False, iteration_count=0,
+                         iteration_count, pass_back=False,
                          dynprompt=None, unique_id=None):
         print(f"[chen] Iteration {iteration_count} / {max_iterations}")
 
@@ -203,6 +206,7 @@ class BatchImageLoopCloseSun:
 
         new_open = graph.lookup_node(open_node)
         new_open.set_input("iteration_count", iteration_count + 1)
+        new_open.set_input("batch_id", os.path.basename(batch_path))
         if pass_back:
             new_open.set_input("previous_image", current_image)
 

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "comfyui_sun_nodes"
-version = "0.1.3"
+version = "0.1.5"
 description = "Custom ComfyUI nodes by SunX.AI"
 authors = [
   {name = "SunX AI", email = "sunx.ai@hotmail.com"}