mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-09 13:50:49 +08:00
47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
import asyncio
|
|
import json
|
|
import logging
|
|
from typing import Optional, Literal
|
|
|
|
import typer
|
|
|
|
from ..cmd.main_pre import args
|
|
from ..cli_args_types import Configuration
|
|
from ..component_model.asyncio_files import stream_json_objects
|
|
from ..client.embedded_comfy_client import Comfy
|
|
from ..component_model.entrypoints_common import configure_application_paths, executor_from_args
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
async def main():
|
|
workflows = args.workflows
|
|
assert len(workflows) > 0, "specify at least one path to a workflow, a literal workflow json starting with `{` or `-` (for standard in) using --workflows cli arg"
|
|
configure_application_paths(args)
|
|
executor = await executor_from_args(args)
|
|
|
|
await run_workflows(executor, workflows)
|
|
|
|
|
|
async def run_workflows(executor, workflows: list[str | Literal["-"]], configuration: Optional[Configuration] = None):
|
|
if configuration is None:
|
|
configuration = args
|
|
async with Comfy(executor=executor, configuration=configuration) as comfy:
|
|
for workflow in workflows:
|
|
obj: dict
|
|
async for obj in stream_json_objects(workflow):
|
|
try:
|
|
res = await comfy.queue_prompt_api(obj)
|
|
typer.echo(json.dumps(res.outputs))
|
|
except asyncio.CancelledError:
|
|
logger.info("Exiting gracefully.")
|
|
break
|
|
|
|
|
|
def entrypoint():
|
|
asyncio.run(main())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
entrypoint()
|