1
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-08-02 23:14:49 +08:00

rework payload from server

This commit is contained in:
m957ymj75urz
2023-03-19 12:54:29 +01:00
parent 5fa514fa51
commit d1138e8ba0
4 changed files with 15 additions and 13 deletions

View File

@@ -111,7 +111,7 @@ class PromptServer():
@routes.get("/view")
async def view_image(request):
if "file" in request.rel_url.query:
if "filename" in request.rel_url.query:
type = request.rel_url.query.get("type", "output")
if type not in ["output", "input", "temp"]:
return web.Response(status=400)
@@ -123,7 +123,7 @@ class PromptServer():
return web.Response(status=403)
output_dir = full_output_dir
file = request.rel_url.query["file"]
file = request.rel_url.query["filename"]
file = os.path.basename(file)
file = os.path.join(output_dir, file)