From 491c020f4502a5b21853a45d1dab8b1b6b196b87 Mon Sep 17 00:00:00 2001 From: Christian Byrne Date: Thu, 1 May 2025 20:01:41 -0700 Subject: [PATCH] Serve SVG files directly (#107) --- server.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/server.py b/server.py index 1b0a73601..973c9d2fe 100644 --- a/server.py +++ b/server.py @@ -417,6 +417,17 @@ class PromptServer(): if os.path.isfile(file): if 'preview' in request.rel_url.query: + extension = os.path.splitext(filename)[-1] + if extension and extension.lower() == ".svg": + with open(file, "r") as f: + return web.Response( + body=f.read(), + content_type="image/svg+xml", + headers={ + "Content-Disposition": f'filename="{filename}"' + }, + ) + with Image.open(file) as img: preview_info = request.rel_url.query['preview'].split(';') image_format = preview_info[0]