298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-04-20 03:13:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update server.py

Browse Source
This commit is contained in:
Maruhi 2024-12-25 20:39:06 +09:00 committed by GitHub
parent 19a64d6291
commit 61ad962154
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 25 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
server.py
View File

@ -327,15 +327,37 @@ class PromptServer():
else:
return web.Response(status=400)
async def get_post_from_request(request):
post = None
if request.content_type == 'application/json':
json_data = await request.json()
post = await request.post()
mutable_post = post.copy()
for key, value in json_data.items():
if key == "image":
file_name = json_data.get("filename")
value = base64.b64decode(value)
import io
image = aiohttp.web.FileField(name=key, filename=file_name, file=io.BytesIO(value), content_type='image/png', headers=None)
mutable_post[key] = image
elif key != "filename":
mutable_post[key] = value
post = mutable_post
elif request.content_type == 'multipart/form-data':
post = await request.post()
else:
raise web.HTTPBadRequest(text="Unsupported content type")
return post
@routes.post("/upload/image")
async def upload_image(request):
post = await request.post()
post = await get_post_from_request(request)
return image_upload(post)
@routes.post("/upload/mask")
async def upload_mask(request):
post = await request.post()
post = await get_post_from_request(request)
def image_save_function(image, post, filepath):
original_ref = json.loads(post.get("original_ref"))