From 579c4457a3cad453194b033af8710efc217d70cf Mon Sep 17 00:00:00 2001
From: Tasha Upchurch <TashaSkyUp@gmail.com>
Date: Fri, 27 Jun 2025 23:29:47 -0600
Subject: [PATCH] feat: add __init__.py

---
 __init__.py | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)
 create mode 100644 __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"
+}