Connect accumulator preview by store key

loop_nodes.py

@@ -646,13 +646,13 @@ class SxCPAccumulator:
             },
         }
 
-    RETURN_TYPES = tuple([ANY_TYPE, "IMAGE", "IMAGE"] + ["IMAGE"] * ACCUMULATOR_IMAGE_GROUPS + ["INT", "STRING"])
+    RETURN_TYPES = tuple([ANY_TYPE, "IMAGE", "IMAGE"] + ["IMAGE"] * ACCUMULATOR_IMAGE_GROUPS + ["INT", "STRING", "STRING"])
     RETURN_NAMES = tuple(
         ["collection", "image_batch", "image_list"]
         + [f"image_batch_{index}" for index in range(1, ACCUMULATOR_IMAGE_GROUPS + 1)]
-        + ["count", "status"]
+        + ["count", "status", "store_key"]
     )
-    OUTPUT_IS_LIST = tuple([False, False, True] + [False] * ACCUMULATOR_IMAGE_GROUPS + [False, False])
+    OUTPUT_IS_LIST = tuple([False, False, True] + [False] * ACCUMULATOR_IMAGE_GROUPS + [False, False, False])
     FUNCTION = "accumulate"
     CATEGORY = "prompt_builder/loop"
 
@@ -771,7 +771,7 @@ class SxCPAccumulator:
         grouped_outputs = image_batches[:ACCUMULATOR_IMAGE_GROUPS]
         grouped_outputs += [None] * (ACCUMULATOR_IMAGE_GROUPS - len(grouped_outputs))
         status = self._status(key, store, image_batch, image_batches)
-        return tuple([self._collection(store), image_batch, images] + grouped_outputs + [len(store), status])
+        return tuple([self._collection(store), image_batch, images] + grouped_outputs + [len(store), status, key])
 
 
 class SxCPAccumulatorPreview:
@@ -792,6 +792,9 @@ class SxCPAccumulatorPreview:
                 "filename_prefix": ("STRING", {"default": "sxcp_accum", "multiline": False}),
                 "clear_after_save": ("BOOLEAN", {"default": False}),
             },
+            "optional": {
+                "store_key_input": ("STRING", {"forceInput": True}),
+            },
             "hidden": {
                 "prompt": "PROMPT",
                 "extra_pnginfo": "EXTRA_PNGINFO",
@@ -838,11 +841,13 @@ class SxCPAccumulatorPreview:
         save_path,
         filename_prefix,
         clear_after_save,
+        store_key_input=None,
         prompt=None,
         extra_pnginfo=None,
         unique_id=None,
     ):
-        key = _accumulator_store_key(store_key, unique_id)
+        key_source = str(store_key_input or "").strip() or store_key
+        key = _accumulator_store_key(key_source, unique_id)
         store = _ACCUMULATOR_STORES.setdefault(key, [])
         removed = self._delete_from_inputs(key, delete_action, delete_entry_id, delete_index)
         images = [entry["image"] for entry in store if entry.get("image") is not None]
@@ -870,6 +875,7 @@ class SxCPAccumulatorPreview:
                 "entries": entries,
                 "status": [status],
                 "saved_paths": saved_paths,
+                "store_key": [key],
             },
             "result": (len(store), status, saved_json),
         }