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)