fusionbrain_api

README

fusionbrain_api

fusionbrain.ai , Kandinsky Библиотек для доступа к API https://fusionbrain.ai .

Настройка

Надо получить API Key и засунуть его в переменные среды FUSIONBRAIN_SECRET_KEY и FUSIONBRAIN_API_KEY или выставить свойства SecretKey и ApiKey обьекту класса Fusionbrain

Примеры

См. директорию examples/

Documentation

Overview

Package fusionbrain_api Предоставляет доступ API на Fusionbrain.ai

Этот пакет сделан для того чтоб генерить картинки в основном нейросетью Кадински от сбера.

Index

• Constants
• type AvailabilityResponse
• type Fusionbrain
  • func NewFusionbrain() *Fusionbrain
  • func (f *Fusionbrain) Generate(query string, negativeQuery string, style string) (GenerateResponse, error)
  • func (f *Fusionbrain) GetModels() (ModelsResponse, error)
  • func (f *Fusionbrain) ImageToFile(response GenerateResponse, imageIndex int, filename string) (*os.File, error)
  • func (f *Fusionbrain) ImageToReader(response GenerateResponse, imageIndex int) (io.Reader, error)
  • func (f *Fusionbrain) СheckStatus(uuidOfRequest string) (GenerateResponse, error)
• type GenerateParams
• type GenerateRequest
• type GenerateResponse
• type ModelItem
• type ModelsResponse
• type StatusResponse
• type StyleItem
• type StyleResponse

Constants

const (
	FusionbrainGenerateStatusINITIAL    = "INITIAL"    //запрос получен, находится в очереди на обработку
	FusionbrainGenerateStatusPROCESSING = "PROCESSING" // запрос находится в процессе обработки
	FusionbrainGenerateDONE             = "DONE"       // задание выполнено
	FusionbrainGenerateFAIL             = "FAIL"       // задание не удалось выполнить.
)

статусы генерации

const (
	FusionbrainModelStatusDisableByQueue = "DISABLED_BY_QUEUE"
)

Доступность сервиса

Types

type AvailabilityResponse

type AvailabilityResponse struct {
	ModelStatus string `json:"model_status"`
}

Доступность

type Fusionbrain

type Fusionbrain struct {
	ApiHost      string
	ApiKey       string
	SecretKey    string
	Style        string
	CurrentModel ModelItem
}

func NewFusionbrain

func NewFusionbrain() *Fusionbrain

func (*Fusionbrain) Generate

func (f *Fusionbrain) Generate(query string, negativeQuery string, style string) (GenerateResponse, error)

запускаем генерацию картинки

func (*Fusionbrain) GetModels

func (f *Fusionbrain) GetModels() (ModelsResponse, error)

func (*Fusionbrain) ImageToFile

func (f *Fusionbrain) ImageToFile(response GenerateResponse, imageIndex int, filename string) (*os.File, error)

Сохраняем картинку из ответа в файл

func (*Fusionbrain) ImageToReader

func (f *Fusionbrain) ImageToReader(response GenerateResponse, imageIndex int) (io.Reader, error)

Декодируем из ответа картинку в байтовый поток

func (*Fusionbrain) СheckStatus

func (f *Fusionbrain) СheckStatus(uuidOfRequest string) (GenerateResponse, error)

type GenerateParams

type GenerateParams struct {
	Query string `json:"query"`
}

Текстовый промпт для генерации

type GenerateRequest

type GenerateRequest struct {
	Type                 string         `json:"type"`
	Style                string         `json:"style,omitempty"`
	Width                int            `json:"width,omitempty"`
	Height               int            `json:"height,omitempty"`
	NumImages            int            `json:"num_images,omitempty"`
	NegativePromptUnclip string         `json:"negativePromptUnclip,omitempty"`
	GenerateParams       GenerateParams `json:"generateParams"`
}

Запрос генерации

type GenerateResponse

type GenerateResponse struct {
	Uuid             string   `json:"uuid"`
	Status           string   `json:"status"`
	StatusTime       int      `json:"status_time,omitempty"`
	Images           []string `json:"images,omitempty"`
	ErrorDescription string   `json:"errorDescription,omitempty"`
	Censored         bool     `json:"censored,omitempty"`
}

Ответ на генерацию

type ModelItem

type ModelItem struct {
	Id      int     `json:"id"`
	Name    string  `json:"name"`
	Version float64 `json:"version"`
	Type    string  `json:"type"`
}

Модель

type ModelsResponse

type ModelsResponse []ModelItem

Ответ со списком моделей

type StatusResponse

type StatusResponse struct {
	Images []string `json:"images"`
}

type StyleItem

type StyleItem struct {
	Name    string `json:"name"`
	Title   string `json:"title"`
	TitleEn string `json:"titleEn"`
	Image   string `json:"image"`
}

Стиль

type StyleResponse

type StyleResponse []StyleItem

Список стилей