moving stuff around

workflows/comfy/ComfyLauncher/ComfyRunner.py

@@ -0,0 +1,68 @@
+import asyncio
+import json
+import os
+import sys
+import threading
+import signal
+
+import aiohttp
+
+current_script_dir = os.path.dirname(os.path.realpath(__file__))
+file_path = os.path.join(current_script_dir, 'prompt', 'prompt.json')
+prompt = json.load(open(file_path))
+
+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
+import folder_paths
+
+
+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)
+
+    async def send_prompt_async():
+        # Load prompt
+        p = {"prompt": prompt}
+        data = json.dumps(p).encode('utf-8')
+        # Send HTTP post request
+        endpoint = "http://127.0.0.1:8188/prompt"
+        async with aiohttp.ClientSession() as session:
+            async with session.post(endpoint, data=data) as resp:
+                print(
+                    await resp.text())
+
+
+    await send_prompt_async()
+
+    while server.prompt_queue.currently_running:
+        await asyncio.sleep(0.05)
+
+if __name__ == "__main__":
+    asyncio.run(run())

workflows/comfy/ComfyLauncher/README.md

@@ -0,0 +1,15 @@
+## ComfyRunner
+
+Run a single ComfyUI request from the command line.
+
+### Usage
+
+Put the request parameters into `prompt/prompt.json` and run `ComfyRunner.py`. The ComfyUI folder must be named "ComfyUI" and be in the same superdirectory
+
+### Usage in backend
+
+It must run with the ComfyUI venv, example:
+```
+COMFY_INFERENCE_ROOT_DIRECTORY="/home/salt/clone/storyteller-ml/image_generation/ComfyUI"
+COMFY_INFERENCE_EXECUTABLE_OR_COMMAND="venv/bin/python ../ComfyLauncher/ComfyLauncher.py"
+```

workflows/comfy/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())

workflows/comfy/required_extensions.txt

@@ -0,0 +1,20 @@
+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
+https://github.com/cubiq/ComfyUI_IPAdapter_plus
+https://github.com/cubiq/ComfyUI_essentials

workflows/comfy/required_requirements.txt

@@ -0,0 +1,4 @@
+numexpr
+numba
+opencv-python
+color_matcher