gepetie_image

make_image.py

#!/usr/bin/env python3

import os
import argparse

from openai import OpenAI


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("prompt")

    args = parser.parse_args()

    client = OpenAI(api_key=os.environ.get("OPENAI_KEY"))

    response = client.images.generate(
       model="dall-e-3",
       prompt=args.prompt,
       size="1024x1024",
       quality="standard",
       n=1,
    )

    print(response.data[0].url)


if __name__ == '__main__':
    main()