Add new FilenameGenerator [hasprompt<prompt1|default><prompt2>..]

2 files changed, 19 insertions, 2 deletions

diff --git a/javascript/hints.js b/javascript/hints.js
index 7f4101b2..730ce7bd 100644
--- a/javascript/hints.js
+++ b/javascript/hints.js
@@ -67,8 +67,8 @@ titles = {
 
     "Interrogate": "Reconstruct prompt from existing image and put it into the prompt field.",
 
-    "Images filename pattern": "Use following tags to define how filenames for images are chosen: [steps], [cfg], [prompt_hash], [prompt], [prompt_no_styles], [prompt_spaces], [width], [height], [styles], [sampler], [seed], [model_hash], [model_name], [prompt_words], [date], [datetime], [datetime<Format>], [datetime<Format><Time Zone>], [job_timestamp]; leave empty for default.",
-    "Directory name pattern": "Use following tags to define how subdirectories for images and grids are chosen: [steps], [cfg],[prompt_hash], [prompt], [prompt_no_styles], [prompt_spaces], [width], [height], [styles], [sampler], [seed], [model_hash], [model_name], [prompt_words], [date], [datetime], [datetime<Format>], [datetime<Format><Time Zone>], [job_timestamp]; leave empty for default.",
+    "Images filename pattern": "Use following tags to define how filenames for images are chosen: [steps], [cfg], [prompt_hash], [prompt], [prompt_no_styles], [prompt_spaces], [width], [height], [styles], [sampler], [seed], [model_hash], [model_name], [prompt_words], [date], [datetime], [datetime<Format>], [datetime<Format><Time Zone>], [job_timestamp], [hasprompt<prompt1|default><prompt2>..]; leave empty for default.",
+    "Directory name pattern": "Use following tags to define how subdirectories for images and grids are chosen: [steps], [cfg],[prompt_hash], [prompt], [prompt_no_styles], [prompt_spaces], [width], [height], [styles], [sampler], [seed], [model_hash], [model_name], [prompt_words], [date], [datetime], [datetime<Format>], [datetime<Format><Time Zone>], [job_timestamp], [hasprompt<prompt1|default><prompt2>..]; leave empty for default.",
     "Max prompt words": "Set the maximum number of words to be used in the [prompt_words] option; ATTENTION: If the words are too long, they may exceed the maximum length of the file path that the system can handle",
 
     "Loopback": "Process an image, use it as an input, repeat.",
diff --git a/modules/images.py b/modules/images.py
index 2da988ee..cdbb77e1 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -349,6 +349,7 @@ class FilenameGenerator:
         'prompt_no_styles': lambda self: self.prompt_no_style(),

         'prompt_spaces': lambda self: sanitize_filename_part(self.prompt, replace_spaces=False),

         'prompt_words': lambda self: self.prompt_words(),

+        'hasprompt': lambda self, *args: self.hasprompt(*args), #accept formats:[hasprompt<prompt1|default><prompt2>..]

     }

     default_time_format = '%Y%m%d%H%M%S'

 

@@ -357,6 +358,22 @@ class FilenameGenerator:
         self.seed = seed

         self.prompt = prompt

         self.image = image

+        

+    def hasprompt(self, *args):

+        lower = self.prompt.lower()

+        if self.p is None or self.prompt is None:

+            return None

+        outres = ""

+        for arg in args:

+            if arg != "":

+                division = arg.split("|")

+                expected = division[0].lower()

+                default = division[1] if len(division) > 1 else ""

+                if lower.find(expected) >= 0:

+                    outres = f'{outres}{expected}'

+                else:

+                    outres = outres if default == "" else f'{outres}{default}'

+        return sanitize_filename_part(outres)

 

     def prompt_no_style(self):

         if self.p is None or self.prompt is None: