From e6bd5af6a8e306a1cdef63402a77a980a04ad6e1 Mon Sep 17 00:00:00 2001
From: Grafting Rayman <156515434+GraftingRayman@users.noreply.github.com>
Date: Fri, 17 Jan 2025 11:06:44 +0000
Subject: Add files via upload

---
 __init__.py | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 __init__.py

(limited to '__init__.py')

diff --git a/__init__.py b/__init__.py
new file mode 100644
index 0000000..3f8982d
--- /dev/null
+++ b/__init__.py
@@ -0,0 +1,39 @@
+import sys
+import os
+
+repo_dir = os.path.dirname(os.path.realpath(__file__))
+sys.path.insert(0, repo_dir)
+original_modules = sys.modules.copy()
+
+# Place aside existing modules if using a1111 web ui
+modules_used = [
+    "modules",
+    "modules.images",
+    "modules.processing",
+    "modules.scripts_postprocessing",
+    "modules.scripts",
+    "modules.shared",
+]
+original_webui_modules = {}
+for module in modules_used:
+    if module in sys.modules:
+        original_webui_modules[module] = sys.modules.pop(module)
+
+# Proceed with node setup
+from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
+
+# Clean up imports
+# Remove repo directory from path
+sys.path.remove(repo_dir)
+# Remove any new modules
+modules_to_remove = []
+for module in sys.modules:
+    if module not in original_modules and not module.startswith("google.protobuf") and not module.startswith("onnx") and not module.startswith("cv2"):
+        modules_to_remove.append(module)
+for module in modules_to_remove:
+    del sys.modules[module]
+
+# Restore original modules
+sys.modules.update(original_webui_modules)
-- 
cgit v1.2.3