commit 579c4457a3cad453194b033af8710efc217d70cf Author: Tasha Upchurch Date: Fri Jun 27 23:29:47 2025 -0600 feat: add __init__.py diff --git a/__init__.py b/__init__.py new file mode 100644 index 0000000..cf3a63c --- /dev/null +++ b/__init__.py @@ -0,0 +1,30 @@ +import random + +class RandomNumberButton: + WEB_DIRECTORY = "js" + CATEGORY = "Agent Assist" + RETURN_TYPES = ("INT",) + RETURN_NAMES = ("random_number",) + FUNCTION = "generate_random" + + @classmethod + def INPUT_TYPES(s): + # No standard inputs are needed for this node's function. + # The button is handled by Javascript triggering a workflow queue. + return {"required": {}} + + def generate_random(self): + """Generates a random integer between 0 and 1000.""" + number = random.randint(0, 1000) + # print(f"Generated random number: {number}") # Optional: for debugging + return (number,) +# ==================================================================================================== +# DO NOT TOUCH THIS SECTION, IT IS REQUIRED FOR THE NODE TO WORK IN COMFYUI +# ==================================================================================================== +NODE_CLASS_MAPPINGS = { + "RandomNumberButton": RandomNumberButton +} + +NODE_DISPLAY_NAME_MAPPINGS = { + "RandomNumberButton": "Random Number Button" +}