Merge pull request #31 from set-soft/block_msg

[ContinueFlow][Added] `message` widget

Author: StableLlama

src/basic_data_handling/control_flow_nodes.py

@@ -217,13 +217,19 @@ class ContinueFlow(ComfyNodeABC):
     This node takes a value and either passes it through or blocks execution
     based on the 'select' parameter. When 'select' is True, the value passes through;
     when False, execution is blocked.
+
+    When a `message` is provided ComfyUI will display it in a dialog.
+    Leave it empty for silent operation.
     """
     @classmethod
     def INPUT_TYPES(cls):
         return {
             "required": {
                 "value": (IO.ANY, {}),
                 "select": (IO.BOOLEAN, {"default": True}),
+            },
+            "optional": {
+                "message": (IO.STRING, {"default": ""}),
             }
         }
 
@@ -233,11 +239,11 @@ def INPUT_TYPES(cls):
     DESCRIPTION = cleandoc(__doc__ or "")
     FUNCTION = "execute"
 
-    def execute(self, value: Any, select: bool = True) -> tuple[Any]:
+    def execute(self, value: Any, select: bool = True, message: str = "") -> tuple[Any]:
         if select:
             return (value,)
         else:
-            return (ExecutionBlocker(None),)
+            return (ExecutionBlocker(message if message else None),)
 
 
 class FlowSelect(ComfyNodeABC):