From 3b2c9eb29eeb31a7230d8934d8b7a2009842df33 Mon Sep 17 00:00:00 2001
From: saltacc <alotofcatz@gmail.com>
Date: Tue, 30 Jan 2024 11:52:44 -0800
Subject: [PATCH] files didnt actually add for some reason

---
 .../initialize_comfy_extensions.py            | 98 +++++++++++++++++++
 image_generation/required_extensions.txt      | 18 ++++
 image_generation/required_requirements.txt    |  4 +
 3 files changed, 120 insertions(+)

diff --git a/image_generation/initialize_comfy_extensions.py b/image_generation/initialize_comfy_extensions.py
index e69de29..68c92e1 100644
--- a/image_generation/initialize_comfy_extensions.py
+++ b/image_generation/initialize_comfy_extensions.py
@@ -0,0 +1,98 @@
+import asyncio
+import json
+import os
+import subprocess
+import sys
+import threading
+
+
+def clone_repos(file_name, destination_folder_name):
+    # Get the directory where the script is located
+    script_dir = os.path.dirname(os.path.realpath(__file__))
+
+    # Construct the full path for the file
+    file_path = os.path.join(script_dir, file_name)
+
+    # Construct the full path for the destination folder and change into it
+    destination_folder = os.path.join(script_dir, destination_folder_name)
+    if not os.path.exists(destination_folder):
+        os.makedirs(destination_folder)
+    os.chdir(destination_folder)
+
+    with open(file_path, 'r') as file:
+        for repo_url in file:
+            repo_url = repo_url.strip()
+            if repo_url:
+                try:
+                    # Execute the git clone command
+                    cmd = ['git', 'clone', repo_url]
+                    subprocess.run(cmd, check=True)
+                    print(f"Successfully cloned {repo_url}")
+                except subprocess.CalledProcessError:
+                    print(f"Failed to clone {repo_url}")
+
+
+# Name of the file containing the Git repos, relative to the script
+file_name = 'required_extensions.txt'  # The file should be in the same directory as the script
+
+# Name of the destination folder, relative to the script
+# destination_folder_name = 'ComfyUI/custom_nodes'
+destination_folder_name = 'ComfyUI/custom_nodes'
+
+# Clone the repositories
+clone_repos(file_name, destination_folder_name)
+
+# install extra requirements
+
+# get executable
+executable = sys.executable
+
+script_dir = os.path.dirname(os.path.realpath(__file__))
+file_path = os.path.join(script_dir, 'required_requirements.txt')
+
+# install
+cmd = [executable, '-m', 'pip', 'install', '-r', file_path]
+subprocess.run(cmd, check=True)
+
+# Init all the nodes
+
+current_script_dir = os.path.dirname(os.path.realpath(__file__))
+comfy_path = os.path.dirname(os.path.abspath(__file__))
+comfy_path = os.path.join(comfy_path, 'ComfyUI')
+
+# change to ComfyUI directory
+os.chdir(comfy_path)
+
+# add ComfyUI to path
+sys.path.append(comfy_path)
+
+from server import PromptServer
+from execution import PromptQueue
+from main import init_custom_nodes, cuda_malloc_warning, prompt_worker, execute_prestartup_script
+import folder_paths
+
+execute_prestartup_script()
+
+async def run():
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
+    server = PromptServer(loop)
+    q = PromptQueue(server)
+
+    init_custom_nodes()
+
+    cuda_malloc_warning()
+
+    server.add_routes()
+
+    threading.Thread(target=prompt_worker, daemon=True, args=(q, server,)).start()
+
+    folder_paths.add_model_folder_path("checkpoints", os.path.join(folder_paths.get_output_directory(), "checkpoints"))
+    folder_paths.add_model_folder_path("clip", os.path.join(folder_paths.get_output_directory(), "clip"))
+    folder_paths.add_model_folder_path("vae", os.path.join(folder_paths.get_output_directory(), "vae"))
+
+    await server.start('127.0.0.1', 8188)
+
+
+if __name__ == "__main__":
+    asyncio.run(run())
diff --git a/image_generation/required_extensions.txt b/image_generation/required_extensions.txt
index e69de29..c46d70b 100644
--- a/image_generation/required_extensions.txt
+++ b/image_generation/required_extensions.txt
@@ -0,0 +1,18 @@
+https://github.com/ltdrdata/ComfyUI-Manager
+https://github.com/ltdrdata/ComfyUI-Impact-Pack
+https://github.com/Fannovel16/comfyui_controlnet_aux
+https://github.com/Fannovel16/ComfyUI-Frame-Interpolation
+https://github.com/jags111/efficiency-nodes-comfyui
+https://github.com/Derfuu/Derfuu_ComfyUI_ModdedNodes
+https://github.com/WASasquatch/was-node-suite-comfyui
+https://github.com/pythongosssss/ComfyUI-Custom-Scripts
+https://github.com/melMass/comfy_mtb
+https://github.com/FizzleDorf/ComfyUI_FizzNodes
+https://github.com/Kosinkadink/ComfyUI-Advanced-ControlNet
+https://github.com/Kosinkadink/ComfyUI-AnimateDiff-Evolved
+https://github.com/Kosinkadink/ComfyUI-VideoHelperSuite
+https://github.com/theUpsider/ComfyUI-Logic
+https://github.com/rgthree/rgthree-comfy
+https://github.com/giriss/comfy-image-saver
+https://github.com/kijai/ComfyUI-KJNodes
+https://github.com/Ttl/ComfyUi_NNLatentUpscale
diff --git a/image_generation/required_requirements.txt b/image_generation/required_requirements.txt
index e69de29..ecf9f9a 100644
--- a/image_generation/required_requirements.txt
+++ b/image_generation/required_requirements.txt
@@ -0,0 +1,4 @@
+numexpr
+numba
+opencv-python
+color_matcher