Commit 8ea165dd authored by comfyanonymous's avatar comfyanonymous
Browse files

Add a way to overwrite images when uploading.

parent dfc74c19
...@@ -127,6 +127,7 @@ class PromptServer(): ...@@ -127,6 +127,7 @@ class PromptServer():
def image_upload(post, image_save_function=None): def image_upload(post, image_save_function=None):
image = post.get("image") image = post.get("image")
overwrite = post.get("overwrite")
image_upload_type = post.get("type") image_upload_type = post.get("type")
upload_dir = get_dir_by_type(image_upload_type) upload_dir = get_dir_by_type(image_upload_type)
...@@ -148,6 +149,9 @@ class PromptServer(): ...@@ -148,6 +149,9 @@ class PromptServer():
split = os.path.splitext(filename) split = os.path.splitext(filename)
filepath = os.path.join(full_output_folder, filename) filepath = os.path.join(full_output_folder, filename)
if overwrite is not None and (overwrite == "true" or overwrite == "1"):
pass
else:
i = 1 i = 1
while os.path.exists(filepath): while os.path.exists(filepath):
filename = f"{split[0]} ({i}){split[1]}" filename = f"{split[0]} ({i}){split[1]}"
......
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