From c822e5c5175603c9388b81b67edf693f7da2cdac Mon Sep 17 00:00:00 2001
From: RyanOnTheInside <7623207+ryanontheinside@users.noreply.github.com>
Date: Tue, 22 Apr 2025 11:33:00 -0400
Subject: [PATCH] add routes

---
 __init__.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/__init__.py b/__init__.py
index 2bbc2a8..ff3cc06 100644
--- a/__init__.py
+++ b/__init__.py
@@ -114,6 +114,15 @@ for node_name in NODE_CLASS_MAPPINGS.keys():
     
     # Assign the final display name to the mappings
     NODE_DISPLAY_NAME_MAPPINGS[node_name] = display_name
+
+# Import server routes
+try:
+    from . import server
+    print("\033[92m" + "[ComfyUI_RealtimeNodes] API routes registered.")
+except Exception as e:
+    print("\033[91m" + f"[ComfyUI_RealtimeNodes] Error registering API routes: {e}")
+    traceback.print_exc()
+
 # --- Original Web Directory and Export ---
 WEB_DIRECTORY = "./web/js" # Adjusted path if needed