Matricule-se
Voltar para o blog
Python

Como Gerar Imagens com IA e Python: Um Passo a Passo Completo

Pedro Surreaux

Aprenda como gerar imagens com IA e Python implementando uma das melhores ferramentas de geração de imagens diretamente no seu código.

A API da OpenAI permite gerar imagens incríveis com poucas linhas de código. Neste post, vamos explorar como você pode usar essa tecnologia, especificamente o modelo DALL-E, para criar imagens personalizadas com Python.

Além disso, vamos mostrar como salvar essas imagens diretamente no seu computador.

Caso prefira esse conteúdo no formato de vídeo-aula, assista ao vídeo abaixo ou acesse o nosso canal do YouTube!

Para fazer o download do(s) arquivo(s) utilizados na aula, preencha com o seu e-mail:

Não vamos te encaminhar nenhum tipo de SPAM! A Hashtag Treinamentos é uma empresa preocupada com a proteção de seus dados e realiza o tratamento de acordo com a Lei Geral de Proteção de Dados (Lei n. 13.709/18). Qualquer dúvida, nos contate.

O que é a API da OpenAI e o DALL-E?

A OpenAI é uma das empresas líderes em inteligência artificial, responsável por ferramentas como o ChatGPT e o DALL-E. O DALL-E é um modelo de geração de imagens que usa descrições escritas (prompts) para criar imagens realistas e criativas.

Com a API da OpenAI, você pode integrar essa funcionalidade diretamente em seus projetos Python.

Por que usar Python para gerar imagens?

Python é uma linguagem de programação versátil e fácil de aprender, amplamente utilizada em automações, análise de dados e, claro, inteligência artificial.

Com bibliotecas como openai e requests, você pode automatizar a geração de imagens e integrá-las em seus fluxos de trabalho com um código curto e fácil de entender.

Passo a Passo: Gerando Imagens com a API da OpenAI

Vamos entender no processo de geração de imagens usando a API da OpenAI e Python. O código é simples e pode ser adaptado para diversas aplicações.

1. Configurando o Ambiente

Antes de começar, você precisará de uma chave de API da OpenAI. Para obtê-la:

  1. Acesse o site da OpenAI e faça login.
  2. Navegue até a seção "API Keys" e crie uma nova chave.
  3. Copie a chave e armazene-a em um arquivo .env no seu projeto.
acessando a chave api no site da open ai

Importante: A API da OpenAI é paga, e o custo varia dependendo do modelo e do tamanho da imagem. Para começar, você pode carregar $5 na sua conta, o que é suficiente para testes e pequenos projetos.

2. Instalando as Bibliotecas Necessárias

Você precisará instalar duas bibliotecas principais:

  • openai: Para interagir com a API da OpenAI.
  • python-dotenv: Para carregar a chave de API do arquivo .env.
Codigo
pip install python-dotenv openai requests

3. Criando o Código para Gerar Imagens

Agora que temos as bibliotecas instaladas, vamos ao código.

O processo é dividido em duas partes: gerar a imagem e salvá-la no computador.

Python
# Importando as bibliotecas
from openai import OpenAI
from dotenv import load_dotenv
import os
import requests

# Carregando a chave de API do arquivo .env
load_dotenv()

# Configurando o cliente da OpenAI
cliente = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))

# Gerando a imagem
resposta = client.images.generate(
    model="dall-e-3",
    prompt="uma imagem de um rato vestido com roupa indiana tocando trompete nos alpes suíços",
    size="1024x1024",

)
# Obtendo o link da imagem gerada
url_imagem = resposta.data[0].url
print("Imagem gerada com sucesso!")
print("Link da imagem:", url_imagem)

# Baixando e salvando a imagem
informacoes_imagem = requests.get(url_imagem)
with open("imagem.jpg", "wb") as arquivo_imagem:
    arquivo_imagem.write(informações_imagem.content)
print("Imagem salva com sucesso!")
print do código para gerar imagens com python
Ícone PythonPython Impressionador

Você vai aprender a linguagem de Programação que mais cresce no Mundo para fazer automações incríveis, desenvolver sites, fazer análises de dados, trabalhar com Ciência de Dados, Inteligência Artificial, mesmo que você nunca tenha tido nenhum contato com Programação na vida.

Começar agoraSeta para a direita
Fundo PythonTelas Python
Luz Python

Explicação do Código

  • Carregando a chave de API: Usamos python-dotenv para carregar a chave de API do arquivo .env.
  • Gerando a imagem: O método client.images.generate envia o prompt para a API e retorna um link para a imagem gerada.
  • Baixando a imagem: Usamos a biblioteca requests para baixar a imagem a partir do link e salvá-la no computador.

Dicas para Melhorar suas Imagens

  1. Seja Específico no Prompt: Quanto mais detalhado for o prompt, melhor será a imagem gerada. Por exemplo, em vez de "um rato tocando trompete", use "um rato realista vestido com roupas tradicionais indianas tocando trompete nos Alpes Suíços com neve ao fundo".
  2. Experimente Diferentes Modelos: O DALL-E 3 oferece imagens de alta qualidade, mas o DALL-E 2 pode ser útil para tarefas específicas, como gerar variações de uma imagem existente.
  3. Ajuste o Tamanho e a Qualidade: Imagens maiores e de alta qualidade consomem mais créditos, mas podem valer a pena dependendo do seu uso.
imagem gerada pelo dall-e com o prompt da aula

Conclusão - Como Gerar Imagens com IA e Python

Gerar imagens com a API da OpenAI e Python é uma tarefa simples e poderosa. Com poucas linhas de código, você pode criar imagens incríveis para uso em projetos pessoais, automações ou até mesmo em aplicações comerciais.

A chave está em experimentar diferentes prompts e ajustar os parâmetros para obter os melhores resultados.

Com este guia, você está pronto para explorar o mundo da geração de imagens com IA. E se quiser ler mais conteúdos como este, não deixe de conferir nossas publicações, com dicas e tutoriais sobre Python e Inteligência Artificial! Boa codificação!

Hashtag Treinamentos

Para acessar outras publicações de Python, clique aqui!

Quer aprender mais sobre Python com um minicurso básico gratuito?

Posts mais recentes de Python

Posts mais recentes da Hashtag Treinamentos

Pedro Surreaux

Autor do artigo

Pedro Surreaux

Redator de Publicações

Redator de conteúdo na Hashtag Treinamentos. Produzo textos didáticos sobre tecnologia, buscando informar e engajar o público de forma acessível e descomplicada.

Ver mais posts