Correct the Content-Disposition Header for preview image, in order to download as corrent filename

To avoid download a webp preview image with a '.png' filename
This commit is contained in:
Maikaze 2024-11-27 02:23:22 +08:00 committed by GitHub
parent 4c82741b54
commit 8fe0e5391e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -420,7 +420,7 @@ class PromptServer():
buffer.seek(0)
return web.Response(body=buffer.read(), content_type=f'image/{image_format}',
headers={"Content-Disposition": f"filename=\"{filename}\""})
headers={"Content-Disposition": f"filename=\"{filename.replace('png', image_format)}\""})
if 'channel' not in request.rel_url.query:
channel = 'rgba'