Golang

Omnicron Golang client library

Omnicron Golang API Library

The Omnicron Golang library provides convenient access to the Omnicron REST API from any Go 1.20+ application. The library includes type definitions for all request parameters and response fields, and offers both synchronous and asynchronous clients.

Installation

bash

go get github.com/kingmariano/omnicron-go

Usage

Use the library and your secret key and your public server URL to run:

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"github.com/kingmariano/omnicron-go"
	"os"
)

func main() {
	apiKey := "YOUR_API_KEY"
	client := omnicron.NewClient(apiKey, omnicron.WithBaseURL("https://omnicron-latest.onrender.com/"))
	imageFile, err := os.Open("image.jpg") //example image, edit
	if err != nil {
		fmt.Printf("Error opening image file: %v\n", err)
		return
	}
	res, err := client.HighImageGeneration(context.Background(), omnicron.HighImageGenerationModelAndParams{
		Model: omnicron.PlaygroundV251024pxAestheticModel,
		Parameters: omnicron.HighImageGenerationParams{
			Prompt:            "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
			NegativePrompt:    omnicron.Ptr("ugly, deformed, noisy, blurry, distorted"),
			ImageFile:         imageFile,
			Width:             omnicron.Ptr(1024),
			Height:            omnicron.Ptr(1024),
			NumOutputs:        omnicron.Ptr(1),
			Scheduler:         omnicron.Ptr("DPMSolver++"),
			NumInferenceSteps: omnicron.Ptr(30),
			GuidanceScale:     omnicron.Ptr(3.1),
			PromptStrength:    omnicron.Ptr(0.7),
		},
	})
	if err != nil {
		fmt.Printf("Error making High ImageGeneration request: %v\n", err)
		return
	}

	// Marshal the response to JSON
	jsonData, err := json.MarshalIndent(res, "", "  ")
	if err != nil {
		fmt.Printf("Error marshaling JSON: %v\n", err)
		return
	}

	// Write JSON data to a file
	file, err := os.Create("file.json")
	if err != nil {
		fmt.Printf("Error creating file: %v\n", err)
		return
	}
	defer file.Close()

	_, err = file.Write(jsonData)
	if err != nil {
		fmt.Printf("Error writing JSON to file: %v\n", err)
		return
	}

	fmt.Println("JSON data written to file.json successfully")
	// dynamically handle the response with the Gabs library: https://github.com/Jeffail/gabs/
	outputText := res.Path("response.output").Data().(map[string]interface{})
	fmt.Println(outputText)

}

Check the full documentation here