mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-04-06 00:36:49 +08:00
d5578efa2b
The /view endpoint returns text/plain for .svg files on some platforms because Python's mimetypes module does not always include SVG by default. Explicitly register image/svg+xml so <img> tags can render SVGs correctly. Amp-Thread-ID: https://ampcode.com/threads/T-019d2da7-6a64-726a-af91-bd9c44e7f43c Co-authored-by: Amp <amp@ampcode.com>
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
"""Centralized MIME type initialization.
|
|
|
|
Call init_mime_types() once at startup to initialize the MIME type database
|
|
and register all custom types used across ComfyUI.
|
|
"""
|
|
|
|
import mimetypes
|
|
|
|
_initialized = False
|
|
|
|
|
|
def init_mime_types():
|
|
"""Initialize the MIME type database and register all custom types.
|
|
|
|
Safe to call multiple times; only runs once.
|
|
"""
|
|
global _initialized
|
|
if _initialized:
|
|
return
|
|
_initialized = True
|
|
|
|
mimetypes.init()
|
|
|
|
# Web types (used by server.py for static file serving)
|
|
mimetypes.add_type('application/javascript; charset=utf-8', '.js')
|
|
mimetypes.add_type('image/webp', '.webp')
|
|
mimetypes.add_type('image/svg+xml', '.svg')
|
|
|
|
# Model and data file types (used by asset scanning / metadata extraction)
|
|
mimetypes.add_type("application/safetensors", ".safetensors")
|
|
mimetypes.add_type("application/safetensors", ".sft")
|
|
mimetypes.add_type("application/pytorch", ".pt")
|
|
mimetypes.add_type("application/pytorch", ".pth")
|
|
mimetypes.add_type("application/pickle", ".ckpt")
|
|
mimetypes.add_type("application/pickle", ".pkl")
|
|
mimetypes.add_type("application/gguf", ".gguf")
|
|
mimetypes.add_type("application/yaml", ".yaml")
|
|
mimetypes.add_type("application/yaml", ".yml")
|