From b5e8061b98a8a406d8d5ad72e860cd553ae6d08d Mon Sep 17 00:00:00 2001
From: Revmagi <Revmagi@users.noreply.github.com>
Date: Thu, 30 Oct 2025 15:24:09 -0400
Subject: [PATCH] Update ddm_nodes.py

---
 ddm_nodes.py | 40 ++++++++++++++++++++++++++++------------
 1 file changed, 28 insertions(+), 12 deletions(-)

diff --git a/ddm_nodes.py b/ddm_nodes.py
index 6565d56..d4c7894 100644
--- a/ddm_nodes.py
+++ b/ddm_nodes.py
@@ -16,7 +16,7 @@ any_type = AnyType("*")
 
 class DDM_SetNode:
     """
-    Store any value with a unique identifier.
+    Store any value with a unique identifier and pass it through.
     """
     
     @classmethod
@@ -32,12 +32,11 @@ class DDM_SetNode:
     RETURN_NAMES = ("value",)
     FUNCTION = "execute"
     CATEGORY = "DDM Bundle"
-    OUTPUT_NODE = True
     
     def execute(self, value, id):
         STORAGE[id] = value
-        print(f"[DDM Set] Stored '{id}' | Storage: {list(STORAGE.keys())}")
-        return {"ui": {"text": [f"Stored: {id}"]}, "result": (value,)}
+        print(f"✓ [DDM Set] Stored '{id}' successfully")
+        return (value,)
 
 
 class DDM_GetNode:
@@ -51,6 +50,9 @@ class DDM_GetNode:
             "required": {
                 "id": ("STRING", {"default": "value"}),
             },
+            "optional": {
+                "default": (any_type,),
+            },
         }
     
     RETURN_TYPES = (any_type,)
@@ -59,19 +61,33 @@ class DDM_GetNode:
     CATEGORY = "DDM Bundle"
     
     @classmethod
-    def IS_CHANGED(cls, id):
+    def IS_CHANGED(cls, id, default=None):
         return float("nan")
     
-    def execute(self, id):
-        print(f"[DDM Get] Looking for '{id}' | Available: {list(STORAGE.keys())}")
+    def execute(self, id, default=None):
+        print(f"→ [DDM Get] Attempting to retrieve '{id}'")
+        print(f"  Current storage: {list(STORAGE.keys())}")
         
         if id not in STORAGE:
-            raise ValueError(
-                f"Get Node Error: '{id}' not found!\n"
-                f"Available IDs: {list(STORAGE.keys())}\n"
-                f"Make sure the Set node with id='{id}' is connected to your workflow."
-            )
+            if default is not None:
+                print(f"⚠ [DDM Get] '{id}' not found, using default value")
+                return (default,)
+            else:
+                error_msg = (
+                    f"\n{'='*60}\n"
+                    f"DDM Get Node Error: ID '{id}' not found!\n"
+                    f"Current storage: {list(STORAGE.keys())}\n\n"
+                    f"Troubleshooting:\n"
+                    f"1. Make sure there's a Set Node with id='{id}'\n"
+                    f"2. The Set Node OUTPUT must be connected to your workflow\n"
+                    f"3. The Set Node must lead to a final output node\n"
+                    f"4. Try adding a 'default' input to this Get Node as backup\n"
+                    f"{'='*60}\n"
+                )
+                print(error_msg)
+                raise ValueError(error_msg)
         
+        print(f"✓ [DDM Get] Successfully retrieved '{id}'")
         return (STORAGE[id],)