Add SeedGenerator

__init__.py

@@ -1,15 +1,18 @@
 from .negi.openai_dalle3 import OpenAiDalle3
 from .negi.openai_translate import OpenAiTranslate
 from .negi.string_function import StringFunction
+from .negi.seed_generator import SeedGenerator
 
 NODE_CLASS_MAPPINGS = {
     "NegiTools_OpenAiDalle3": OpenAiDalle3,
     "NegiTools_OpenAiTranslate": OpenAiTranslate,
     "NegiTools_StringFunction": StringFunction,
+    "NegiTools_SeedGenerator": SeedGenerator,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "NegiTools_OpenAiDalle3": "OpenAI DALLe3 🧅",
     "NegiTools_OpenAiTranslate": "OpenAI Translate to English 🧅",
-    "NegiTools_StringFunction": "String Function 🧅"
+    "NegiTools_StringFunction": "String Function 🧅",
+    "NegiTools_SeedGenerator": "Seed Generator 🧅",
 }

negi/seed_generator.py

@@ -0,0 +1,25 @@
+import random
+
+
+class SeedGenerator:
+    def __init__(self):
+        self.__previous_seed = random.randint(0, 0xffffffffffffffff)
+
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "generation_mode": (["random", "keep_previous"],),
+            }
+        }
+
+    RETURN_TYPES = ("INT",)
+    RETURN_NAMES = ("SEED",)
+    FUNCTION = "doit"
+    OUTPUT_NODE = True
+    CATEGORY = "utils"
+
+    def doit(self, generation_mode):
+        if generation_mode == "random":
+            self.__previous_seed = random.randint(0, 0xffffffffffffffff)
+        return (self.__previous_seed,)