previews: nicer builds

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()
 

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

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