images-example.go

package main

import (
	"fmt"
	"os"

	"github.com/Kardbord/gopenai/authentication"
	"github.com/Kardbord/gopenai/images"
	_ "github.com/joho/godotenv/autoload"
)

const OpenAITokenEnv = "OPENAI_API_KEY"

func init() {
	key := os.Getenv(OpenAITokenEnv)
	authentication.SetAPIKey(key)
}

func create(model, size string) (*images.Response, error) {
	const prompt = "A cute baby sea otter"

	fmt.Printf("Creating from model=\"%s\", prompt=\"%s\"\n", model, prompt)
	resp, _, err := images.MakeModeratedCreationRequest(&images.CreationRequest{
		Prompt: prompt,
		Size:   size,
		User:   "https://github.com/Kardbord/gopenai",
		Model:  model,
	}, nil)
	if err != nil {
		return nil, err
	}

	fmt.Printf("Generated: %s\n", resp.Data[0].URL)
	return resp, nil
}

func variation(imagename, image string) error {

	fmt.Printf("Generating a variation...")
	resp, err := images.MakeVariationRequest(&images.VariationRequest{
		Image:     image,
		ImageName: imagename,
		Size:      images.Dalle2SmallImage,
		User:      "https://github.com/Kardbord/gopenai",
	}, nil)
	if err != nil {
		return err
	}

	fmt.Printf("Generated: %s\n", resp.Data[0].URL)
	return nil
}

func main() {
	resp, err := create(images.ModelDalle2, images.Dalle2SmallImage)
	if err != nil {
		fmt.Println(err)
		return
	}

	err = variation("Original", resp.Data[0].URL)
	if err != nil {
		fmt.Println(err)
		return
	}

	_, err = create(images.ModelDalle3, images.Dalle3SquareImage)
	if err != nil {
		fmt.Println(err)
		return
	}
}