From 87a4afd1087cadb606f4c0bbfe512dd264f3b864 Mon Sep 17 00:00:00 2001
From: limbicnation <gero@limbicnation.com>
Date: Fri, 29 Nov 2024 18:43:09 +0100
Subject: [PATCH] feat: enhance __init__.py with version tracking, web
 directory config and improved documentation

- Add module docstring
- Add version tracking (__version__)
- Add web directory config for UI components
- Add explicit __all__ exports list
- Improve code organization
---
 __init__.py | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/__init__.py b/__init__.py
index 2a2fa22..c36e481 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,11 +1,30 @@
+"""
+ComfyUI Depth Estimation Node
+A custom node for depth map estimation using transformer models.
+"""
+
 from .depth_estimation_node import DepthEstimationNode
 
+# Version info
+__version__ = "1.0.0"
+
+# Node class mapping for ComfyUI registration
 NODE_CLASS_MAPPINGS = {
     "DepthEstimationNode": DepthEstimationNode
 }
 
+# Display names for UI
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "DepthEstimationNode": "Depth Estimation Node"
+    "DepthEstimationNode": "Depth Estimation"
 }
 
-__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
+# Web extension info for ComfyUI
+WEB_DIRECTORY = "./js"
+
+# Module exports
+__all__ = [
+    "NODE_CLASS_MAPPINGS",
+    "NODE_DISPLAY_NAME_MAPPINGS",
+    "__version__",
+    "WEB_DIRECTORY"
+]