Commit f4b93906 authored by comfyanonymous's avatar comfyanonymous
Browse files

Add a random string to the temp prefix for PreviewImage.

parent 2b2a1474
...@@ -7,6 +7,7 @@ import hashlib ...@@ -7,6 +7,7 @@ import hashlib
import traceback import traceback
import math import math
import time import time
import random
from PIL import Image, ImageOps from PIL import Image, ImageOps
from PIL.PngImagePlugin import PngInfo from PIL.PngImagePlugin import PngInfo
...@@ -1116,6 +1117,7 @@ class SaveImage: ...@@ -1116,6 +1117,7 @@ class SaveImage:
def __init__(self): def __init__(self):
self.output_dir = folder_paths.get_output_directory() self.output_dir = folder_paths.get_output_directory()
self.type = "output" self.type = "output"
self.prefix_append = ""
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
...@@ -1133,6 +1135,7 @@ class SaveImage: ...@@ -1133,6 +1135,7 @@ class SaveImage:
CATEGORY = "image" CATEGORY = "image"
def save_images(self, images, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=None): def save_images(self, images, filename_prefix="ComfyUI", prompt=None, extra_pnginfo=None):
filename_prefix += self.prefix_append
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir, images[0].shape[1], images[0].shape[0]) full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, self.output_dir, images[0].shape[1], images[0].shape[0])
results = list() results = list()
for image in images: for image in images:
...@@ -1160,6 +1163,7 @@ class PreviewImage(SaveImage): ...@@ -1160,6 +1163,7 @@ class PreviewImage(SaveImage):
def __init__(self): def __init__(self):
self.output_dir = folder_paths.get_temp_directory() self.output_dir = folder_paths.get_temp_directory()
self.type = "temp" self.type = "temp"
self.prefix_append = "_temp_" + ''.join(random.choice("abcdefghijklmnopqrstupvxyz") for x in range(5))
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment