wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-25 05:40:15 +08:00
Code Issues Actions 16 Packages Projects Releases Wiki Activity

Fix mask uploads

Browse Source
This commit is contained in:
doctorpangloss 2025-01-28 16:34:59 -08:00
parent 044dff6887
commit d24098cd9b
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
comfy/cmd/server.py
View File

@ -2,6 +2,7 @@ from __future__ import annotations
import asyncio import asyncio
import glob import glob
import io
import ipaddress import ipaddress
import json import json
import logging import logging
@ -343,7 +344,7 @@ class PromptServer(ExecutorToClientProgress):
if not image_is_duplicate: if not image_is_duplicate:
if image_save_function is not None: if image_save_function is not None:
image_save_function(image, post, filepath) await image_save_function(image, post, filepath)
else: else:
async with aiofiles.open(filepath, mode='wb') as file: async with aiofiles.open(filepath, mode='wb') as file:
await file.write(image.file.read()) await file.write(image.file.read())
@ -361,7 +362,7 @@ class PromptServer(ExecutorToClientProgress):
async def upload_mask(request): async def upload_mask(request):
post = await request.post() post = await request.post()
def image_save_function(image, post, filepath): async def image_save_function(image, post, filepath):
original_ref = json.loads(post.get("original_ref")) original_ref = json.loads(post.get("original_ref"))
filename, output_dir = folder_paths.annotated_filepath(original_ref['filename']) filename, output_dir = folder_paths.annotated_filepath(original_ref['filename'])
@ -399,9 +400,14 @@ class PromptServer(ExecutorToClientProgress):
# alpha copy # alpha copy
new_alpha = mask_pil.getchannel('A') new_alpha = mask_pil.getchannel('A')
original_pil.putalpha(new_alpha) original_pil.putalpha(new_alpha)
original_pil.save(filepath, compress_level=4, pnginfo=metadata)
return image_upload(post, image_save_function) output_buffer = io.BytesIO()
original_pil.save(output_buffer, format='PNG', compress_level=4, pnginfo=metadata)
output_data = output_buffer.getvalue()
async with aiofiles.open(filepath, mode='wb') as file_dest:
await file_dest.write(output_data)
return await image_upload(post, image_save_function)
@routes.get("/view") @routes.get("/view")
async def view_image(request): async def view_image(request):