Commit 8a1f32b6 authored by w-e-w's avatar w-e-w
Browse files

image hash

parent 541ef924
Loading
Loading
Loading
Loading
+6 −1
Original line number Diff line number Diff line
@@ -368,7 +368,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 +449,10 @@ 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 apply(self, x):
        res = ''