From 1a8c571646b4cfe8a680d1cc1fd856cfc41d2386 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E9=9B=AA=E5=B3=B0?= <lldancing@163.com>
Date: Thu, 11 Jan 2024 19:31:33 +0800
Subject: [PATCH] =?UTF-8?q?=E7=94=9F=E6=88=90=E7=9A=84=E5=9B=BE=E7=89=87?=
 =?UTF-8?q?=E5=85=83=E4=BF=A1=E6=81=AF=E4=B8=8D=E8=A6=81=E5=8C=85=E5=90=AB?=
 =?UTF-8?q?=E8=BE=93=E5=85=A5=E7=9A=84base64=E5=9B=BE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 easyapi/ImageNode.py |  4 ++++
 easyapi/logScript.py | 15 +++++++++++++++
 2 files changed, 19 insertions(+)

diff --git a/easyapi/ImageNode.py b/easyapi/ImageNode.py
index e1c6063..927d094 100644
--- a/easyapi/ImageNode.py
+++ b/easyapi/ImageNode.py
@@ -94,6 +94,10 @@ class ImageToBase64Advanced:
             if not args.disable_metadata:
                 metadata = PngInfo()
                 if prompt is not None:
+                    for idx in prompt:
+                        node = prompt[idx]
+                        if node['class_type'] == 'Base64ToImage' or node['class_type'] == 'Base64ToMask':
+                            node['inputs']['base64Images'] = ""
                     metadata.add_text("prompt", json.dumps(prompt))
                 if extra_pnginfo is not None:
                     for x in extra_pnginfo:
diff --git a/easyapi/logScript.py b/easyapi/logScript.py
index 04f56da..79a24bd 100644
--- a/easyapi/logScript.py
+++ b/easyapi/logScript.py
@@ -2,6 +2,8 @@ import sys
 
 from datetime import datetime as dt
 
+from server import PromptServer
+
 old_stdout = sys.stdout
 old_stderr = sys.stderr
 
@@ -38,6 +40,19 @@ class StdTimeFilter:
             old_stderr.flush()
 
 
+def socket_wrap(func):
+    def wrap_func(event, data, sid=None):
+        if event == "executed" or event == "executing":
+            print("send message begin, type={}, node={}".format(event, data['node']))
+        else:
+            print("send message begin, {}".format(event))
+        func(event, data, sid)
+        print("send message end, {}".format(event))
+    return wrap_func
+
+
 def log_wrap():
     sys.stdout = StdTimeFilter(True)
     sys.stderr = StdTimeFilter(False)
+    # old_send_sync = PromptServer.instance.send_sync
+    # PromptServer.instance.send_sync = socket_wrap(old_send_sync)