From fbf8388437245f4594c5ef87582a4278fdfd4391 Mon Sep 17 00:00:00 2001 From: David Marx Date: Sun, 31 Dec 2023 22:06:16 -0800 Subject: [PATCH] Update __init__.py --- __init__.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) diff --git a/__init__.py b/__init__.py index 7432280..bb346ac 100644 --- a/__init__.py +++ b/__init__.py @@ -1,3 +1,36 @@ from nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS __all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] + +import os +import subprocess +import importlib.util +import sys + +# scavenged install sequence from https://github.com/FizzleDorf/ComfyUI_FizzNodes/blob/main/__init__.py +def is_installed(package, package_overwrite=None): + try: + spec = importlib.util.find_spec(package) + except ModuleNotFoundError: + pass + + package = package_overwrite or package + + python = sys.executable + if spec is None: + print(f"Installing {package}...") + command = f'"{python}" -m pip install {package}' + + result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, env=os.environ) + + if result.returncode != 0: + print(f"Couldn't install\nCommand: {command}\nError code: {result.returncode}") + +# to do: read from requirements.txt +is_installed("scipy") +is_installed("scikit-learn") +is_installed("librosa") + + +print(os.environ.get('COMFYUI_DEBUG_MODE')) +