xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-05-24 00:49:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #466 from VipCrazyMaster/fix-bytesio-import

Browse Source 
Fix BytesIO import error caused by comfy_api.latest.io
This commit is contained in:
Jukka Seppänen 2025-12-19 19:36:07 +02:00 committed by GitHub
commit ac2da893e9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes/curve_nodes.py
View File

@ -7,6 +7,8 @@ from ..utility.utility import pil2tensor, tensor2pil
import folder_paths import folder_paths
import io import io
import base64 import base64
from io import BytesIO
def parse_color(color): def parse_color(color):
if isinstance(color, str) and ',' in color: if isinstance(color, str) and ',' in color:
@ -309,7 +311,7 @@ output types:
else: else:
transform = transforms.ToPILImage() transform = transforms.ToPILImage()
image = transform(bg_image[0].permute(2, 0, 1)) image = transform(bg_image[0].permute(2, 0, 1))
buffered = io.BytesIO() buffered = BytesIO()
image.save(buffered, format="JPEG", quality=75) image.save(buffered, format="JPEG", quality=75)
# Encode the image bytes to a Base64 string # Encode the image bytes to a Base64 string
@ -1526,7 +1528,7 @@ you can clear the image from the context menu by right clicking on the canvas
else: else:
transform = transforms.ToPILImage() transform = transforms.ToPILImage()
image = transform(bg_image[0].permute(2, 0, 1)) image = transform(bg_image[0].permute(2, 0, 1))
buffered = io.BytesIO() buffered = BytesIO()
image.save(buffered, format="JPEG", quality=75) image.save(buffered, format="JPEG", quality=75)
# Step 3: Encode the image bytes to a Base64 string # Step 3: Encode the image bytes to a Base64 string

4
nodes/image_nodes.py
View File

@ -12,6 +12,8 @@ import re
import json import json
import importlib import importlib
from PIL.PngImagePlugin import PngInfo from PIL.PngImagePlugin import PngInfo
from io import BytesIO
try: try:
import cv2 import cv2
except: except:
@ -3378,7 +3380,7 @@ class FastPreview:
def preview(self, image, format, quality): def preview(self, image, format, quality):
pil_image = to_pil_image(image[0].permute(2, 0, 1)) pil_image = to_pil_image(image[0].permute(2, 0, 1))
with io.BytesIO() as buffered: with BytesIO() as buffered:
pil_image.save(buffered, format=format, quality=quality) pil_image.save(buffered, format=format, quality=quality)
img_bytes = buffered.getvalue() img_bytes = buffered.getvalue()

3
nodes/nodes.py
View File

@ -11,6 +11,7 @@ from nodes import MAX_RESOLUTION
from comfy.utils import common_upscale, ProgressBar, load_torch_file from comfy.utils import common_upscale, ProgressBar, load_torch_file
from comfy.comfy_types.node_typing import IO from comfy.comfy_types.node_typing import IO
from comfy_api.latest import io from comfy_api.latest import io
from io import BytesIO
script_directory = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) script_directory = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
folder_paths.add_model_folder_path("kjnodes_fonts", os.path.join(script_directory, "fonts")) folder_paths.add_model_folder_path("kjnodes_fonts", os.path.join(script_directory, "fonts"))
@ -1632,7 +1633,7 @@ or a .txt file with RealEstate camera intrinsics and coordinates, in a 3D plot.
plt.title('') plt.title('')
plt.draw() plt.draw()
buf = io.BytesIO() buf = BytesIO()
plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0) plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0)
buf.seek(0) buf.seek(0)
img = Image.open(buf) img = Image.open(buf)