Un sencillo generador de imágenes mediante la API de OpenAI
Estos son algunos ejemplos
import os # Importamos la libreria OS, para poder usar comandos de consola
import random # Importamos la libreria random, para poder generar un nombre de archivo aleatorio
import string # Importamos la libreria string, para poder asignar unos parametros a la variable del nombre de archivo
import openai # Importamos la libreria de open ai, la cual nos permite usar la api de open ai
import requests # Importamos la libreria requests, para poder descargar la imagen generada
from colorama import Fore # Y por ultimo importamos Fore desde la libreria de Colorama, para darle un poco de color a la consola
os.system("cls") # Limpiamos la consola
# Y aquí definimos las variables de los colores para tenerlos mas ordenados
w = Fore.LIGHTWHITE_EX
r = Fore.LIGHTRED_EX
g = Fore.LIGHTGREEN_EX
b = Fore.LIGHTBLUE_EX
y = Fore.LIGHTYELLOW_EX
m = Fore.LIGHTMAGENTA_EX
c = Fore.LIGHTCYAN_EX
black = Fore.LIGHTBLACK_EX
openai.organization = "ORGANIZACION_ID_AQUI" # Aquí debes poner el ID de tu organizacion que puedes encontrar en -> https://beta.openai.com/account/org-settings
openai.api_key = "OPENAI-APIKEY-AQUI" # Aquí debes poner tu API KEY que puedes encontrar en -> https://beta.openai.com/account/api-keys
openai.Model.list()
os.system("title franafp - Image Generator - OPEN AI") # Aquí le damos un titulo a la consola
print(f"{m}[{w}1{m}] {black}Generador de imágenes")
def image_gen(): # Aquí definimos la funcion de la generacion de imagenes
image=input(f"{m}[{w}>>>{m}] {black}Deja tus argumentos aquí?:{w} ") # Aquí pedimos los argumentos para la imagen
print(f"{m}[{w}>{m}] {black}Generando imagen{m}...")
r = openai.Image.create(prompt=image, n=2, size="1024x1024") # Aquí le decimos a la api que genere la imagen con los argumentos que le dimos
image_url = r["data"][0]["url"] # Aquí le decimos que guarde la url de la imagen generada
os.system("cls")
print(f"{m}[{w}!{m}] {black}Imagen generada: {y}{image_url}")
print(f"{image_url}")
save=input(f"{m}[{w}>>>{m}] {black}Quieres guardar la imagen? [y/n]:{w} ") # Aquí le preguntamos si quiere guardar la imagen
if save == "y": # Si la respuesta es "y" entonces...
path="./img/" # Aquí le decimos que guarde la imagen en la carpeta "img"
nombre_random= "".join(random.choice(string.ascii_letters + string.digits) for i in range(10)) # Aquí le decimos que genere un nombre aleatorio para la imagen
nombre_archivo = f"{path}{nombre_random}.png" # Aquí le decimos que guarde la imagen con el nombre aleatorio y la extension .png
imagen_save= requests.get(image_url).content # Aquí le decimos que descargue la imagen"
with open(nombre_archivo, 'wb') as i:
i.write(imagen_save)
print(f"{m}[{w}!{m}] {black}Imagen guardada en: {y}{nombre_archivo}")
openask=input("¿Quieres abrir la imagen? [y/n]: ") # Aquí le preguntamos si quiere abrir la imagen en el explorador
if openask == "y": # Si la respuesta es "y" entonces...
os.system(f"""explorer {image_url}""") # Aquí le decimos que abra la imagen en el explorador
else: # Y si la respuesta es cualquier otra, entonces simplemente le decimos que salga del programa pulsando cualquier tecla
print("OK")
print("Pulsa cualquier tecla para salir")
os.system("pause >nul")
os.system("exit")
image_gen()