From 61c26e496d8b098ed4c47b93495f95b2c325660b Mon Sep 17 00:00:00 2001
From: VipCrazyMaster <vipcrazydavid@gmail.com>
Date: Fri, 19 Dec 2025 22:04:54 +0800
Subject: [PATCH] Fix BytesIO import error caused by comfy_api.latest.io

---
 nodes/curve_nodes.py | 6 ++++--
 nodes/image_nodes.py | 4 +++-
 nodes/nodes.py       | 3 ++-
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/nodes/curve_nodes.py b/nodes/curve_nodes.py
index ceea047..ed9fef3 100644
--- a/nodes/curve_nodes.py
+++ b/nodes/curve_nodes.py
@@ -7,6 +7,8 @@ from ..utility.utility import pil2tensor, tensor2pil
 import folder_paths
 import io
 import base64
+from io import BytesIO
+
 
 def parse_color(color):
     if isinstance(color, str) and ',' in color:
@@ -309,7 +311,7 @@ output types:
         else:
             transform = transforms.ToPILImage()
             image = transform(bg_image[0].permute(2, 0, 1))
-            buffered = io.BytesIO()
+            buffered = BytesIO()
             image.save(buffered, format="JPEG", quality=75)
 
             # 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:
             transform = transforms.ToPILImage()
             image = transform(bg_image[0].permute(2, 0, 1))
-            buffered = io.BytesIO()
+            buffered = BytesIO()
             image.save(buffered, format="JPEG", quality=75)
 
             # Step 3: Encode the image bytes to a Base64 string
diff --git a/nodes/image_nodes.py b/nodes/image_nodes.py
index 320a953..aac70d4 100644
--- a/nodes/image_nodes.py
+++ b/nodes/image_nodes.py
@@ -12,6 +12,8 @@ import re
 import json
 import importlib
 from PIL.PngImagePlugin import PngInfo
+from io import BytesIO
+
 try:
     import cv2
 except:
@@ -3378,7 +3380,7 @@ class FastPreview:
     def preview(self, image, format, quality):        
         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)
             img_bytes = buffered.getvalue()
 
diff --git a/nodes/nodes.py b/nodes/nodes.py
index 25397f2..06ea1cf 100644
--- a/nodes/nodes.py
+++ b/nodes/nodes.py
@@ -11,6 +11,7 @@ from nodes import MAX_RESOLUTION
 from comfy.utils import common_upscale, ProgressBar, load_torch_file
 from comfy.comfy_types.node_typing import IO
 from comfy_api.latest import io
+from io import BytesIO
 
 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"))
@@ -1632,7 +1633,7 @@ or a .txt file with RealEstate camera intrinsics and coordinates, in a 3D plot.
         
         plt.title('')
         plt.draw()
-        buf = io.BytesIO()
+        buf = BytesIO()
         plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0)
         buf.seek(0)
         img = Image.open(buf)