|
|
import sys
|
|
|
import os
|
|
|
|
|
|
from io import StringIO
|
|
|
|
|
|
import traceback
|
|
|
|
|
|
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.")
|
|
|
|
|
|
|
|
|
os.environ["RUNNING_IN_COMFYUI"] = "TRUE"
|
|
|
|
|
|
from comfyui_to_python import ComfyUItoPython
|
|
|
|
|
|
sys.path.append(os.path.dirname(os.path.dirname(ext_dir)))
|
|
|
|
|
|
import server
|
|
|
|
|
|
WEB_DIRECTORY = "js"
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
|
|
|
|
|
|
|
@server.PromptServer.instance.routes.post("/saveasscript")
|
|
|
async def save_as_script(request):
|
|
|
try:
|
|
|
data = await request.json()
|
|
|
name = data["name"]
|
|
|
workflow = data["workflow"]
|
|
|
|
|
|
sio = StringIO()
|
|
|
ComfyUItoPython(workflow=workflow, output_file=sio)
|
|
|
|
|
|
sio.seek(0)
|
|
|
data = sio.read()
|
|
|
|
|
|
return web.Response(text=data, status=200)
|
|
|
except Exception as e:
|
|
|
traceback.print_exc()
|
|
|
return web.Response(text=str(e), status=500)
|
|
|
|
