AI Image Generation API

from openai import OpenAI

# Use OpenAI SDK with VoltageGPU
client = OpenAI(
    base_url="https://api.voltagegpu.com/v1",
    api_key="YOUR_VOLTAGE_API_KEY"
)

# Generate an image with FLUX.1 Schnell
response = client.images.generate(
    model="black-forest-labs/FLUX.1-schnell",
    prompt="A futuristic cityscape at sunset, "
           "cyberpunk aesthetic, neon lights reflecting "
           "on wet streets, ultra detailed, 8k",
    size="1024x1024",
    n=1,
)

image_url = response.data[0].url
print(f"Generated image: {image_url}")

# Batch generation example
prompts = [
    "Product photo of white sneakers on marble background",
    "Minimalist logo design for a tech startup",
    "Illustration of a mountain landscape in watercolor style",
]

for prompt in prompts:
    result = client.images.generate(
        model="black-forest-labs/FLUX.1-schnell",
        prompt=prompt,
        size="1024x1024",
    )
    print(f"Image: {result.data[0].url}")