Criando Imagens de IA Utilizando a API da OpenAI e DALL-E

Você já imaginou criar imagens incríveis utilizando apenas algumas linhas de código? Com a API da OpenAI e o modelo DALL-E, isso é possível! Neste post, vamos explorar como você, que está começando na área de Inteligência Artificial, pode gerar imagens com a API da OpenAI e DALL-E de forma simples e prática. Vamos lá?

O que é a API da OpenAI?

Antes de iniciarmos a criação de imagens, é importante entender o que é uma API. API, ou Interface de Programação de Aplicações, é um conjunto de regras que permite que diferentes softwares se comuniquem entre si. No caso da OpenAI, a API permite que você acesse modelos de inteligência artificial avançados, como o ChatGPT e o DALL-E, para realizar diversas tarefas, incluindo a geração de imagens.

Introdução ao DALL-E

O DALL-E é um modelo de inteligência artificial desenvolvido pela OpenAI que pode criar imagens a partir de descrições textuais. Imagine descrever uma cena e ver essa descrição se transformar em uma imagem realista ou artística! O DALL-E 3 é a versão mais recente e oferece ainda mais precisão e qualidade na geração de imagens.

Configurando o Ambiente

Para começar a criar imagens com a API da OpenAI e DALL-E, você precisará configurar seu ambiente de desenvolvimento. Vamos utilizar a linguagem de programação Python, que é simples e poderosa.

Instalando as Bibliotecas Necessárias

Primeiro, vamos instalar as bibliotecas necessárias. Você precisará da biblioteca openai para acessar a API e da Pillow para manipular as imagens.

pip install openai Pillow

Configurando a Chave de API

Para utilizar a API da OpenAI, você precisará de uma chave de API. Siga os passos abaixo para gerar a sua:

1. Acesse o site da OpenAI e faça login.
2. Vá até a seção de API Keys e clique em “Create new secret key”.
3. Copie a chave gerada e guarde-a em um local seguro.

Inicializando a Biblioteca da OpenAI

Com a chave de API em mãos, vamos inicializar a biblioteca da OpenAI no nosso código Python.

import openai
from dotenv import load_dotenv, find_dotenv

_ = load_dotenv(find_dotenv())

client = openai.Client()

Criando Imagens com a API da OpenAI e DALL-E

Agora que nosso ambiente está configurado, vamos criar nossa primeira imagem utilizando a API da OpenAI e DALL-E.

Gerando uma Imagem

Vamos começar gerando uma imagem simples. O código abaixo cria uma imagem de um campo de pastagem com uma leve elevação ao fundo.

import requests
from PIL import Image

nome = 'bosque'
modelo = 'dall-e-3'
prompt = 'Crie uma imagem de um campo de pastagem, amplo com uma leve elevação ao fundo.'
qualidade = 'hd'
style = 'natural'

resposta = client.images.generate(
    model=modelo,
    prompt=prompt,
    size='1024x1024',
    quality=qualidade,
    style=style,
    n=1
)

# Salvando a imagem gerada
nome_arquivo = f'{nome}_{modelo}_{qualidade}_{style}.jpg'
image_url = resposta.data[0].url
img_data = requests.get(image_url).content
with open(nome_arquivo, 'wb') as f:
    f.write(img_data)

# Visualizando a imagem
image = Image.open(nome_arquivo)
image.show()

Editando uma Imagem

Além de criar imagens do zero, o DALL-E também permite editar imagens existentes. Para isso, precisamos criar uma máscara que indica a área da imagem que queremos modificar. Para gerar a mask, podemos utilizar este site.

resposta = client.images.edit(
    model='dall-e-2',
    image=open('arquivos/imagens/original.png', 'rb'),
    mask=open('arquivos/imagens/mask.png', 'rb'),
    prompt='Adicione uma vaca e um terneirinho na imagem fornecida',
    n=1,
    size='1024x1024'
)

# Salvando a imagem editada
nome_arquivo = 'editada.jpg'
image_url = resposta.data[0].url
img_data = requests.get(image_url).content
with open(nome_arquivo, 'wb') as f:
    f.write(img_data)

# Visualizando a imagem editada
image = Image.open(nome_arquivo)
image.show()

Criando Variações de uma Imagem

Outra funcionalidade interessante do DALL-E é a capacidade de criar variações de uma imagem existente. Veja como fazer isso:

resposta = client.images.create_variation(
    image=open('arquivos/imagens/bosque_dall-e-3_hd_natural.jpg', 'rb'),
    n=1,
    size='1024x1024'
)

# Salvando a variação da imagem
nome_arquivo = 'variacao.jpg'
image_url = resposta.data[0].url
img_data = requests.get(image_url).content
with open(nome_arquivo, 'wb') as f:
    f.write(img_data)

# Visualizando a variação da imagem
image = Image.open(nome_arquivo)
image.show()

Conclusão

Criar imagens com a API da OpenAI e DALL-E é uma ótima maneira de explorar o poder da inteligência artificial de forma prática e acessível. Com apenas algumas linhas de código, você pode gerar, editar e criar variações de imagens, abrindo um mundo de possibilidades criativas.

Se você está começando na área de IA, essa é uma excelente oportunidade para entender como as APIs podem simplificar o uso de modelos avançados sem a necessidade de treinamento de algoritmos complexos. Experimente criar suas próprias imagens e veja até onde sua criatividade pode te levar!

Se você gostou deste post e quer aprender mais sobre inteligência artificial e programação, continue acompanhando nosso blog e explore nossos cursos na Asimov Academy. Até a próxima!