package main

import (
    "context"
    "fmt"
    "log"

    "github.com/plexusone/omniimage"
    "github.com/plexusone/omniimage/provider"
)

func main() {
    // Create client with OpenAI provider
    client, err := omniimage.NewClient(omniimage.ClientConfig{
        Providers: []omniimage.ProviderConfig{
            {
                Provider: omniimage.ProviderNameOpenAI,
                APIKey:   "sk-...", // or use OPENAI_API_KEY env var
            },
        },
    })
    if err != nil {
        log.Fatal(err)
    }
    defer client.Close()

    // Generate an image
    resp, err := client.Generate(context.Background(), &provider.GenerateRequest{
        Model:   omniimage.ModelDALLE3,
        Prompt:  "A serene mountain landscape at sunset",
        Size:    provider.Size1024x1024,
        Quality: provider.QualityHD,
    })
    if err != nil {
        log.Fatal(err)
    }

    fmt.Printf("Generated image: %s\n", resp.Images[0].URL)
}

client, _ := omniimage.NewClient(omniimage.ClientConfig{
    Providers: []omniimage.ProviderConfig{
        {
            Provider: omniimage.ProviderNameOpenAI,
            APIKey:   "sk-...", // or OPENAI_API_KEY env var
        },
    },
})

// DALL-E 3 - high quality, single image
resp, _ := client.Generate(ctx, &provider.GenerateRequest{
    Model:   omniimage.ModelDALLE3,
    Prompt:  "A futuristic cityscape",
    Size:    provider.Size1024x1024,
    Quality: provider.QualityHD,
    Style:   provider.StyleVivid,
})

// DALL-E 2 - multiple images, edit, variations
resp, _ := client.Generate(ctx, &provider.GenerateRequest{
    Model:  omniimage.ModelDALLE2,
    Prompt: "A cute robot",
    N:      4,
    Size:   provider.Size512x512,
})

client, _ := omniimage.NewClient(omniimage.ClientConfig{
    Providers: []omniimage.ProviderConfig{
        {
            Provider: omniimage.ProviderNameFal,
            APIKey:   "...", // or FAL_KEY env var
        },
    },
})

// FLUX Pro - high quality
resp, _ := client.Generate(ctx, &provider.GenerateRequest{
    Model:  omniimage.ModelFluxPro,
    Prompt: "A photorealistic portrait",
    Size:   provider.Size1024x1024,
})

// FLUX Schnell - fast generation
resp, _ := client.Generate(ctx, &provider.GenerateRequest{
    Model:  omniimage.ModelFluxSchnell,
    Prompt: "Abstract art",
    N:      4,
})

// Upscale an image
upscaled, _ := client.Upscale(ctx, &provider.UpscaleRequest{
    Model: omniimage.ModelClarityUpscaler,
    Image: "https://example.com/image.png",
    Scale: 2,
})