From f8f8a36ef9e9bda8541e7665e4e2f0a489106d13 Mon Sep 17 00:00:00 2001
From: lilly <lilly1987@naver.com>
Date: Mon, 20 Mar 2023 18:07:53 +0900
Subject: [PATCH] Update __init__.py

---
 __init__.py | 73 +++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 54 insertions(+), 19 deletions(-)

diff --git a/__init__.py b/__init__.py
index 339befb..1cfc077 100644
--- a/__init__.py
+++ b/__init__.py
@@ -1,24 +1,59 @@
 print("ComfyUI_node_Lilly.__init__")
 
-import importlib
-#from custom_nodes.ComfyUI_node_Lilly import eval(f"{name}")
+import os
+import sys
+#print(f"sys.modules : {sys.modules}")
 
-NODE_CLASS_MAPPINGS = {
-}
+#mn=os.path.abspath(__main__)
+#print("nm ", mn)
 
-def add(name):
-    print(f"Load : {name}")
-    try:        
-        pkg = importlib.import_module(f"custom_nodes.ComfyUI_node_Lilly.{name}")
-        NODE_CLASS_MAPPINGS[name]=eval(f"pkg.{name}")
-        print(f"ok   : {name}")
-    except Exception as e:         
-        print(f"Exception : {e}")
+#filePath = __file__
+#print("This script file path is ", filePath)
+#
+#absFilePath = os.path.abspath(__file__)
+#print("This script absolute path is ", absFilePath)
+#
+#realFilePath = os.path.realpath(__file__)
+#print("This script real path is ", realFilePath)
+#
+#path, filename = os.path.split(absFilePath)
+#print("Script file path is {}, filename is {}".format(path, filename))
+    
+nm=os.path.abspath(__name__)
+print("abspath __name__ : ", nm)
 
-add("CheckpointLoaderRandom")
-add("CheckpointLoaderSimpleText")
-add("CLIPTextEncodeWildcards")
-add("LoraLoaderText")
-add("LoraLoaderTextRandom")
-add("Random_Sampler")
-add("VAELoaderDecode")
\ No newline at end of file
+wd = os.getcwd()
+print("working directory : ", wd)
+
+md=nm.replace(wd+"\\","")
+print("import name", md)
+
+#if md in sys.modules:
+#    print(f"{md!r} already in sys.modules")
+#else:
+#    print(f"{md!r} not in sys.modules")
+        
+if not md.startswith("custom_nodes."):
+
+    import importlib
+    #from custom_nodes.ComfyUI_node_Lilly import eval(f"{name}")
+
+    NODE_CLASS_MAPPINGS = {
+    }
+
+    def add(name):
+        print(f"Load : {name}")
+        try:        
+            pkg = importlib.import_module(f"{md}.{name}")
+            NODE_CLASS_MAPPINGS[name]=eval(f"pkg.{name}")
+            print(f"ok   : {name}")
+        except Exception as e:         
+            print(f"Exception : {e}")
+
+    add("CheckpointLoaderRandom")
+    add("CheckpointLoaderSimpleText")
+    add("CLIPTextEncodeWildcards")
+    add("LoraLoaderText")
+    add("LoraLoaderTextRandom")
+    add("Random_Sampler")
+    add("VAELoaderDecode")
\ No newline at end of file