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