Adding in filename prefix

OneButtonPromptNodes.py

@@ -3,6 +3,7 @@ import os
 import folder_paths
 from datetime import datetime
 import uuid
+import platform
  
 custom_nodes_path = os.path.join(folder_paths.base_path, "custom_nodes")
 onebuttonprompt_path = os.path.join(custom_nodes_path, "OneButtonPrompt")
@@ -267,12 +268,15 @@ class CreatePromptVariant:
 # Let us create our own prompt saver. Not everyone has WAS installed
 class SavePromptToFile:
     def __init__(self):
-        pass
+        self.output_dir = folder_paths.get_output_directory()
+        self.type = "output"
+        self.prefix_append = ""
 
     @classmethod
     def INPUT_TYPES(s):
         return {
             "required": {
+                "filename_prefix": ("STRING", {"default": "Prompt"}),
                 "positive_prompt": ("STRING",{"multiline": True}),
                 "negative_prompt": ("STRING",{"multiline": True}),
             },
@@ -284,8 +288,61 @@ class SavePromptToFile:
 
     CATEGORY = "OneButtonPrompt"
 
-    def saveprompttofile(self, positive_prompt, negative_prompt):
-        
+    def saveprompttofile(self, positive_prompt, negative_prompt, filename_prefix):
+        # Some stuff for the prefix
+        filename_prefix += self.prefix_append
+
+        # turns out there is some hardcoded stuff on saveimage we have to kind of repeat here
+        # Find the %date:yyyy-M-d% pattern using regular expression
+        pattern = r'%date:([^\%]+)%'
+        match = re.search(pattern, filename_prefix)
+
+        if match:
+            # Extract the date format from the match
+            date_format = match.group(1)
+
+            # Get the current date
+            current_date = datetime.now()
+
+            # convert the ComfyUI standard into Python standard format.
+            # What a crazy way of doing this
+            # first lol, I got to make sure it doesn't overlap things
+            date_format = date_format.replace('M', 'X')
+            date_format = date_format.replace('m', 'Z')
+            
+            # This is so bad
+
+            # lets make it even worse, it work differently on windows than in Linux
+            if(platform.system() == 'Windows'):
+
+                date_format = date_format.replace('yyyy', '%Y')
+                date_format = date_format.replace('yy', '%#y')
+                date_format = date_format.replace('X', '%#m')
+                date_format = date_format.replace('d', '%#d')
+                date_format = date_format.replace('h', '%#H')
+                date_format = date_format.replace('Z', '%#M')
+                date_format = date_format.replace('s', '%#S')
+            else:
+                date_format = date_format.replace('yyyy', '%Y')
+                date_format = date_format.replace('yy', '%-y')
+                date_format = date_format.replace('X', '%-m')
+                date_format = date_format.replace('d', '%-d')
+                date_format = date_format.replace('h', '%-H')
+                date_format = date_format.replace('Z', '%-M')
+                date_format = date_format.replace('s', '%-S')
+
+
+            # Format the date using the extracted format
+            formatted_date = current_date.strftime(date_format)
+
+            # Replace the matched pattern with the formatted date
+            filename_prefix = re.sub(pattern, formatted_date, filename_prefix)
+            
+
+           
+
+        full_output_folder, filename_short, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir)
+
         # make the filename, from from a to the first comma
         # find the index of the first comma after "of a" or end of the prompt
         if(positive_prompt.find("of a ") != -1):
@@ -297,8 +354,6 @@ class SavePromptToFile:
             start_index = 0
             end_index = 128
   
-        
-
         # extract the desired substring using slicing
         filename = positive_prompt[start_index:end_index]
 
@@ -310,6 +365,7 @@ class SavePromptToFile:
         filename = filename.replace("<", "")
         filename = filename.replace(">", "")
         filename = filename.replace(":", "_")
+        filename = filename.replace(".", "")
         filename = re.sub(r'[0-9]+', '', filename)
 
         safe_characters = set("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_.")
@@ -321,15 +377,20 @@ class SavePromptToFile:
         if(filename==""):
             filename = str(uuid.uuid4())
         
+        if(filename_prefix == ""):
         # create a datetime object for the current date and time
-        now = datetime.now()
-        filenamecomplete = now.strftime("%Y%m%d%H%M%S") + "_" + filename.replace(" ", "_").strip() + ".txt"
-       
+        # if there is no prefix
+            now = datetime.now()
+            filenamecomplete = now.strftime("%Y%m%d%H%M%S") + "_" + filename.replace(" ", "_").strip() + ".txt"
         
-        # Get the output folder from comfy
-        output_directory = folder_paths.output_directory
+        else:
+            # lol since we insert a file, the counter of the image goes up by 1.
+            # So we add 1 here, so the prompt file matches the image file
+            formatted_counter = str(counter + 1).zfill(5)
+            filenamecomplete = filename_short + "_" + formatted_counter + "_" + filename.replace(" ", "_").strip() + ".txt"
+    
         
-        directoryandfilename = os.path.abspath(os.path.join(output_directory, filenamecomplete))
+        directoryandfilename = os.path.abspath(os.path.join(full_output_folder, filenamecomplete))
         
 
         with open(directoryandfilename, 'w', encoding="utf-8") as file: