Merge pull request #2 from get-salt-AI/update-audioviz

Update AudioViz
get-salt-AI · May 23, 2024 · 2a1d902 · 2a1d902
2 parents 436d648 + 687ad7f
commit 2a1d902
Show file tree

Hide file tree

Showing 12 changed files with 3,563 additions and 236 deletions.
diff --git a/__init__.py b/__init__.py
@@ -1,16 +1,21 @@
 import os
 
-from SaltAI_AudioViz.modules.node_importer import ModuleLoader
+# Create package logger
+from .modules.log import create_logger
+logger = create_logger()
 
 ROOT = os.path.abspath(os.path.dirname(__file__))
 NAME = "Salt.AI AudioViz"
 PACKAGE = "SaltAI_AudioViz"
+MENU_NAME = "SALT"
+SUB_MENU_NAME = "AudioViz"
 NODES_DIR = os.path.join(ROOT, 'nodes')
 EXTENSION_WEB_DIRS = {}
 NODE_CLASS_MAPPINGS = {}
 NODE_DISPLAY_NAME_MAPPINGS = {}
 
 # Load modules
+from SaltAI_AudioViz.modules.node_importer import ModuleLoader
 module_timings = {}
 module_loader = ModuleLoader(PACKAGE)
 module_loader.load_modules(NODES_DIR)
@@ -20,9 +25,9 @@
 NODE_DISPLAY_NAME_MAPPINGS = module_loader.NODE_DISPLAY_NAME_MAPPINGS
 
 # Timings and such
-print("")
+logger.info("")
 module_loader.report(NAME)
-print("")
+logger.info("")
 
 # Export nodes
 __all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']