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