diff --git a/__init__.py b/__init__.py index 587ecba..016ae40 100644 --- a/__init__.py +++ b/__init__.py @@ -57,6 +57,20 @@ async def update_exclude_modules(request): return web.json_response({"status": "success"}) except Exception as e: return web.json_response({"status": "error", "message": str(e)}, status=500) + +@PromptServer.instance.routes.get("/hotreload/get_all_modules") +async def get_all_modules(request): + try: + modules = [] + for item in os.listdir(CUSTOM_NODE_ROOT[0]): + item_path = os.path.join(CUSTOM_NODE_ROOT[0], item) + if os.path.isdir(item_path) and not item.startswith('.'): + if os.path.exists(os.path.join(item_path, '__init__.py')): + modules.append(item) + return web.json_response({"modules": modules}) + except Exception as e: + return web.json_response({"status": "error", "message": str(e)}, status=500) + if (HOTRELOAD_EXCLUDE := os.getenv("HOTRELOAD_EXCLUDE", None)) is not None: EXCLUDE_MODULES.update(x for x in HOTRELOAD_EXCLUDE.split(',') if x) HOTRELOAD_OBSERVE_ONLY: set[str] = set(x for x in os.getenv("HOTRELOAD_OBSERVE_ONLY", '').split(',') if x) diff --git a/config.json b/config.json index bccd3a3..38278c8 100644 --- a/config.json +++ b/config.json @@ -3,12 +3,14 @@ "ComfyUI-IF_AI_tools", "rgthree-comfy", "comfyui-kjnodes", - "ComfyUI-LG_HotReload", - "AIGODLIKE-ComfyUI-Translation", - "comfyui_layerstyle", "ComfyUI-Manager", + "ComfyUI-Depth-Pro", + "ComfyUI-Depth-Anything-Tensorrt", + "comfyui_layerstyle", "ComfyUI-Crystools", - "ComfyUI_Custom_Nodes_AlekPet", - "Comfyui-LG_Relight" + "AIGODLIKE-ComfyUI-Translation", + "ComfyUI-LG_HotReload", + "ComfyUI-Easy-Use", + "ComfyUI_Custom_Nodes_AlekPet" ] } \ No newline at end of file