# First, specify the base Docker image.
# You can see the Docker images from Apify at https://hub.docker.com/r/apify/.
# You can also use any other image from Docker Hub.
FROM apify/actor-python:3.11

# Second, copy just requirements.txt into the actor image,
# since it should be the only file that affects the dependency install in the next step,
# in order to speed up the build
COPY requirements.txt ./

# Install the packages specified in requirements.txt,
# Print the installed Python version, pip version
# and all installed packages with their versions for debugging
RUN echo "Python version:" \
 &amp;&amp; python --version \
 &amp;&amp; echo "Pip version:" \
 &amp;&amp; pip --version \
 &amp;&amp; echo "Installing dependencies:" \
 &amp;&amp; pip install -r requirements.txt \
 &amp;&amp; echo "All installed Python packages:" \
 &amp;&amp; pip freeze

# Next, copy the remaining files and directories with the source code.
# Since we do this after installing the dependencies, quick build will be really fast
# for most source file changes.
COPY . ./

# Specify how to launch the source code of your actor.
# By default, the "python3 -m src" command is run
CMD ["python3", "-m", "src"]


.actor/Dockerfile

{
	"actorSpecification": 1,
	"name": "dalle-2-image-generation",
	"title": "DALL-E 2 image generation",
	"description": "Generates images using OpenAI's DALL-E 2.",
	"version": "0.0",
	"meta": {
		"templateId": "python-start"
	},
	"input": "./input_schema.json",
	"dockerfile": "./Dockerfile"
}

.actor/actor.json

{
 "title": "DALL-E 2 image generation",
 "type": "object",
 "schemaVersion": 1,
 "properties": {
 "openai_api_key": {
 "title": "OpenAI API key",
 "type": "string",
 "description": "An API key from https://platform.openai.com/account/api-keys",
 "editor": "textfield",
 "isSecret": true
 },
 "prompt": {
 "title": "Prompt",
 "type": "string",
 "description": "The text prompt used to generate the image(s).",
 "editor": "textfield"
 },
 "count": {
 "title": "Count",
 "type": "integer",
 "description": "The number of images to generate.",
 "editor": "number",
 "default": 1
 },
 "width": {
 "title": "Width",
 "type": "integer",
 "description": "Width of the images.",
 "editor": "number",
 "default": 512
 },
 "height": {
 "title": "Height",
 "type": "integer",
 "description": "Height of the images.",
 "editor": "number",
 "default": 512
 }
 },
 "required": ["openai_api_key", "prompt", "count", "width", "height"]
}

.actor/input_schema.json

src/__init__.py

import asyncio
import logging

from apify.log import ActorLogFormatter

from .main import main

handler = logging.StreamHandler()
handler.setFormatter(ActorLogFormatter())

apify_client_logger = logging.getLogger('apify_client')
apify_client_logger.setLevel(logging.INFO)
apify_client_logger.addHandler(handler)

apify_logger = logging.getLogger('apify')
apify_logger.setLevel(logging.DEBUG)
apify_logger.addHandler(handler)

asyncio.run(main())

src/__main__.py

import openai, requests
from apify import Actor


async def main():
 async with Actor:
 actor_input = await Actor.get_input() or {}

 openai_api_key = actor_input.get('openai_api_key')
 prompt = actor_input.get('prompt')
 count = actor_input.get('count')
 width = actor_input.get('width')
 height = actor_input.get('height')

 try:
 response = openai.Image.create(
 openai_api_key,
 prompt = prompt,
 n = count,
 size = f"{width}x{height}",
 )
 image_url_data = response['data']
 i = 0
 for image_url_obj in image_url_data:
 url = image_url_obj['url']
 image = requests.get(url)
 await Actor.set_value(f'image_{i}.png', image.content)
 i += 1

 except openai.error.OpenAIError as e:
 print(e.http_status)
 print(e.error)

src/main.py

# configurations
.idea

# crawlee and apify storage folders
apify_storage
crawlee_storage
storage

# installed files
.venv

# git folder
.git


.dockerignore

root = true

[*]
indent_style = space
indent_size = 4
charset = utf-8
trim_trailing_whitespace = true
insert_final_newline = true
end_of_line = lf


.editorconfig

# This file tells Git which files shouldn't be added to source control

.idea
.DS_Store

apify_storage
storage

.venv/
.env/
__pypackages__
dist/
build/
*.egg-info/
*.egg

__pycache__

.mypy_cache
.dmypy.json
dmypy.json
.pytest_cache

.scrapy
*.log

storage

.gitignore

# DALL-E 2 image generation

You need to obtain an API key from https://platform.openai.com/account/api-keys

Then you can generate images and they will be stored in the key-value store.

README.md

# Add your dependencies here.
# See https://pip.pypa.io/en/latest/reference/requirements-file-format/
# for how to format them
apify ~= 1.0.0
openai ~= 0.27.2


requirements.txt

# First, specify the base Docker image.
# You can see the Docker images from Apify at https://hub.docker.com/r/apify/.
# You can also use any other image from Docker Hub.
FROM apify/actor-python:3.11

# Second, copy just requirements.txt into the actor image,
# since it should be the only file that affects the dependency install in the next step,
# in order to speed up the build
COPY requirements.txt ./

# Install the packages specified in requirements.txt,
# Print the installed Python version, pip version
# and all installed packages with their versions for debugging
RUN echo "Python version:" \
 && python --version \
 && echo "Pip version:" \
 && pip --version \
 && echo "Installing dependencies:" \
 && pip install -r requirements.txt \
 && echo "All installed Python packages:" \
 && pip freeze

# Next, copy the remaining files and directories with the source code.
# Since we do this after installing the dependencies, quick build will be really fast
# for most source file changes.
COPY . ./

# Specify how to launch the source code of your actor.
# By default, the "python3 -m src" command is run
CMD ["python3", "-m", "src"]


Generates images using OpenAI's DALL-E 2.

dalle-2-image-generation

DALL-E 2 image generation

Count

Height

An API key from https://platform.openai.com/account/api-keys

OpenAI API key

The text prompt used to generate the image(s).

Prompt

Width

import asyncio
import logging

from apify.log import ActorLogFormatter

from .main import main

handler = logging.StreamHandler()
handler.setFormatter(ActorLogFormatter())

apify_client_logger = logging.getLogger('apify_client')
apify_client_logger.setLevel(logging.INFO)
apify_client_logger.addHandler(handler)

apify_logger = logging.getLogger('apify')
apify_logger.setLevel(logging.DEBUG)
apify_logger.addHandler(handler)

asyncio.run(main())


import openai, requests
from apify import Actor


async def main():
    async with Actor:
        actor_input = await Actor.get_input() or {}

        openai_api_key = actor_input.get('openai_api_key')
        prompt = actor_input.get('prompt')
        count = actor_input.get('count')
        width = actor_input.get('width')
        height = actor_input.get('height')

        try:
            response = openai.Image.create(
                openai_api_key,
                prompt = prompt,
                n = count,
                size = f"{width}x{height}",
            )
            image_url_data = response['data']
            i = 0
            for image_url_obj in image_url_data:
                url = image_url_obj['url']
                image = requests.get(url)
                await Actor.set_value(f'image_{i}.png', image.content)
                i += 1

        except openai.error.OpenAIError as e:
            print(e.http_status)
            print(e.error)




# DALL-E 2 image generation

You need to obtain an API key from https://platform.openai.com/account/api-keys

Then you can generate images and they will be stored in the key-value store.


DALL-E 2 Image Generation

Where next?