From 3957839ff145fe226cc132b618f1954301c09797 Mon Sep 17 00:00:00 2001
From: Christian Byrne <cbyrne@comfy.org>
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 f64ec27d4..33112c227 100644
--- a/server.py
+++ b/server.py
@@ -416,6 +416,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]