Complete Guide to using DALL-E API in Python

In this article, we will see how to use and call OpenAI DALL-E API in Python. We will cover how to use Google Colab notebook to call DALL-E API in Python.

We will demonstrate calling DALL-E API in Python for all its features, like new image generation, editing an image, and generating image variations.

Let’s dive in.

Requirements for calling DALL-E API in Python

You could run all the code from this article locally on your machine too.

However, for demonstration assumes you are using a Google Colab Notebook.

If you are going to run it locally, you need to make sure you have Python 3.x or higher installed.

You also need an OpenAI Account.

Setup

Open up a new Google Colab Notebook.

The rest of the steps below require you to type the code in the Google Colab and execute that particular cell before going to the next step.

Install OpenAI’s Python module by running this code

!pip install openai

import os
import openai
from IPython.display import Image
from IPython import display
from base64 import b64decode

openai.api_key = ""

response = openai.Image.create(
  prompt="A cute baby sea otter",
  n=1,
  size="256x256",
  response_format="b64_json"
)

This code generates a single variant image for the prompt “a cute sea otter”. You can go ahead and change the prompt to whatever you like.

the response is a JSON object that is stored in the response variable.

If you print the response, you will output it like this.

We will now use the base64 string that was returned to us by the DALL-E API to display the image in the Google Colab Notebook itself.

First, we extract the base64 string from the response JSON object

image_b64 = response['data'][0]['b64_json']