From 715ea6135567e6ac453f0da76df1b455ec7d8140 Mon Sep 17 00:00:00 2001
From: aimfordeb <133010426+aimfordeb@users.noreply.github.com>
Date: Fri, 13 Mar 2026 17:27:24 +0200
Subject: [PATCH] Additional prompts?

---
 funpack.py | 27 ++++++++++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/funpack.py b/funpack.py
index 4429e26b..87096401 100644
--- a/funpack.py
+++ b/funpack.py
@@ -599,6 +599,11 @@ class FunPackStoryWriter:
         return {
             "required": {
                 "user_prompt": ("STRING", {"multiline": True, "default": "A photo of a [subject] in a [setting]. [action]."}),
+                "prompt1": ("STRING", {"multiline": False, "default": ""}),
+                "prompt2": ("STRING", {"multiline": False, "default": ""}),
+                "prompt3": ("STRING", {"multiline": False, "default": ""}),
+                "prompt4": ("STRING", {"multiline": False, "default": ""}),
+                "prompt5": ("STRING", {"multiline": False, "default": ""}),
                 "story_system_prompt": ("STRING", {
                     "multiline": True,
                     "default": ""
@@ -634,7 +639,7 @@ class FunPackStoryWriter:
     FUNCTION = "write_story"
     CATEGORY = "FunPack"
 
-    def write_story(self, user_prompt, story_system_prompt, sequence_system_prompt, model_path_type, model_path, llm_safetensors_file, prompt_count, top_p, top_k, min_p, temperature, max_new_tokens, repetition_penalty, mode, vision_input, sanity_check, sanity_check_system_prompt, disable_continuity, provide_current_id):
+    def write_story(self, user_prompt, prompt1, prompt2, prompt3, prompt4, prompt5, story_system_prompt, sequence_system_prompt, model_path_type, model_path, llm_safetensors_file, prompt_count, top_p, top_k, min_p, temperature, max_new_tokens, repetition_penalty, mode, vision_input, sanity_check, sanity_check_system_prompt, disable_continuity, provide_current_id):
         llm_model = None
         llm_tokenizer = None
         llm_model_device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -669,6 +674,14 @@ class FunPackStoryWriter:
 
             outputs = [""] * 5
 
+            prompts = [""] * 5
+
+            prompts[0] = prompt1
+            prompts[1] = prompt2
+            prompts[2] = prompt3
+            prompts[3] = prompt4
+            prompts[4] = prompt5
+
             recommended_loras = None
 
             # ── Initialize messages ONCE, depending on mode ────────────────────────────
@@ -718,18 +731,26 @@ class FunPackStoryWriter:
                         {"role": "user", "content": user_prompt}
                     ]
 
+                    if prompts[seq_idx]:
+                        messages.append({"role": "user", "content": prompts[seq_idx]})
+
                 elif disable_continuity == True and provide_current_id == True:
                     messages = [
                         {"role": "system", "content": sequence_system_prompt},
-                        {"role": "user", "content": f"""Current request ID: {seq_idx+1}\nUser's instruction: {user_prompt}"""}
+                        {"role": "user", "content": f"""Current request ID: {seq_idx+1}\nUser's message: {user_prompt}"""}
                     ]
+                    
+                    if prompts[seq_idx]:
+                        messages.append({"role": "user", "content": prompts[seq_idx]})
                 
                 else:
                     messages = [
                         {"role": "system", "content": sequence_system_prompt},
-                        {"role": "user", "content": f"""Total amount of requests in this batch: {prompt_count}\nCurrently generating request ID {seq_idx+1} out of {prompt_count}\nRequests left in queue: {prompt_count - seq_idx - 1}\nUser's instruction: {user_prompt}"""},
+                        {"role": "user", "content": f"""Total amount of requests in this batch: {prompt_count}\nCurrently generating request ID {seq_idx+1} out of {prompt_count}\nRequests left in queue: {prompt_count - seq_idx - 1}\nUser's message: {user_prompt}"""},
                         {"role": "assistant", "content": f"""History:{chr(10).join([f"ID {i}: {text}" for i, text in enumerate(outputs[:seq_idx])]) if seq_idx > 0 else "No history available."}"""}
                     ]
+                    if prompts[seq_idx]:
+                        messages.append({"role": "user", "content": prompts[seq_idx]})
 
                 if vision_input is not None:
                     messages.append({"role": "user", "content": f"""Reference image description (this is the starting image in the video batch): {vision_input}"""})