We will be using OpenAI’s Dall-E model to generate images. At the time of writing this article Langchain does not provide an integration for Dall-E. If you’re interested head over to langchain to open an PR!

You will need 3 things to authenticate with OpenAI, or Azure OpenAI if you’re using that.

I will be covering the Azure side of things.

OPENAI_API_VERSION=
AZURE_OPENAI_ENDPOINT=
AZURE_OPENAI_API_KEY=

Now you will need the openai package

from openai import AzureOpenAI

client = AzureOpenAI(
    api_version = OPENAI_API_VERSION,
    api_key = AZURE_OPENAI_API_KEY,
    azure_endpoint = AZURE_OPENAI_ENDPOINT,
)

Next, we create this function to show us the generated image

import json
from PIL import Image
import requests
from io import BytesIO

def image_generator(prompt):
    # Generate image with DALL-E 3
    response = client.images.generate(
        model = "dall-e-3",
        prompt = prompt,
    )
    
    # Extract image URL
    json_response = json.loads(response.model_dump_json())
    image_url = json_response["data"][0]["url"]
    response = requests.get(image_url)
    img = Image.open(BytesIO(response.content))
    img.show()

Now let us try it out

image_generator(prompt="Green Apple")