mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-11 23:00:51 +08:00
a38968f098
- Validation errors that occur early in the lifecycle of prompt execution now get propagated to their callers in the EmbeddedComfyClient. This includes error messages about missing node classes. - The execution context now includes the node_id and the prompt_id - Latent previews are now sent with a node_id. This is not backwards compatible with old frontends. - Dependency execution errors are now modeled correctly. - Distributed progress encodes image previews with node and prompt IDs. - Typing for models - The frontend was updated to use node IDs with previews - Improvements to torch.compile experiments - Some controlnet_aux nodes were upstreamed
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from __future__ import annotations
|
|
|
|
import struct
|
|
from io import BytesIO
|
|
from typing import Literal
|
|
|
|
import PIL.Image
|
|
from PIL import Image, ImageOps
|
|
|
|
|
|
def encode_preview_image(image: PIL.Image.Image, image_type: Literal["JPEG", "PNG"], max_size: int, node_id: str = "", task_id: str = "") -> bytes:
|
|
if max_size is not None:
|
|
if hasattr(Image, 'Resampling'):
|
|
resampling = Image.Resampling.BILINEAR
|
|
else:
|
|
resampling = Image.LANCZOS
|
|
|
|
image = ImageOps.contain(image, (max_size, max_size), resampling)
|
|
|
|
has_ids = (node_id is not None and len(node_id) > 0) or (task_id is not None and len(task_id) > 0)
|
|
|
|
if image_type == "JPEG":
|
|
type_num = 3 if has_ids else 1
|
|
elif image_type == "PNG":
|
|
type_num = 4 if has_ids else 2
|
|
else:
|
|
raise ValueError(f"Unsupported image type: {image_type}")
|
|
|
|
bytesIO = BytesIO()
|
|
|
|
if has_ids:
|
|
# Pack the header with type_num, node_id length, task_id length
|
|
node_id = node_id or ""
|
|
task_id = task_id or ""
|
|
header = struct.pack(">III", type_num, len(node_id), len(task_id))
|
|
bytesIO.write(header)
|
|
bytesIO.write(node_id.encode('utf-8'))
|
|
bytesIO.write(task_id.encode('utf-8'))
|
|
else:
|
|
# Pack only the type_num for types 1 and 2
|
|
header = struct.pack(">I", type_num)
|
|
bytesIO.write(header)
|
|
|
|
image.save(bytesIO, format=image_type, quality=95, compress_level=1)
|
|
preview_bytes = bytesIO.getvalue()
|
|
return preview_bytes
|