diff --git a/src/basic_data_handling/control_flow_nodes.py b/src/basic_data_handling/control_flow_nodes.py index 21c80c1..5aca5d8 100644 --- a/src/basic_data_handling/control_flow_nodes.py +++ b/src/basic_data_handling/control_flow_nodes.py @@ -213,6 +213,9 @@ 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): @@ -220,6 +223,9 @@ def INPUT_TYPES(cls): "required": { "value": (IO.ANY, {}), "select": (IO.BOOLEAN, {"default": True}), + }, + "optional": { + "message": (IO.STRING, {"default": ""}), } } @@ -229,11 +235,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):