From 5fe795c13e751addef147d0e2fa5ba90e6c84caa Mon Sep 17 00:00:00 2001
From: atmaranto <35581227+atmaranto@users.noreply.github.com>
Date: Thu, 7 Mar 2024 03:59:46 -0600
Subject: [PATCH] Reduce requirements, use install script

---
 __init__.py      | 14 +++++++++++++-
 requirements.txt | 12 ------------
 2 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/__init__.py b/__init__.py
index ca899b4..416be60 100644
--- a/__init__.py
+++ b/__init__.py
@@ -2,7 +2,6 @@ import sys
 import os
 
 from io import StringIO
-import json
 
 import traceback
 
@@ -11,6 +10,19 @@ from aiohttp import web
 ext_dir = os.path.dirname(__file__)
 sys.path.append(ext_dir)
 
+try:
+    import black
+except ImportError:
+    print("Unable to import requirements for ComfyUI-SaveAsScript.")
+    print("Installing...")
+
+    import importlib
+    spec = importlib.util.spec_from_file_location('impact_install', os.path.join(os.path.dirname(__file__), 'install.py'))
+    impact_install = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(impact_install)
+
+    print("Successfully installed. Hopefully, at least.")
+
 # Prevent reimporting of custom nodes
 os.environ["RUNNING_IN_COMFYUI"] = "TRUE"
 
diff --git a/requirements.txt b/requirements.txt
index 5bb9376..b3ca909 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,13 +1 @@
-torch
-torchdiffeq
-torchsde
-einops
-transformers>=4.25.1
-safetensors>=0.3.0
-aiohttp
-accelerate
-pyyaml
-Pillow
-scipy
-tqdm
 black
\ No newline at end of file