Add content disposition header so images have name

3c25f5dc · pythongosssss · f67c0062 · 3c25f5dc
Commit 3c25f5dc authored Mar 22, 2023 by pythongosssss
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)