TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix image filename to match actual image type.

Browse Source
This commit is contained in:
Christian Bastian
2024-07-19 23:17:16 -04:00
parent 2cdb62a587
commit 823903a38f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
__init__.py
View File

@@ -393,9 +393,10 @@ async def get_model_preview(request):
image.save(image_bytes, format=response_image_format, exif=exif, pnginfo=metadata, quality=quality) image.save(image_bytes, format=response_image_format, exif=exif, pnginfo=metadata, quality=quality)
image_data = image_bytes.getvalue() image_data = image_bytes.getvalue()
response_file_name = os.path.splitext(file_name)[0] + '.' + response_image_format.lower()
return web.Response( return web.Response(
headers={ headers={
"Content-Disposition": f"inline; filename={file_name}", "Content-Disposition": f"inline; filename={response_file_name}",
}, },
body=image_data, body=image_data,
content_type="image/" + response_image_format.lower(), content_type="image/" + response_image_format.lower(),