First commit with core code

Core developed code and functionality of LevelPixel nodes
2024-11-05 01:35:36 +06:00
parent 675353dcdb
commit 418db2c60d
21 changed files with 13123 additions and 12 deletions
@@ -0,0 +1,69 @@
+import os
+import pkg_resources
+import sys
+import subprocess
+import folder_paths
+import importlib
+
+supported_LLava_extensions = set(['.gguf'])
+
+try:
+    folder_paths.folder_names_and_paths["LLavacheckpoints"] = (folder_paths.folder_names_and_paths["LLavacheckpoints"][0], supported_LLava_extensions)
+except:
+    if not os.path.isdir(os.path.join(folder_paths.models_dir, "LLavacheckpoints")):
+        os.mkdir(os.path.join(folder_paths.models_dir, "LLavacheckpoints"))
+        
+    folder_paths.folder_names_and_paths["LLavacheckpoints"] = ([os.path.join(folder_paths.models_dir, "LLavacheckpoints")], supported_LLava_extensions)
+
+def check_requirements_installed(requirements_path):
+    with open(requirements_path, 'r') as f:
+        requirements = [pkg_resources.Requirement.parse(line.strip()) for line in f if line.strip()]
+
+    installed_packages = {pkg.key: pkg for pkg in pkg_resources.working_set}
+    installed_packages_set = set(installed_packages.keys())
+    missing_packages = []
+    for requirement in requirements:
+        if requirement.key not in installed_packages_set or not installed_packages[requirement.key] in requirement:
+            missing_packages.append(str(requirement))
+
+    if missing_packages:
+        print(f"Missing or outdated packages: {', '.join(missing_packages)}")
+        print("Installing/Updating missing packages...")
+        subprocess.check_call([sys.executable, '-s', '-m', 'pip', 'install', *missing_packages])
+
+requirements_path  = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requirements.txt")
+check_requirements_installed(requirements_path)
+
+from .install_init import init, get_system_info, install_llama
+
+system_info = get_system_info()
+install_llama(system_info)
+llama_cpp_agent_path  = os.path.join(os.path.dirname(os.path.realpath(__file__)), "cpp_agent_req.txt")
+check_requirements_installed(llama_cpp_agent_path)
+
+init()
+
+node_list = [
+    "image.image_utils_LP",
+    "io.numbers_utils_LP",
+    "io.folder_workers_LP",
+    "io.image_loaders_LP",
+    "io.text_utils_LP",
+    "llm.llm_LP",
+    "tags.tags_utils_LP",
+    "text.text_utils_LP",
+    "unloaders.model_unloaders_LP",
+    "vlm.llava_LP",
+]
+
+NODE_CLASS_MAPPINGS = {}
+NODE_DISPLAY_NAME_MAPPINGS = {}
+
+for module_name in node_list:
+    imported_module = importlib.import_module(f".nodes.{module_name}", __name__)
+
+    NODE_CLASS_MAPPINGS = {**NODE_CLASS_MAPPINGS, **imported_module.NODE_CLASS_MAPPINGS}
+    NODE_DISPLAY_NAME_MAPPINGS = {**NODE_DISPLAY_NAME_MAPPINGS, **imported_module.NODE_DISPLAY_NAME_MAPPINGS}
+
+WEB_DIRECTORY = "./web"
+__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS", "WEB_DIRECTORY"]