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

Added filename to right-click open preview.

Browse Source
This commit is contained in:
Christian Bastian
2024-04-03 23:53:56 -04:00
parent 9484344467
commit 1b5713ab2d
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
__init__.py
View File

@@ -277,6 +277,7 @@ async def get_model_preview(request):
uri = request.query.get("uri") uri = request.query.get("uri")
image_path = no_preview_image image_path = no_preview_image
image_type = "png" image_type = "png"
file_name = os.path.split(no_preview_image)[1]
if uri != "no-preview": if uri != "no-preview":
sep = os.path.sep sep = os.path.sep
uri = uri.replace("/" if sep == "\\" else "/", sep) uri = uri.replace("/" if sep == "\\" else "/", sep)
@@ -285,9 +286,11 @@ async def get_model_preview(request):
if os.path.exists(path): if os.path.exists(path):
image_path = path image_path = path
image_type = extension.rsplit(".", 1)[1] image_type = extension.rsplit(".", 1)[1]
file_name = os.path.split(head)[1] + "." + image_type
elif os.path.exists(head) and head.endswith(".safetensors"): elif os.path.exists(head) and head.endswith(".safetensors"):
image_path = head image_path = head
image_type = extension.rsplit(".", 1)[1] image_type = extension.rsplit(".", 1)[1]
file_name = os.path.splitext(os.path.split(head)[1])[0] + "." + image_type
w = request.query.get("width") w = request.query.get("width")
h = request.query.get("height") h = request.query.get("height")
@@ -340,7 +343,13 @@ async def get_model_preview(request):
image.save(image_bytes, format=image.format, exif=exif, pnginfo=metadata) image.save(image_bytes, format=image.format, exif=exif, pnginfo=metadata)
image_data = image_bytes.getvalue() image_data = image_bytes.getvalue()
return web.Response(body=image_data, content_type="image/" + image_type) return web.Response(
headers={
"Content-Disposition": f"inline; filename={file_name}",
},
body=image_data,
content_type="image/" + image_type,
)
@server.PromptServer.instance.routes.get("/model-manager/image/extensions") @server.PromptServer.instance.routes.get("/model-manager/image/extensions")