from openai import OpenAI
from dotenv import load_dotenv
import os

load_dotenv()

# Rufe den API-Key sicher ab
api_key = os.getenv("OPENAI_API_KEY")
if not api_key:
    raise ValueError("API Key ist nicht gesetzt. Pruefe deine .env Datei oder die Umgebungsvariablen.")

client = OpenAI(api_key=api_key)

# Prompt
n = 3
user = "test-1"
size = "256x256"
img = open("C:\Pythonprojekte\image_variations_with_plot_output\dino.png", "rb")
 
response = client.images.create_variation(image=img, n=n, user=user, size=size)

import matplotlib.pyplot as plt
import imageio.v3 as iio

# Ausgabe
for i in range(n):
    img = iio.imread(response.data[i].url)
    plt.imshow(img)
    plt.axis('off')  # Optional, um die Achsen auszublenden
    plt.show()  # Zeigt die Grafik an