ai-infra / API Reference

ImageGen

from ai_infra.imagegen import ImageGen

ai_infra.imagegen

Provider-agnostic image generation. Supports OpenAI (DALL-E), Google (Imagen), Stability AI, and Replicate. Auto-detects provider from environment variables if not specified.

python

# Zero-config: auto-detects from env vars
    gen = ImageGen()
    images = gen.generate("A sunset over mountains")

    # Explicit provider and model
    gen = ImageGen(provider="google", model="imagen-4.0-generate-001")
    images = gen.generate("A futuristic city", n=2)

    # Save generated image
    images[0].save("output.png")

Environment Variables: - OPENAI_API_KEY: For OpenAI DALL-E - GOOGLE_API_KEY or GEMINI_API_KEY: For Google Imagen - STABILITY_API_KEY: For Stability AI - REPLICATE_API_TOKEN: For Replicate

Constructor

ImageGen(provider: str | None = None, model: str | None = None, api_key: str | None = None, kwargs: Any = {}) -> None

Parameter	Type	Default	Description
`provider`	`str\|None`	None	Provider name ("openai", "google", "stability", "replicate"). Auto-detected from env vars if not specified.
`model`	`str\|None`	None	Model name. Uses provider default if not specified.
`api_key`	`str\|None`	None	API key. Uses env var if not specified.
`kwargs`	`Any`	{}	—

Methods

# Zero-config: auto-detects from env vars gen = ImageGen() images = gen.generate("A sunset over mountains") # Explicit provider and model gen = ImageGen(provider="google", model="imagen-4.0-generate-001") images = gen.generate("A futuristic city", n=2) # Save generated image images[0].save("output.png")

Parameter

Type

Default

Description

provider

str|None

None

Provider name ("openai", "google", "stability", "replicate"). Auto-detected from env vars if not specified.

model

str|None

None

Model name. Uses provider default if not specified.

api_key

str|None

None

API key. Uses env var if not specified.

kwargs

Any

{}

—