xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-09 11:20:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

encode image to be sent with json

Browse Source
This commit is contained in:
RandomGitUser321 2024-09-20 02:21:15 -04:00 committed by GitHub
parent 63f0910eea
commit 98cffee8b3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
server.py
View File

@ -2,6 +2,7 @@ import os
import sys import sys
import asyncio import asyncio
import traceback import traceback
import base64
import nodes import nodes
import folder_paths import folder_paths
@ -741,6 +742,12 @@ class PromptServer():
async def send(self, event, data, sid=None): async def send(self, event, data, sid=None):
if event == BinaryEventTypes.UNENCODED_PREVIEW_IMAGE: if event == BinaryEventTypes.UNENCODED_PREVIEW_IMAGE:
await self.send_image(data, sid=sid) await self.send_image(data, sid=sid)
elif event == "latent_preview":
img = data[1]
buffered = BytesIO()
img.save(buffered, format="JPEG")
img_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
await self.send_json("latent_preview", img_base64, sid=sid)
elif isinstance(data, (bytes, bytearray)): elif isinstance(data, (bytes, bytearray)):
await self.send_bytes(event, data, sid) await self.send_bytes(event, data, sid)
else: else: