From 24fb30f0dd9747d9d33541622309993f546d4ec0 Mon Sep 17 00:00:00 2001
From: kex0 <46696708+kex0@users.noreply.github.com>
Date: Mon, 18 Sep 2023 16:46:03 +0200
Subject: [PATCH] Fix #130

Seed in OneButtonPrompt node now works
---
 OneButtonPromptNodes.py | 4 ++--
 build_dynamic_prompt.py | 6 ++++--
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/OneButtonPromptNodes.py b/OneButtonPromptNodes.py
index 95f50e6..a76852d 100644
--- a/OneButtonPromptNodes.py
+++ b/OneButtonPromptNodes.py
@@ -219,8 +219,8 @@ class OneButtonPrompt:
 
     CATEGORY = "OneButtonPrompt"
     
-    def Comfy_OBP(self, insanitylevel, custom_subject,seed, artist,imagetype, subject, imagemodechance, humanoids_gender, subject_subtype_objects, subject_subtypes_humanoids, subject_subtypes_concepts, emojis):
-        generatedprompt = build_dynamic_prompt(insanitylevel,subject,artist,imagetype,False,"","","",1,"",custom_subject,True,"",imagemodechance, humanoids_gender, subject_subtype_objects, subject_subtypes_humanoids, subject_subtypes_concepts, False, emojis)
+    def Comfy_OBP(self, insanitylevel, custom_subject, seed, artist, imagetype, subject, imagemodechance, humanoids_gender, subject_subtype_objects, subject_subtypes_humanoids, subject_subtypes_concepts, emojis):
+        generatedprompt = build_dynamic_prompt(insanitylevel,subject,artist,imagetype,False,"","","",1,"",seed,custom_subject,True,"",imagemodechance, humanoids_gender, subject_subtype_objects, subject_subtypes_humanoids, subject_subtypes_concepts, False, emojis)
         #print(generatedprompt)
         return (generatedprompt,)
 
diff --git a/build_dynamic_prompt.py b/build_dynamic_prompt.py
index cc9b33b..086e363 100644
--- a/build_dynamic_prompt.py
+++ b/build_dynamic_prompt.py
@@ -10,9 +10,11 @@ from random_functions import *
 # insanity level controls randomness of propmt 0-10
 # forcesubject van be used to force a certain type of subject
 # Set artistmode to none, to exclude artists 
-def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all", imagetype = "all", onlyartists = False, antivalues = "", prefixprompt = "", suffixprompt ="",promptcompounderlevel ="1", seperator = "comma", givensubject="",smartsubject = True,giventypeofimage="", imagemodechance = 20, gender = "all", subtypeobject="all", subtypehumanoid="all", subtypeconcept="all", advancedprompting=True, hardturnoffemojis=False):
+def build_dynamic_prompt(insanitylevel = 5, forcesubject = "all", artists = "all", imagetype = "all", onlyartists = False, antivalues = "", prefixprompt = "", suffixprompt ="",promptcompounderlevel ="1", seperator = "comma", seed=0, givensubject="",smartsubject = True,giventypeofimage="", imagemodechance = 20, gender = "all", subtypeobject="all", subtypehumanoid="all", subtypeconcept="all", advancedprompting=True, hardturnoffemojis=False):
+
+    # set seed
+    random.seed(seed)
 
-    
     # load the config file
     config = load_config_csv()