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