298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-25 15:55:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

import location change, to shorter code

Browse Source
This commit is contained in:
Pupbani 2025-01-02 09:21:13 +09:00
parent fa2c56ac60
commit eeb20ac9d8
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
server.py
View File

@ -14,6 +14,7 @@ import struct
import ssl import ssl
import socket import socket
import ipaddress import ipaddress
import io,base64
from PIL import Image, ImageOps from PIL import Image, ImageOps
from PIL.PngImagePlugin import PngInfo from PIL.PngImagePlugin import PngInfo
from io import BytesIO from io import BytesIO
@ -333,7 +334,6 @@ class PromptServer():
""" """
Convert Image Byte Stream to PIL Image Convert Image Byte Stream to PIL Image
""" """
import io,base64
img_data = base64.b64decode(img_str) img_data = base64.b64decode(img_str)
img_io = io.BytesIO(img_data) img_io = io.BytesIO(img_data)
@ -353,7 +353,7 @@ class PromptServer():
image = str2pil(post.get("img_str")) image = str2pil(post.get("img_str"))
file_name = post.get("file_name") file_name = post.get("file_name")
if not(file_name.endswith(".png") or file_name.endswith(".jpg")): if not file_name.endswith((".png",".jpg")):
return web.Response(text="The file name must end in .jpg or .png.", status=400) return web.Response(text="The file name must end in .jpg or .png.", status=400)
# PATH # PATH