Upload 2 files

Danbooru Prompt Selector/TEST2024/NAIA_0203_testv2.py

@@ -1896,6 +1896,9 @@ class App(customtkinter.CTk):
                         gen_request["count"] =self.generation_count
                     if app.auto_count_left_flag == True:
                         app.auto_count_left -= 1
+                    if self.image_generation_repeat_flag == True:
+                        gen_request["repeat"] = self.image_generation_repeat_current -1
+                        gen_request["repeat_max"] = self.image_generation_repeat
                     app.image_generation_button.configure(border_width = 2)
                     pass_bit = False
                     retry_count = 0

Danbooru Prompt Selector/TEST2024/NAIA_generation.py

@@ -291,6 +291,21 @@ def generate(gen_request):
         rating = gen_request["rating"]
         params["negative_prompt"] = parse_and_execute_commands(positive, params["negative_prompt"], user_input, rating)
 
+    if "repeat" in gen_request:
+        max = gen_request["repeat_max"]
+        keywords = [key.strip() for key in positive.split(',')]
+
+        for i, key in enumerate(keywords):
+            if "->" in key:
+                instant_keyword = [k for k in key.split('->')]
+                if len(instant_keyword) > gen_request["repeat"]:
+                    current_key = instant_keyword[gen_request["repeat"]]
+                else:
+                    current_key = instant_keyword[gen_request["repeat"] % len(instant_keyword)]
+                keywords[i] = current_key
+
+        positive = ', '.join(keywords)
+
     filename_rule = gen_request["png_rule"]
     save_folder = gen_request["save_folder"]