Merge pull request #12639 from AUTOMATIC1111/more-hash

More hash filename patterns

1 files changed, 13 insertions, 2 deletions

diff --git a/modules/images.py b/modules/images.py
index 019c1d60..a6b4fb1e 100644
--- a/modules/images.py
+++ b/modules/images.py
@@ -355,7 +355,9 @@ class FilenameGenerator:
         'date': lambda self: datetime.datetime.now().strftime('%Y-%m-%d'),

         'datetime': lambda self, *args: self.datetime(*args),  # accepts formats: [datetime], [datetime<Format>], [datetime<Format><Time Zone>]

         'job_timestamp': lambda self: getattr(self.p, "job_timestamp", shared.state.job_timestamp),

-        'prompt_hash': lambda self: hashlib.sha256(self.prompt.encode()).hexdigest()[0:8],

+        'prompt_hash': lambda self, *args: self.string_hash(self.prompt, *args),

+        'negative_prompt_hash': lambda self, *args: self.string_hash(self.p.negative_prompt, *args),

+        'full_prompt_hash': lambda self, *args: self.string_hash(f"{self.p.prompt} {self.p.negative_prompt}", *args),  # a space in between to create a unique string

         'prompt': lambda self: sanitize_filename_part(self.prompt),

         'prompt_no_styles': lambda self: self.prompt_no_style(),

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

@@ -368,7 +370,8 @@ class FilenameGenerator:
         'denoising': lambda self: self.p.denoising_strength if self.p and self.p.denoising_strength else NOTHING_AND_SKIP_PREVIOUS_TEXT,

         'user': lambda self: self.p.user,

         'vae_filename': lambda self: self.get_vae_filename(),

-        'none': lambda self: '', # Overrides the default so you can get just the sequence number

+        'none': lambda self: '',  # Overrides the default, so you can get just the sequence number

+        'image_hash': lambda self, *args: self.image_hash(*args)  # accepts formats: [image_hash<length>] default full hash

     }

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

 

@@ -448,6 +451,14 @@ class FilenameGenerator:
 

         return sanitize_filename_part(formatted_time, replace_spaces=False)

 

+    def image_hash(self, *args):

+        length = int(args[0]) if (args and args[0] != "") else None

+        return hashlib.sha256(self.image.tobytes()).hexdigest()[0:length]

+

+    def string_hash(self, text, *args):

+        length = int(args[0]) if (args and args[0] != "") else 8

+        return hashlib.sha256(text.encode()).hexdigest()[0:length]

+

     def apply(self, x):

         res = ''