<!-- * @Author: SpenserCai * @Date: 2023-08-12 01:27:12 * @version: * @LastEditors: SpenserCai, drgrib * @LastEditTime: 2023-09-13 10:51:18 * @Description: 文件内容 --> <div align="center"> <img src="https://yellow-cdn.veclightyear.com/35dd4d3f/1cd842c2-43be-422c-980c-fc0eab8f408c.png" width="200" height="200" alt="sd-webui-go">

SD-WEBUI-GO

Golang SDK for stable-diffusion-webui 的 API

贡献者中有 AUTOMATIC1111，这是我的一个小玩笑😂。

在此，诚挚感谢 AUTOMATIC1111 对 AIGC 的巨大贡献

</div> <p align="center"> <a href="https://raw.githubusercontent.com/SpenserCai/sd-webui-go/main/LICENSE"> <img src="https://yellow-cdn.veclightyear.com/35dd4d3f/8353e3bd-5450-43ae-8f1f-6f08f7287c92.png" alt="license"> </a> <img src="https://yellow-cdn.veclightyear.com/35dd4d3f/beddb696-0ea3-4a4b-88f9-851892d9b874.19+-blue" alt="go"> <a href="https://github.com/SpenserCai/sd-webui-go/releases"> <img src="https://yellow-cdn.veclightyear.com/35dd4d3f/53b07c61-1304-4f99-93c6-358f7f7d5a8a.png" alt="release"> </a> <a href="https://goreportcard.com/report/github.com/SpenserCai/sd-webui-go"> <img src="https://yellow-cdn.veclightyear.com/35dd4d3f/cd8bbb5d-f0a2-4bed-8318-014bca341f4d.png" alt="GoReportCard"> </a> <a href="https://discord.gg/nN6b7QAcVW"> <img src="https://yellow-cdn.veclightyear.com/35dd4d3f/aa704d4d-e33c-4258-8149-be83606b20b3.png?style=shield" alt="Discord Server"> </a> </p>

这是一个基于 stable-diffusion-webui 的 Go 语言版本 SDK。在您的代码中，可以通过面向对象的操作直接使用 stable-diffusion-webui 的 API 接口，而不需要处理繁琐的 JSON。

支持扩展 API ！

<b>您可以在这个 wiki 页面 查看 intersvc 的支持列表。</b>

使用方法

有两种方法可以使用这个 SDK。

• 使用 'intersvc'，它为 'sd-webui' 提供了高度封装的 API 接口。使用此方法感觉像是在直接使用 'sd-webui' 的 Go 语言版本。

• 使用 'go-swagger'，虽然仍然涉及面向对象的操作，但比 'intersvc' 略复杂。

第二种方法几乎支持所有接口，而第一种方法正逐步得到支持。

实际上，大部分接口可以通过 'intersvc' 使用，但需要定义响应。'sd-webui' 的 API 文档没有提供响应的内容，所以需要手动定义。您可以参考下面的 "参与" 部分的具体说明。

go-swagger 的 text2img

package main

import (
	"encoding/base64"
	"fmt"

	SdClient "github.com/SpenserCai/sd-webui-go"
	SdApiOperation "github.com/SpenserCai/sd-webui-go/stablediffusion/client/operations"
	SdApiModel "github.com/SpenserCai/sd-webui-go/stablediffusion/models"
)

func MustBeNil(err error) {
	if err != nil {
		panic(err)
	}
}

var URL = "127.0.0.1:7860"

func main() {
	sdClient := SdClient.NewStableDiffInterface(URL)

	rd := SdApiOperation.NewText2imgapiSdapiV1Txt2imgPostParams()
	rd.Body = &SdApiModel.StableDiffusionProcessingTxt2Img{
		Prompt:         "dog",
		NegativePrompt: "ugly",
		ScriptArgs:     []interface{}{},
	}

	resp, err := sdClient.Client.Operations.Text2imgapiSdapiV1Txt2imgPost(rd)
	MustBeNil(err)

	for i, s := range resp.Payload.Images {
		b, err := base64.StdEncoding.DecodeString(s)
		MustBeNil(err)
		err = os.WriteFile(fmt.Sprintf("%d.png", i), b, 0644)
		MustBeNil(err)
	}
}

intersvc 的 deoldify

import (
	SdClient "github.com/SpenserCai/sd-webui-go"
	"github.com/SpenserCai/sd-webui-go/intersvc"
)

func main() {
	// 创建客户端
	sdClient := SdClient.NewStableDiffInterface("127.0.0.1:7860")

	var f_factor int64 = 20
	var artistic bool = false

	// 设置请求
	deoldify_inter := &intersvc.DeoldifyImage{
		RequestItem: &intersvc.DeoldifyImageRequest{
			InputImage:   "https://media.discordapp.net/attachments/1138408545277190237/1138508881635577947i7krs1njekla1.jpg",
			RenderFactor: &f_factor,
			Artistic:     &artistic,
		},
	}


	deoldify_inter.Action(sdClient)
	if deoldify_inter.Error != nil {
		panic(deoldify_inter.Error)
	}

	response := deoldify_inter.GetResponse()
}

完整示例代码: intersvc_example

go-swagger 的 deoldify

import (
	"encoding/base64"
	"os"
	
	SdClient "github.com/SpenserCai/sd-webui-go"
	intersvc "github.com/SpenserCai/sd-webui-go/intersvc"
	SdApiOperation "github.com/SpenserCai/sd-webui-go/stablediffusion/client/operations"
	SdApiModel "github.com/SpenserCai/sd-webui-go/stablediffusion/models"
)

type DeoldifyResponse struct {
	Image string `json:"image"`
}

func main() {
	var (
		err error
	)

	// 初始化客户端
	sdClient := SdClient.NewStableDiffInterface("127.0.0.1:7860")

	var f_factor int64 = 20
	var artistic bool = false

	// 设置请求数据
	RequestData := SdApiOperation.NewDeoldifyImageDeoldifyImagePostParams()
	RequestData.Body = &SdApiModel.BodyDeoldifyImageDeoldifyImagePost{
		InputImage:   "https://media.discordapp.net/attachments/1138408545277190237/1138508881635577947/i7krs1njekla1.jpg",
		RenderFactor: &f_factor,
		Artistic:     &artistic,
	}

	// 发送请求
	Response, err := sdClient.Client.Operations.DeoldifyImageDeoldifyImagePost(RequestData)
	if err != nil {
		panic(err)
	}

	// 转换响应
	deoldifyRep, err := intersvc.ConvertResponse(Response.Payload, &DeoldifyResponse{})
	if err != nil {
		panic(err)
	}

    response := deoldifyRep.(*DeoldifyResponse)
}

完整示例代码: go-swagger_example

参与

大部分 intersvc 的代码是使用代码生成器生成的。但是，由于 sd-webui 的 API 文档中缺乏响应信息，这些需要手动编写。

如何提交 PR

您需要 fork 当前代码库的 dev 分支，进行必要的代码更新，在自己的仓库中创建一个名为 dev-[model filename] 的分支，然后提交一个 pull request 到当前代码库的 dev 分支。

如何定义响应模型

在 sd-webui-go/intersvc 中您可以看到一些像 ***_model.go 的文件。这些文件定义了响应模型。您可以参照以下示例来定义响应模型。

从

type DeoldifyImageResponse struct {
	
}

到

type DeoldifyImageResponse struct {
	Image string `json:"image"`
}