From e3c6ab971c302704a608866974b6f1287ae50926 Mon Sep 17 00:00:00 2001
From: Kasisnu <5044120+kasisnu@users.noreply.github.com>
Date: Wed, 10 Jul 2024 12:47:35 +0530
Subject: [PATCH] previews: nicer builds

---
 workflows/preview_server/initialize_comfy_extensions.py | 5 +++--
 workflows/preview_server/preview_server.py              | 4 +---
 workflows/preview_server/server_requirements.txt        | 1 -
 3 files changed, 4 insertions(+), 6 deletions(-)

diff --git a/workflows/preview_server/initialize_comfy_extensions.py b/workflows/preview_server/initialize_comfy_extensions.py
index bae7c54..adad2b8 100644
--- a/workflows/preview_server/initialize_comfy_extensions.py
+++ b/workflows/preview_server/initialize_comfy_extensions.py
@@ -16,8 +16,9 @@ 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
+from main import cuda_malloc_warning, prompt_worker, execute_prestartup_script
 import folder_paths
+from nodes import init_extra_nodes
 
 execute_prestartup_script()
 
@@ -27,7 +28,7 @@ async def run():
     server = PromptServer(loop)
     q = PromptQueue(server)
 
-    init_custom_nodes()
+    init_extra_nodes()
 
     cuda_malloc_warning()
 
diff --git a/workflows/preview_server/preview_server.py b/workflows/preview_server/preview_server.py
index cbde3e8..a1bf654 100644
--- a/workflows/preview_server/preview_server.py
+++ b/workflows/preview_server/preview_server.py
@@ -4,7 +4,6 @@ import time
 
 import shutil
 from pathlib import Path
-import asyncio
 import uuid
 import os
 import json
@@ -15,7 +14,6 @@ import mimetypes
 from fastapi import FastAPI, File, UploadFile, HTTPException, Form, Request, Response
 import requests
 from jsonpath_ng import jsonpath, parse
-from moviepy.editor import ImageClip
 from fastapi.middleware.cors import CORSMiddleware
 import urllib
 import websockets
@@ -399,7 +397,7 @@ if __name__ == "__main__":
 
     signal.signal(signal.SIGINT, signal_handler)
     signal.signal(signal.SIGTERM, signal_handler)
-    comfyui_process = subprocess.Popen(["venv/bin/python", "main.py", "--highvram", "--disable-metadata"], cwd=COMFY_ROOT_DIRECTORY)
+    comfyui_process = subprocess.Popen(["../venv/bin/python", "main.py", "--highvram", "--disable-metadata"], cwd=COMFY_ROOT_DIRECTORY)
 
 
     # block until comfyUI server is up
diff --git a/workflows/preview_server/server_requirements.txt b/workflows/preview_server/server_requirements.txt
index 0523c7c..16ff43f 100644
--- a/workflows/preview_server/server_requirements.txt
+++ b/workflows/preview_server/server_requirements.txt
@@ -4,5 +4,4 @@ fastapi~=0.110.0
 python-multipart~=0.0.9
 pydantic~=2.6.4
 jsonpath-ng~=1.6.1
-moviepy
 websockets
\ No newline at end of file