From 01ab8f4ac2a22b2804b03d14b89a5fb8f9d1df1f Mon Sep 17 00:00:00 2001
From: yolain <yolain@163.com>
Date: Tue, 4 Feb 2025 12:32:51 +0800
Subject: [PATCH] Renamed api.py to routes.py

---
 __init__.py              |  2 +-
 py/{api.py => routes.py} | 11 ++---------
 2 files changed, 3 insertions(+), 10 deletions(-)
 rename py/{api.py => routes.py} (97%)

diff --git a/__init__.py b/__init__.py
index 7c6235a..bac8b1d 100644
--- a/__init__.py
+++ b/__init__.py
@@ -8,7 +8,7 @@ import importlib
 NODE_CLASS_MAPPINGS = {}
 NODE_DISPLAY_NAME_MAPPINGS = {}
 
-importlib.import_module('.py.api', __name__)
+importlib.import_module('.py.routes', __name__)
 importlib.import_module('.py.server', __name__)
 nodes_list = ["util", "seed", "prompt", "loaders", "adapter", "inpaint", "preSampling", "samplers", "fix", "pipe", "xyplot", "image", "logic", "api", "deprecated"]
 for module_name in nodes_list:
diff --git a/py/api.py b/py/routes.py
similarity index 97%
rename from py/api.py
rename to py/routes.py
index e577550..ef9b92c 100644
--- a/py/api.py
+++ b/py/routes.py
@@ -4,7 +4,7 @@ import sys
 import json
 import shutil
 import folder_paths
-from folder_paths import get_directory_by_type
+from aiohttp import web
 from server import PromptServer
 from .config import RESOURCES_DIR, FOOOCUS_STYLES_DIR, FOOOCUS_STYLES_SAMPLES
 from .libs.model import easyModelManager
@@ -12,13 +12,6 @@ from .libs.utils import getMetadata, cleanGPUUsedForce, get_local_filepath
 from .libs.cache import remove_cache
 from .libs.translate import has_chinese, zh_to_en
 
-try:
-    import aiohttp
-    from aiohttp import web
-except ImportError:
-    print("Module 'aiohttp' not installed. Please install it via:")
-    print("pip install aiohttp")
-    sys.exit()
 
 @PromptServer.instance.routes.post("/easyuse/cleangpu")
 def cleanGPU(request):
@@ -268,7 +261,7 @@ async def save_preview(request):
 
     body = await request.json()
 
-    dir = get_directory_by_type(body.get("type", "output"))
+    dir = folder_paths.get_directory_by_type(body.get("type", "output"))
     subfolder = body.get("subfolder", "")
     full_output_folder = os.path.join(dir, os.path.normpath(subfolder))