91 lines
2.6 KiB
JavaScript
91 lines
2.6 KiB
JavaScript
|
import { basicApi } from "./apiBasic";
|
|||
|
|
import { define } from "../define/define";
|
|||
|
|
import { promises as fspromises } from 'fs'
|
|||
|
|
import { errorMessage, successMessage } from "../main/generalTools";
|
|||
|
|
|
|||
|
|
|
|||
|
|
export class SdApi {
|
|||
|
|
constructor() {
|
|||
|
|
this.baseUrl = global.config?.webui_api_url;
|
|||
|
|
this.sd_setting = null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取当前SD的服务器中所有的lora信息
|
|||
|
|
* @returns
|
|||
|
|
*/
|
|||
|
|
async getAllLoras(baseURL = null) {
|
|||
|
|
let url = this.baseUrl + "sdapi/v1/loras";
|
|||
|
|
if (baseURL != null) {
|
|||
|
|
url = baseURL + "sdapi/v1/loras";
|
|||
|
|
}
|
|||
|
|
return await basicApi.get(url);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取当前的所有的checkpoint模型
|
|||
|
|
* @param {*} baseURL
|
|||
|
|
*/
|
|||
|
|
async getAllSDModel(baseURL = null) {
|
|||
|
|
let url = this.baseUrl + "sdapi/v1/sd-models";
|
|||
|
|
if (baseURL != null) {
|
|||
|
|
url = baseURL + "sdapi/v1/sd-models";
|
|||
|
|
}
|
|||
|
|
return await basicApi.get(url);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取当前连接的所有的samplers（采样器）
|
|||
|
|
* @param {*} baseURL
|
|||
|
|
* @returns
|
|||
|
|
*/
|
|||
|
|
async getAllSamplers(baseURL = null) {
|
|||
|
|
try {
|
|||
|
|
|
|||
|
|
let url = this.baseUrl + "sdapi/v1/samplers";
|
|||
|
|
if (baseURL != null) {
|
|||
|
|
url = baseURL + "sdapi/v1/samplers";
|
|||
|
|
}
|
|||
|
|
return await basicApi.get(url);
|
|||
|
|
} catch (error) {
|
|||
|
|
throw error;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async txt2img(data, baseURL = null) {
|
|||
|
|
try {
|
|||
|
|
|
|||
|
|
if (this.sd_setting == null) {
|
|||
|
|
this.sd_setting = JSON.parse(await fspromises.readFile(define.sd_setting, 'utf-8'));
|
|||
|
|
this.baseUrl = this.sd_setting.setting.webui_api_url;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 加上通用前缀
|
|||
|
|
data.prompt = this.sd_setting.webui.prompt + data.prompt
|
|||
|
|
|
|||
|
|
data.negative_prompt = this.sd_setting.webui.negative_prompt;
|
|||
|
|
data.sampler_name = this.sd_setting.webui.sampler_name;
|
|||
|
|
data.cfg_scale = this.sd_setting.webui.cfg_scale;
|
|||
|
|
data.n_iter = 1;
|
|||
|
|
data.steps = this.sd_setting.webui.steps;
|
|||
|
|
data.save_images = false;
|
|||
|
|
data.batch_size = data.batch_size ? data.batch_size : 1;
|
|||
|
|
if (data.width == null) {
|
|||
|
|
data.width = 512;
|
|||
|
|
}
|
|||
|
|
if (data.height == null) {
|
|||
|
|
data.height = 512;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
let url = this.baseUrl + "sdapi/v1/txt2img";
|
|||
|
|
if (baseURL != null) {
|
|||
|
|
url = baseURL + "sdapi/v1/txt2img";
|
|||
|
|
}
|
|||
|
|
let res = await basicApi.post(url, data);
|
|||
|
|
return res;
|
|||
|
|
} catch (error) {
|
|||
|
|
throw error;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|