ComfyUI/doc-nodes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update randomness

Browse Source
This commit is contained in:
Francesco Bellini 2025-10-07 22:06:20 +02:00
parent 1abda678a2
commit 289cf61268
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nodes/DOC_RandomPromptChoice.py
View File

@ -9,7 +9,8 @@ class DOC_RandomPromptChoice:
def INPUT_TYPES(cls): def INPUT_TYPES(cls):
return { return {
"required": { "required": {
**{f"prompt_{i+1}": ("STRING", {"default": "", "tooltip": f"Prompt {i+1}"}) for i in range(20)} **{f"prompt_{i+1}": ("STRING", {"default": "", "tooltip": f"Prompt {i+1}"}) for i in range(20)},
"seed": ("INT", {"default": datetime.datetime.now().timestamp(), "tooltip": "The seed for random number generation."}),
} }
} }
@ -19,11 +20,11 @@ class DOC_RandomPromptChoice:
CATEGORY = "DOC" CATEGORY = "DOC"
DISPLAY_NAME = "Random Prompt Choice" DISPLAY_NAME = "Random Prompt Choice"
def choose_random_prompt(self, **kwargs): def choose_random_prompt(self, seed, **kwargs):
prompts = [v for v in kwargs.values() if v.strip()] prompts = [v for v in kwargs.values() if v.strip()]
if not prompts: if not prompts:
return ("",) return ("",)
sysrand = random.SystemRandom() sysrand = random.SystemRandom()
sysrand.seed(datetime.datetime.now().timestamp()) sysrand.seed(seed)
choice = sysrand.choice(prompts) choice = sysrand.choice(prompts)
return (choice,) return (choice,)