Merge branch 'image-content-disposition-filename' of https://github.com/pythongosssss/ComfyUI

8a3d2067 · comfyanonymous · a8aa623a · 3c25f5dc · 8a3d2067
Commit 8a3d2067 authored Mar 22, 2023 by comfyanonymous
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 4 deletions

server.py server.py +4 -4

No files found.
--- a/server.py
+++ b/server.py
@@ -129,12 +129,12 @@ class PromptServer():
                        return web.Response(status=403)
                    output_dir = full_output_dir
-                file = request.rel_url.query["filename"]
+                filename = request.rel_url.query["filename"]
-                file = os.path.basename(file)
+                filename = os.path.basename(filename)
-                file = os.path.join(output_dir, file)
+                file = os.path.join(output_dir, filename)
                if os.path.isfile(file):
-                    return web.FileResponse(file)
+                    return web.FileResponse(file, headers={"Content-Disposition": f"filename=\"{filename}\""})
            return web.Response(status=404)