|
|
|
@@ -11,7 +11,7 @@ import pprint
|
|
|
|
|
import piexif
|
|
|
|
|
import piexif.helper
|
|
|
|
|
|
|
|
|
|
version = 2.6.5
|
|
|
|
|
version = 2.74
|
|
|
|
|
|
|
|
|
|
avif_supported = False
|
|
|
|
|
jxl_supported = False
|
|
|
|
@@ -20,23 +20,27 @@ jxl_supported = False
|
|
|
|
|
try:
|
|
|
|
|
import pillow_avif
|
|
|
|
|
except:
|
|
|
|
|
print(f"\033[92m[save_image_extended]\033[0m AVIF is not supported. To add it: pip install pillow pillow-avif-plugin\033[0m")
|
|
|
|
|
print(f"\033[92m[💾 save_image_extended]\033[0m AVIF is not supported. To add it: pip install pillow pillow-avif-plugin\033[0m")
|
|
|
|
|
pass
|
|
|
|
|
else:
|
|
|
|
|
print(f"\033[92m[save_image_extended] AVIF is supported! Woohoo!\033[0m")
|
|
|
|
|
print(f"\033[92m[💾 save_image_extended] AVIF is supported! Woohoo!\033[0m")
|
|
|
|
|
avif_supported = True
|
|
|
|
|
|
|
|
|
|
# Jxl requires jxlpy wheel to be compiled, and a valid MSVC environment, which is complex task
|
|
|
|
|
try:
|
|
|
|
|
from jxlpy import JXLImagePlugin
|
|
|
|
|
# jxlpy is in early stages of development. None one has ever compiled it on Windows AFAIK
|
|
|
|
|
# from jxlpy import JXLImagePlugin
|
|
|
|
|
# from imagecodecs import (jpegxl_encode, jpegxl_decode, jpegxl_check, jpegxl_version, JPEGXL)
|
|
|
|
|
import pillow_jxl
|
|
|
|
|
except:
|
|
|
|
|
print(f"\033[92m[save_image_extended]\033[0m JXL is not supported. To add it: pip install jxlpy\033[0m")
|
|
|
|
|
print(f"\033[92m[save_image_extended]\033[0m You will need a valid MSVC env to build the wheel\033[0m")
|
|
|
|
|
print(f"\033[92m[💾 save_image_extended]\033[0m JXL is not supported. To add it: pip install jxlpy\033[0m")
|
|
|
|
|
print(f"\033[92m[💾 save_image_extended]\033[0m You will need a valid MSVC env to build the wheel\033[0m")
|
|
|
|
|
pass
|
|
|
|
|
else:
|
|
|
|
|
print(f"\033[92m[💾 save_image_extended] JPEGXL is supported! YeePee!\033[0m")
|
|
|
|
|
jxl_supported = True
|
|
|
|
|
|
|
|
|
|
# PIL must be loaded after pillow_avif
|
|
|
|
|
# PIL must be loaded after pillow plugins
|
|
|
|
|
from PIL import Image, ExifTags
|
|
|
|
|
from PIL.PngImagePlugin import PngInfo
|
|
|
|
|
|
|
|
|
@@ -51,26 +55,44 @@ class SaveImageExtended:
|
|
|
|
|
OUTPUT_NODE = True
|
|
|
|
|
CATEGORY = 'image'
|
|
|
|
|
DESCRIPTION = """
|
|
|
|
|
### subfolders:
|
|
|
|
|
- you can use / or ./ or even ../ to start from parent
|
|
|
|
|
- you can also use / as the separator
|
|
|
|
|
- if your widget name has subfolders like SDXL/whatnot, you will have subfolders too
|
|
|
|
|
### Sample datetime formats: see [man datetime](https://www.man7.org/linux/man-pages/man1/date.1.html)
|
|
|
|
|
## Advice
|
|
|
|
|
It is strongly advised to enable Badge numbers in the Manager: _#ID Nickname_
|
|
|
|
|
|
|
|
|
|
### Default behavior
|
|
|
|
|
- Workflows with multiple of the same node:
|
|
|
|
|
- if you don't specify the node number in your keys like _13.sampler_name_,
|
|
|
|
|
- then the highest node number value will be returned
|
|
|
|
|
- if your checkpoint/controlnet is in subfolders like SDXL/whatnot, `ckpt_name` will **not** have subfolders
|
|
|
|
|
- to get the subfolders from controlnets and checkpoints, use _ckpt_path_ and _control_net_path_
|
|
|
|
|
|
|
|
|
|
### subfolders
|
|
|
|
|
- you can use `/` as the separator, or use keys such as `/name` or `name/` as a folder separator
|
|
|
|
|
- you can use `/` or `./` or even `../` anywhere in your keys
|
|
|
|
|
- there can be subfolders in _filename\_keys_ as well, altough it will mess the order of your keys
|
|
|
|
|
|
|
|
|
|
### Datetime formats
|
|
|
|
|
See [man datetime](https://www.man7.org/linux/man-pages/man1/date.1.html) for all possible values
|
|
|
|
|
- %F = %Y-%m-%d = 2024-05-22
|
|
|
|
|
- %H-%M-%S = 09-13-58
|
|
|
|
|
- %D = 05/22/24 (subfolders)
|
|
|
|
|
|
|
|
|
|
### 💾 Prompt Embedded
|
|
|
|
|
Prompt and Workflows are embedded in every files except BMP.
|
|
|
|
|
They are saved in Exif tags _Make_ [0x010f] and _ImageDescription_ [0x010e].
|
|
|
|
|
ComfyUI can only load PNG and WebP at the moment, AVIF is a PR still not merged.
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
|
|
|
|
|
type = 'output'
|
|
|
|
|
|
|
|
|
|
png_compress_level = 9
|
|
|
|
|
avif_quality = 60
|
|
|
|
|
webp_quality = 75
|
|
|
|
|
jpeg_quality = 91
|
|
|
|
|
jxl_quality = 91
|
|
|
|
|
webp_quality = 90
|
|
|
|
|
jpeg_quality = 90
|
|
|
|
|
jxl_quality = 90
|
|
|
|
|
j2k_quality = 90
|
|
|
|
|
# tiff_quality has no impact unless you start meddling with the compressions algorithms
|
|
|
|
|
tiff_quality = 91
|
|
|
|
|
tiff_quality = 90
|
|
|
|
|
# optimize_image only works for jpeg, with like just 2% reduction in size
|
|
|
|
|
optimize_image = True
|
|
|
|
|
|
|
|
|
@@ -90,23 +112,23 @@ class SaveImageExtended:
|
|
|
|
|
image_preview = True
|
|
|
|
|
extToRemove = ['.safetensors', '.ckpt', '.pt', '.bin', '.pth']
|
|
|
|
|
output_ext = '.webp'
|
|
|
|
|
output_exts = ['.webp', '.png', '.jpg', '.jpeg', '.gif', '.tiff', '.bmp']
|
|
|
|
|
quality = 75
|
|
|
|
|
output_exts = ['.webp', '.png', '.jpg', '.jpeg', '.j2k', '.jp2', '.gif', '.tiff', '.bmp']
|
|
|
|
|
quality = 90
|
|
|
|
|
|
|
|
|
|
print(f"\033[92m[save_image_extended]\033[0m version: {version}\033[0m")
|
|
|
|
|
print(f"\033[92m[💾 save_image_extended]\033[0m version: {version}\033[0m")
|
|
|
|
|
if jxl_supported:
|
|
|
|
|
output_exts.insert(0, '.jxl')
|
|
|
|
|
# if pillow_avif not in sys.modules:
|
|
|
|
|
if avif_supported:
|
|
|
|
|
# no matter what people say, jxl is far away from being integrated in browsers. I bet on AVIF.
|
|
|
|
|
output_exts.insert(0, '.avif')
|
|
|
|
|
|
|
|
|
|
def __init__(self):
|
|
|
|
|
self.output_dir = folder_paths.get_output_directory()
|
|
|
|
|
self.prefix_append = ''
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
"""
|
|
|
|
|
Return a dictionary which contains config for all input fields.
|
|
|
|
|
INPUT_TYPES Return a dictionary which contains config for all input fields.
|
|
|
|
|
Some types (string): "MODEL", "VAE", "CLIP", "CONDITIONING", "LATENT", "IMAGE", "INT", "STRING", "FLOAT".
|
|
|
|
|
Input types "INT", "STRING" or "FLOAT" are special values for fields on the node.
|
|
|
|
|
The type can be a list for selection.
|
|
|
|
@@ -121,6 +143,8 @@ class SaveImageExtended:
|
|
|
|
|
"""
|
|
|
|
|
@classmethod
|
|
|
|
|
def INPUT_TYPES(self):
|
|
|
|
|
# this is checked by ComfyUI/execution.py: validate_inputs(prompt, item, validated):
|
|
|
|
|
# must also define VALIDATE_INPUTS(self) so we can fill in missing inputs when new inputs are added by new versions
|
|
|
|
|
return {
|
|
|
|
|
'required': {
|
|
|
|
|
'images': ('IMAGE', ),
|
|
|
|
@@ -165,6 +189,25 @@ class SaveImageExtended:
|
|
|
|
|
'hidden': {'prompt': 'PROMPT', 'extra_pnginfo': 'EXTRA_PNGINFO'},
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
# This class serves no purpose, it can only test 1 element. you always get all errors even if only one element is bad
|
|
|
|
|
# @classmethod
|
|
|
|
|
# def VALIDATE_INPUTS(self, output_ext, quality, **kwargs):
|
|
|
|
|
# print(f"VALIDATE_INPUTS output_ext = x{output_ext}x")
|
|
|
|
|
# print(f"VALIDATE_INPUTS quality = x{quality}x")
|
|
|
|
|
# if output_ext == None or output_ext == '':
|
|
|
|
|
# return "cannot be empty"
|
|
|
|
|
# if output_ext not in self.output_exts:
|
|
|
|
|
# return "extension invalid"
|
|
|
|
|
# else:
|
|
|
|
|
# return True
|
|
|
|
|
# if quality == None or quality == '' or quality == 0:
|
|
|
|
|
# return "cannot be empty or 0"
|
|
|
|
|
# return True
|
|
|
|
|
|
|
|
|
|
# TODO: see how that works and how that can help
|
|
|
|
|
# @classmethod
|
|
|
|
|
# def IS_CHANGED(self, **kwargs):
|
|
|
|
|
# return float("nan")
|
|
|
|
|
|
|
|
|
|
def get_subfolder_path(self, image_path, output_path):
|
|
|
|
|
image_path = Path(image_path).resolve()
|
|
|
|
@@ -204,9 +247,19 @@ class SaveImageExtended:
|
|
|
|
|
|
|
|
|
|
# find_keys_recursively is a self-updating recursive method, that will update the dict found_values
|
|
|
|
|
def find_keys_recursively(self, prompt={}, keys_to_find=[], found_values={}):
|
|
|
|
|
# print(f"debug find_keys_recursively: keys_to_find={keys_to_find} found_values={found_values}")
|
|
|
|
|
for key, value in prompt.items():
|
|
|
|
|
if key in keys_to_find:
|
|
|
|
|
found_values[key] = value
|
|
|
|
|
# print(f"debug find_keys_recursively: found key={key}")
|
|
|
|
|
if key == 'ckpt_name' and 'ckpt_path' in keys_to_find:
|
|
|
|
|
value_path = Path(value)
|
|
|
|
|
found_values['ckpt_path'] = self.cleanup_fileName(str(value_path.parent))
|
|
|
|
|
# print(f"debug find_keys_recursively: ckpt_name={value_path.name} ckpt_path={str(value_path.parent)}")
|
|
|
|
|
elif key == 'control_net_name' and 'control_net_path' in keys_to_find:
|
|
|
|
|
value_path = Path(value)
|
|
|
|
|
found_values['control_net_path'] = self.cleanup_fileName(str(value_path.parent))
|
|
|
|
|
else:
|
|
|
|
|
found_values[key] = self.cleanup_fileName(value)
|
|
|
|
|
elif isinstance(value, dict):
|
|
|
|
|
self.find_keys_recursively(value, keys_to_find, found_values)
|
|
|
|
|
|
|
|
|
@@ -275,6 +328,7 @@ class SaveImageExtended:
|
|
|
|
|
# 'inputs': {'cfg': 1.6,
|
|
|
|
|
# 'denoise': 1.0, ...
|
|
|
|
|
for key in keys_to_extract:
|
|
|
|
|
# empty comma
|
|
|
|
|
if not key: continue
|
|
|
|
|
|
|
|
|
|
value = None
|
|
|
|
@@ -282,7 +336,7 @@ class SaveImageExtended:
|
|
|
|
|
|
|
|
|
|
# check if this is a subfolder: starts with ./ or /, can also end with /
|
|
|
|
|
# we also exclude datetime formats like "%A %d. %B %Y"
|
|
|
|
|
if '/' in key and not '%' in key:
|
|
|
|
|
if os.sep in key and not '%' in key:
|
|
|
|
|
# key is a subfolder: ./subfolder or ../subfolder or /subfolder
|
|
|
|
|
value = key
|
|
|
|
|
else:
|
|
|
|
@@ -299,13 +353,23 @@ class SaveImageExtended:
|
|
|
|
|
if node in prompt:
|
|
|
|
|
# print(f"debug generate_custom_name: --node.nodeKey = {node}.{nodeKey}")
|
|
|
|
|
# splitKey[0] = #node number found in prompt, we will recurse only in that node:
|
|
|
|
|
if nodeKey == 'ckpt_path':
|
|
|
|
|
self.find_keys_recursively(prompt[node], ['ckpt_name', nodeKey], found_values)
|
|
|
|
|
elif nodeKey == 'control_net_path':
|
|
|
|
|
self.find_keys_recursively(prompt[node], ['control_net_name', nodeKey], found_values)
|
|
|
|
|
else:
|
|
|
|
|
self.find_keys_recursively(prompt[node], [nodeKey], found_values)
|
|
|
|
|
else:
|
|
|
|
|
# if splitKey[0] = #num node not found in prompt; #num could have changed or user made a typo
|
|
|
|
|
# if splitKey[0] = #num node not found in prompt; #num could have changed or user made a typo. Fallback to normal key search
|
|
|
|
|
print(f"SaveImageExtended info: node #{node} not found")
|
|
|
|
|
if nodeKey == 'ckpt_path':
|
|
|
|
|
self.find_keys_recursively(prompt, ['ckpt_name', nodeKey], found_values)
|
|
|
|
|
elif nodeKey == 'control_net_path':
|
|
|
|
|
self.find_keys_recursively(prompt, ['control_net_name', nodeKey], found_values)
|
|
|
|
|
else:
|
|
|
|
|
self.find_keys_recursively(prompt, [nodeKey], found_values)
|
|
|
|
|
else:
|
|
|
|
|
# key is in the form string.string = fixed string
|
|
|
|
|
# key is in the form string.string = fixed string; still, we remove extra stuff and known extensions
|
|
|
|
|
value = self.cleanup_fileName(key)
|
|
|
|
|
else:
|
|
|
|
|
# key is in the form ".string" or "string." or "." - we won't clean that up and keep as is, maybe it's a separator
|
|
|
|
@@ -317,20 +381,24 @@ class SaveImageExtended:
|
|
|
|
|
# nodeKey is not a datetime, folder, has no dot, or multiple dots, could be a valid key to find, could be a fixed string - keep as is
|
|
|
|
|
else:
|
|
|
|
|
nodeKey = key
|
|
|
|
|
if nodeKey == 'ckpt_path':
|
|
|
|
|
self.find_keys_recursively(prompt, ['ckpt_name', nodeKey], found_values)
|
|
|
|
|
elif nodeKey == 'control_net_path':
|
|
|
|
|
self.find_keys_recursively(prompt, ['control_net_name', nodeKey], found_values)
|
|
|
|
|
else:
|
|
|
|
|
self.find_keys_recursively(prompt, [nodeKey], found_values)
|
|
|
|
|
# is key num.widget_name
|
|
|
|
|
# is key subfolder
|
|
|
|
|
|
|
|
|
|
# at this point we have a nodeKey, or a value, or both
|
|
|
|
|
# print(f"debug generate_custom_name: ----key: {nodeKey}")
|
|
|
|
|
# print(f"debug generate_custom_name: ----nodeKey: {nodeKey}")
|
|
|
|
|
# print(f"debug generate_custom_name: ----value: {value}")
|
|
|
|
|
# print(f"debug generate_custom_name: ----found_values: {found_values}")
|
|
|
|
|
if value is None:
|
|
|
|
|
if nodeKey is not None:
|
|
|
|
|
if nodeKey in found_values: value = found_values[nodeKey]
|
|
|
|
|
if value is None:
|
|
|
|
|
value = nodeKey
|
|
|
|
|
else:
|
|
|
|
|
value = self.cleanup_fileName(value)
|
|
|
|
|
|
|
|
|
|
# at this point, value is not None anymore
|
|
|
|
|
# now we analyze each value found and format them accordingly:
|
|
|
|
@@ -340,14 +408,32 @@ class SaveImageExtended:
|
|
|
|
|
# now we build the custom_name:
|
|
|
|
|
if isinstance(value, str):
|
|
|
|
|
# prefix and keys can very well be subfolders ending or starting with a /
|
|
|
|
|
if custom_name.endswith('/'):
|
|
|
|
|
if custom_name.endswith(os.sep):
|
|
|
|
|
# for subfolders, do not start filename with a delimiter...
|
|
|
|
|
delim = ''
|
|
|
|
|
else:
|
|
|
|
|
# for subfolders in keys, do not clean the filename...
|
|
|
|
|
if '/' in value and not value.endswith('/'):
|
|
|
|
|
# print(f"debug generate_custom_name: ---------: {os.sep} in value? {nodeKey}={value}")
|
|
|
|
|
if os.sep in value and not value.endswith(os.sep):
|
|
|
|
|
# print(f"debug generate_custom_name: ---------: folder")
|
|
|
|
|
delim= ''
|
|
|
|
|
|
|
|
|
|
# now process the custom cases ckpt_path and control_net_path; maybe we should do that to `image` as well?
|
|
|
|
|
# the recursive function creates ckpt_path and ckpt_name already; we just need to eliminate path if == '.'
|
|
|
|
|
if nodeKey in ['ckpt_path', 'control_net_path']:
|
|
|
|
|
# smth_path is most likely placed before smth_name, therefore value cannot be resolved during the previous check
|
|
|
|
|
value = found_values[nodeKey]
|
|
|
|
|
# and indeed we bypass anything that does not have a subfolder
|
|
|
|
|
if value == '.':
|
|
|
|
|
continue
|
|
|
|
|
else:
|
|
|
|
|
# terminate the subfolder with os.sep so it creates an actual subfolder.
|
|
|
|
|
# Because find_keys_recursively will only return the path part: 'path' or 'path/subpath'
|
|
|
|
|
# Therefore, enclosing this value with os.sep ensure we get a subfolder
|
|
|
|
|
value = os.sep + value + os.sep
|
|
|
|
|
else:
|
|
|
|
|
value = self.cleanup_fileName(value)
|
|
|
|
|
|
|
|
|
|
# ".string" case
|
|
|
|
|
if value.startswith('.'):
|
|
|
|
|
delim = ''
|
|
|
|
@@ -358,7 +444,7 @@ class SaveImageExtended:
|
|
|
|
|
custom_name += f"{delim}{value}"
|
|
|
|
|
# print(f"debug generate_custom_name: ------custom_name: {custom_name}")
|
|
|
|
|
# for each key
|
|
|
|
|
return custom_name.strip(delimiter).strip('.').strip('/').strip(delimiter)
|
|
|
|
|
return custom_name.strip(delimiter).strip('.').strip(os.sep).strip(delimiter)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def save_job_to_json(self, save_job_data, prompt, filename_prefix, positive_text_opt, negative_text_opt, job_custom_text, resolution, output_path, filename, timestamp=datetime.now()):
|
|
|
|
@@ -474,7 +560,7 @@ class SaveImageExtended:
|
|
|
|
|
metadata.update(extra_pnginfo)
|
|
|
|
|
|
|
|
|
|
## For Comfy to load an image, it need a PNG tag at the root: Prompt={prompt}
|
|
|
|
|
## For AVIF WebP Jpeg, it's only Exif that's available... and UserComment is the best choice.
|
|
|
|
|
## For AVIF WebP Jpeg JXL, it's only Exif that's available... and UserComment is the best choice.
|
|
|
|
|
|
|
|
|
|
## This method gives good results, as long as you save the prompt/workflow in 2 separate Exif tags
|
|
|
|
|
## Otherwise, [ExifTool] will issue Warning: Invalid EXIF text encoding for UserComment
|
|
|
|
@@ -534,34 +620,39 @@ class SaveImageExtended:
|
|
|
|
|
return exif_dat
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def save_image(self, image_path, img, prompt, save_metadata=save_metadata, extra_pnginfo=None, quality=75):
|
|
|
|
|
def save_image(self, image_path, img, prompt, save_metadata=save_metadata, extra_pnginfo=None, quality=quality):
|
|
|
|
|
# print(f"debug save_images: image_path={image_path}")
|
|
|
|
|
output_ext = os.path.splitext(os.path.basename(image_path))[1]
|
|
|
|
|
if quality == 0:
|
|
|
|
|
quality = self.quality
|
|
|
|
|
# output_ext = os.path.splitext(os.path.basename(image_path))[1]
|
|
|
|
|
output_ext = Path(image_path).suffix
|
|
|
|
|
metadata = None
|
|
|
|
|
kwargs = dict()
|
|
|
|
|
|
|
|
|
|
# TODO: see if convert_hdr_to_8bit=False make a change
|
|
|
|
|
# https://pillow.readthedocs.io/en/stable/handbook/image-file-formats.html
|
|
|
|
|
|
|
|
|
|
# match is python 3.10+
|
|
|
|
|
if output_ext in ['.avif']:
|
|
|
|
|
if output_ext in ['.avif', '.webp', '.jxl']:
|
|
|
|
|
if save_metadata: kwargs["exif"] = self.get_metadata_exif(img, prompt, extra_pnginfo)
|
|
|
|
|
if quality == 100:
|
|
|
|
|
kwargs["lossless"] = True
|
|
|
|
|
else:
|
|
|
|
|
kwargs["quality"] = quality
|
|
|
|
|
if quality == 100: kwargs["lossless"] = True
|
|
|
|
|
elif output_ext in ['.webp']:
|
|
|
|
|
if output_ext in ['.j2k', '.jp2', '.jpc', '.jpf', '.jpx', '.j2c']:
|
|
|
|
|
if save_metadata: kwargs["exif"] = self.get_metadata_exif(img, prompt, extra_pnginfo)
|
|
|
|
|
if quality < 100:
|
|
|
|
|
kwargs["irreversible"] = True
|
|
|
|
|
# there is no such thing as compression level in JPEG2000. Read https://comprimato.com/blog/2017/06/22/bitrate-control-quality-layers-jpeg2000/
|
|
|
|
|
# kwargs["quality_mode"] = 'rates' or 'dB'
|
|
|
|
|
# kwargs["quality_layers"] = [0,1,2] no refence online. i tried all values from 0 to 100 and no change in filesize
|
|
|
|
|
else:
|
|
|
|
|
kwargs["quality"] = quality
|
|
|
|
|
if quality == 100: kwargs["lossless"] = True
|
|
|
|
|
elif output_ext in ['.jpg', '.jpeg']:
|
|
|
|
|
if save_metadata: kwargs["exif"] = self.get_metadata_exif(img, prompt, extra_pnginfo)
|
|
|
|
|
# https://stackoverflow.com/questions/19303621/why-is-the-quality-of-jpeg-images-produced-by-pil-so-poor
|
|
|
|
|
kwargs["subsampling"] = 0
|
|
|
|
|
kwargs["quality"] = quality
|
|
|
|
|
kwargs["optimize"] = self.optimize_image
|
|
|
|
|
elif output_ext in ['.jxl']:
|
|
|
|
|
if save_metadata: kwargs["exif"] = self.get_metadata_exif(img, prompt, extra_pnginfo)
|
|
|
|
|
kwargs["quality"] = quality
|
|
|
|
|
if quality == 100: kwargs["lossless"] = True
|
|
|
|
|
elif output_ext in ['.tiff']:
|
|
|
|
|
# tiff: i suspect no quality either
|
|
|
|
|
kwargs["quality"] = quality
|
|
|
|
|
kwargs["optimize"] = self.optimize_image
|
|
|
|
|
elif output_ext in ['.png', '.gif']:
|
|
|
|
|
# png/gif: no quality
|
|
|
|
@@ -598,17 +689,21 @@ class SaveImageExtended:
|
|
|
|
|
positive_text_opt=None,
|
|
|
|
|
extra_pnginfo=None,
|
|
|
|
|
prompt=None,
|
|
|
|
|
quality=75,
|
|
|
|
|
quality=quality,
|
|
|
|
|
):
|
|
|
|
|
|
|
|
|
|
# print(f"filename_prefix = x{filename_prefix}x")
|
|
|
|
|
# print(f"filename_keys = x{filename_keys}x")
|
|
|
|
|
# print(f"foldername_prefix = x{foldername_prefix}x")
|
|
|
|
|
# print(f"foldername_keys = x{foldername_keys}x")
|
|
|
|
|
# print(f"delimiter = x{delimiter}x")
|
|
|
|
|
# print(f"save_job_data = x{save_job_data}x")
|
|
|
|
|
# print(f"job_data_per_image = x{job_data_per_image}x")
|
|
|
|
|
# print(f"output_ext = x{output_ext}x")
|
|
|
|
|
# print(f"save_images filename_prefix = x{filename_prefix}x")
|
|
|
|
|
# print(f"save_images filename_keys = x{filename_keys}x")
|
|
|
|
|
# print(f"save_images foldername_prefix = x{foldername_prefix}x")
|
|
|
|
|
# print(f"save_images foldername_keys = x{foldername_keys}x")
|
|
|
|
|
# print(f"save_images delimiter = x{delimiter}x")
|
|
|
|
|
# print(f"save_images save_job_data = x{save_job_data}x")
|
|
|
|
|
# print(f"save_images job_data_per_image = x{job_data_per_image}x")
|
|
|
|
|
# print(f"save_images output_ext = x{output_ext}x")
|
|
|
|
|
# print(f"save_images quality = x{quality}x")
|
|
|
|
|
|
|
|
|
|
# bugfix: sometimes on load, quality == 0
|
|
|
|
|
if quality == 0: quality = self.quality
|
|
|
|
|
|
|
|
|
|
# apply default values: we replicate the default save image box
|
|
|
|
|
if not filename_prefix and not filename_keys: filename_prefix=self.filename_prefix
|
|
|
|
@@ -621,8 +716,8 @@ class SaveImageExtended:
|
|
|
|
|
# pprint.pprint(prompt)
|
|
|
|
|
##########################################################################
|
|
|
|
|
timestamp = datetime.now()
|
|
|
|
|
custom_filename = self.generate_custom_name(filename_keys_to_extract, filename_prefix, delimiter, prompt, timestamp)
|
|
|
|
|
custom_foldername = self.generate_custom_name(foldername_keys_to_extract, foldername_prefix, delimiter, prompt, timestamp)
|
|
|
|
|
custom_filename = self.generate_custom_name(filename_keys_to_extract, filename_prefix, delimiter, prompt, timestamp)
|
|
|
|
|
|
|
|
|
|
# Get set resolution value
|
|
|
|
|
i = 255. * images[0].cpu().numpy()
|
|
|
|
@@ -631,11 +726,17 @@ class SaveImageExtended:
|
|
|
|
|
|
|
|
|
|
# Create folders, count images, save images
|
|
|
|
|
try:
|
|
|
|
|
full_output_folder, filename, _, _, custom_filename = folder_paths.get_save_image_path(custom_filename, self.output_dir, images[0].shape[1], images[0].shape[0])
|
|
|
|
|
output_path = os.path.join(full_output_folder, custom_foldername)
|
|
|
|
|
# print(f"debug save_images: full_output_folder={full_output_folder}")
|
|
|
|
|
# folder_paths.get_save_image_path() is kind of magic, I don't like that. Outpout is wrong anyway, when custom_filename contains folders
|
|
|
|
|
# full_output_folder, filename, _, _, custom_filename = folder_paths.get_save_image_path(custom_filename, self.output_dir, images[0].shape[1], images[0].shape[0])
|
|
|
|
|
# output_path = os.path.join(full_output_folder, custom_foldername)
|
|
|
|
|
|
|
|
|
|
output_path = Path(os.path.join(self.output_dir, custom_foldername, custom_filename)).parent
|
|
|
|
|
filename = Path(os.path.join(self.output_dir, custom_foldername, custom_filename)).name
|
|
|
|
|
# print(f"debug save_images: custom_foldername= {custom_foldername}")
|
|
|
|
|
# print(f"debug save_images: custom_filename= {custom_filename}")
|
|
|
|
|
# print(f"debug save_images: output_path= {output_path}")
|
|
|
|
|
# print(f"debug save_images: filename= {filename}")
|
|
|
|
|
|
|
|
|
|
os.makedirs(output_path, exist_ok=True)
|
|
|
|
|
counter = self.get_latest_counter(one_counter_per_folder, output_path, filename, counter_digits, counter_position, output_ext)
|
|
|
|
|
# print(f"debug save_images: counter for {output_ext}: {counter}")
|
|
|
|
@@ -646,18 +747,18 @@ class SaveImageExtended:
|
|
|
|
|
img = Image.fromarray(np.clip(i, 0, 255).astype(np.uint8))
|
|
|
|
|
|
|
|
|
|
if counter_position == 'last':
|
|
|
|
|
file = f'{filename}{delimiter}{counter:0{counter_digits}}{output_ext}'
|
|
|
|
|
image_name = f'{filename}{delimiter}{counter:0{counter_digits}}{output_ext}'
|
|
|
|
|
else:
|
|
|
|
|
file = f'{counter:0{counter_digits}}{delimiter}{filename}{output_ext}'
|
|
|
|
|
image_name = f'{counter:0{counter_digits}}{delimiter}{filename}{output_ext}'
|
|
|
|
|
|
|
|
|
|
image_path = os.path.join(output_path, file)
|
|
|
|
|
image_path = os.path.join(output_path, image_name)
|
|
|
|
|
self.save_image(image_path, img, prompt, save_metadata, extra_pnginfo, quality)
|
|
|
|
|
|
|
|
|
|
if save_job_data != 'disabled' and job_data_per_image:
|
|
|
|
|
self.save_job_to_json(save_job_data, prompt, filename_prefix, positive_text_opt, negative_text_opt, job_custom_text, resolution, output_path, f'{file.removesuffix(output_ext)}.json', timestamp)
|
|
|
|
|
self.save_job_to_json(save_job_data, prompt, filename_prefix, positive_text_opt, negative_text_opt, job_custom_text, resolution, output_path, f'{image_name.removesuffix(output_ext)}.json', timestamp)
|
|
|
|
|
|
|
|
|
|
subfolder = self.get_subfolder_path(image_path, self.output_dir)
|
|
|
|
|
results.append({ 'filename': file, 'subfolder': subfolder, 'type': self.type})
|
|
|
|
|
results.append({ 'filename': image_name, 'subfolder': subfolder, 'type': self.type})
|
|
|
|
|
counter += 1
|
|
|
|
|
|
|
|
|
|
if save_job_data != 'disabled' and not job_data_per_image:
|
|
|
|
|
audioscavenger