Function main

src/gpt_image_cli/cli.py:257–290

()

def main() -> int:
    args = parse_args()

    _load_env_chain()
    if "OPENAI_API_KEY" not in os.environ:
        print(
            "error: OPENAI_API_KEY not set. Add it to ~/.env or `export OPENAI_API_KEY=...`.",
            file=sys.stderr,
        )
        return 2

    if args.mask and not args.image:
        print("error: --mask requires --image (edits endpoint only)", file=sys.stderr)
        return 2

    ext = args.output_format or "png"
    out_path = Path(args.file).expanduser().resolve() if args.file else default_output_path(args.prompt, ext)

    client = OpenAI()  # auto-reads OPENAI_API_KEY

    try:
        result = call_edit(client, args) if args.image else call_generate(client, args)
    except APIError as e:
        print(f"error: {type(e).__name__}: {e}", file=sys.stderr)
        return 1

    data = result.data or []
    if not data:
        print(f"error: no image data in response: {result}", file=sys.stderr)
        return 1

    for p in write_outputs(data, out_path, args.n):
        print(p)
    return 0


if __name__ == "__main__":