How to add custom routes outside server.py

[CodingOctocat]

Every time I need to add a new route, I have to modify server.py. Does ComfyUI support loading external files? For example, can I add a my_routes.py file?

[CodingOctocat]

I found the answer.

📂 ComfyUI/
└── 📂 custom_nodes/
    ├── 📝 __init__.py
    ├── 📝 your_routes1.py
    ├── 📝 your_routes2.py
    └── ...

# __init__.py

import os
import importlib
from pathlib import Path

current_dir = Path(__file__).parent

for file in current_dir.glob("*.py"):
    if file.name != "__init__.py":
        module_name = file.stem
        importlib.import_module(f".{module_name}", package=__package__)

NODE_CLASS_MAPPINGS = {}
NODE_DISPLAY_NAME_MAPPINGS = {}

# your_routes.py

import os
import asyncio

from aiohttp import web
import folder_paths
import server

import logging

logger = logging.getLogger(__name__)

routes = server.PromptServer.instance.routes

@routes.get("/your_api")
async def your_api(request):
    pass

Then, simply restart ComfyUI.