Dall-E 3 command line image generator

dalle.py

# Install packages
#   pip install python-dotenv openai
# Add your API key to the .env file.
#   OPENAI_API_KEY="sk-xxxxxx"
# Run
#   python .\dalle.py "a bear in a forest"

import os
import sys
from datetime import datetime
import urllib.request
from dotenv import load_dotenv
import openai
from openai import OpenAI

load_dotenv() 

# Fail if no commandline argument is provided
if len(sys.argv) < 2:
    print("Please provide a prompt")
    exit(1)

text_prompt = sys.argv[1]

client = OpenAI()

image_url = ""

try:
    response = client.images.generate(
    model="dall-e-3",
    prompt=text_prompt,
    size="1024x1024",
    quality="standard",
    n=1,
    )
    image_url = response.data[0].url
except openai.OpenAIError as e:
  print(e.http_status)
  print(e.error)
  exit(1)

print(image_url)

# Download image using python and save it to file in the dalle folder
if not os.path.isdir("dalle"):
    os.mkdir("dalle")
with urllib.request.urlopen(image_url) as response:
    cur_time = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
    with open(f"dalle/{cur_time}.png", "wb") as f:
        f.write(response.read())
    with open(f"dalle/{cur_time}.txt", "wb") as f:
        f.write(text_prompt.encode("utf-8"))