From b92df4241cc6ddb5c7dedb981bbfecabf252e133 Mon Sep 17 00:00:00 2001
From: bilt <vanderbilt@gmail.com>
Date: Wed, 20 Aug 2025 12:24:47 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8=20PIL=20=E6=89=93=E5=BC=80?=
 =?UTF-8?q?=E4=B8=8A=E4=BC=A0=E7=9A=84=E5=9B=BE=E7=89=87=E6=B5=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 server.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/server.py b/server.py
index ec34a17ae..53b11872b 100644
--- a/server.py
+++ b/server.py
@@ -329,8 +329,10 @@ class PromptServer():
                         image_save_function(image, post, filepath)
                     else:
                         print("image_save_function is None")
-                        with open(filepath, "wb") as f:
-                            f.write(image.file.read())
+                        # 使用 PIL 打开上传的图片流
+                        img = PILImage.open(image.file)
+                        # 使用被你的插件修改过的 save 方法进行保存，这将触发加密
+                        img.save(filepath)
 
                 return web.json_response({"name" : filename, "subfolder": subfolder, "type": image_upload_type})
             else: