274 lines
9.8 KiB
JavaScript
274 lines
9.8 KiB
JavaScript
|
import axios from "axios";
|
|||
|
|
import path from "path";
|
|||
|
|
import { DEFINE_STRING } from "../../define/define_string";
|
|||
|
|
import { define } from "../../define/define";
|
|||
|
|
import { ImageStyleDefine } from "../../define/iamgeStyleDefine";
|
|||
|
|
import { cloneDeep } from 'lodash';
|
|||
|
|
let fspromises = require("fs").promises;
|
|||
|
|
const sharp = require('sharp');
|
|||
|
|
// const {
|
|||
|
|
// createCanvas,
|
|||
|
|
// loadImage
|
|||
|
|
// } = require('canvas');
|
|||
|
|
import { SdSettingDefine } from "../../define/setting/sdSettingDefine";
|
|||
|
|
import { PublicMethod } from "./publicMethod";
|
|||
|
|
import { Tools } from "../tools";
|
|||
|
|
|
|||
|
|
export class SD {
|
|||
|
|
constructor(global) {
|
|||
|
|
this.global = global;
|
|||
|
|
this.pm = new PublicMethod(global);
|
|||
|
|
this.tools = new Tools();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取图片风格菜单
|
|||
|
|
* @returns 返回图片风格菜单
|
|||
|
|
*
|
|||
|
|
* */
|
|||
|
|
async GetImageStyleMenu() {
|
|||
|
|
try {
|
|||
|
|
let style = ImageStyleDefine.getImageStyleMenu();
|
|||
|
|
return {
|
|||
|
|
code: 1,
|
|||
|
|
data: style
|
|||
|
|
}
|
|||
|
|
} catch (error) {
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取指定的ID的风格信息，传入的是一个数组
|
|||
|
|
* @param {*} value id集合
|
|||
|
|
*/
|
|||
|
|
async GetImageStyleInfomation(value) {
|
|||
|
|
try {
|
|||
|
|
if (value) {
|
|||
|
|
value = JSON.parse(value);
|
|||
|
|
} else {
|
|||
|
|
value = [];
|
|||
|
|
}
|
|||
|
|
value = value ? value : [];
|
|||
|
|
let style = ImageStyleDefine.getAllSubStyle();
|
|||
|
|
let tmp = [];
|
|||
|
|
for (let i = 0; i < value.length; i++) {
|
|||
|
|
const element = value[i];
|
|||
|
|
for (let j = 0; j < style.length; j++) {
|
|||
|
|
const item = style[j];
|
|||
|
|
if (item.id == element) {
|
|||
|
|
tmp.push(item);
|
|||
|
|
break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
let newSubStyle = cloneDeep(tmp);
|
|||
|
|
for (let i = 0; i < newSubStyle.length; i++) {
|
|||
|
|
const element = newSubStyle[i];
|
|||
|
|
element.image = path.join(define.image_path, "style/" + element.image);
|
|||
|
|
}
|
|||
|
|
return {
|
|||
|
|
code: 1,
|
|||
|
|
data: newSubStyle
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} catch (error) {
|
|||
|
|
return {
|
|||
|
|
code: 0,
|
|||
|
|
message: error.toString()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 获取指定ID的分类的子风格信息
|
|||
|
|
* @param {*} value ID
|
|||
|
|
* @returns 返回ID对应的子风格的详细信息
|
|||
|
|
*/
|
|||
|
|
async GetStyleImageSubList(value) {
|
|||
|
|
try {
|
|||
|
|
let subStyle = ImageStyleDefine.getImagePathById(value);
|
|||
|
|
let newSubStyle = cloneDeep(subStyle);
|
|||
|
|
for (let i = 0; i < newSubStyle.length; i++) {
|
|||
|
|
const element = newSubStyle[i];
|
|||
|
|
element.image = path.join(define.image_path, "style/" + element.image);
|
|||
|
|
}
|
|||
|
|
return {
|
|||
|
|
code: 1,
|
|||
|
|
data: newSubStyle
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} catch (error) {
|
|||
|
|
return {
|
|||
|
|
code: 0,
|
|||
|
|
message: error.toString()
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 生成一次图片的方法。可以区分模式
|
|||
|
|
* @param {图片名称 } image
|
|||
|
|
* @param {任务队列信息} task_list 301198499
|
|||
|
|
*/
|
|||
|
|
async OneImageGeneration(image, task_list, seed = -1) {
|
|||
|
|
let taskPath = path.join(this.global.config.project_path, "scripts/task_list.json")
|
|||
|
|
try {
|
|||
|
|
let imageJson = JSON.parse(await fspromises.readFile(image + '.json', 'utf-8'));
|
|||
|
|
let sd_setting = JSON.parse(await fspromises.readFile(define.sd_setting, 'utf-8'));
|
|||
|
|
let model = imageJson.model;
|
|||
|
|
let image_json = JSON.parse(await fspromises.readFile(image + '.json', 'utf-8'));
|
|||
|
|
let image_path = "";
|
|||
|
|
let target_image_path = "";
|
|||
|
|
|
|||
|
|
if (image_json.name) {
|
|||
|
|
image_path = path.join(this.global.config.project_path, `tmp/${task_list.out_folder}/tmp_${image_json.name}`)
|
|||
|
|
target_image_path = path.join(this.global.config.project_path, `tmp/${task_list.out_folder}/${image_json.name}`)
|
|||
|
|
} else {
|
|||
|
|
image_path = image.replaceAll("input_crop", task_list.out_folder).split(".png")[0] + "_tmp.png";
|
|||
|
|
target_image_path = image.replaceAll("input_crop", task_list.out_folder);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// let prompt = "";
|
|||
|
|
// // 拼接提示词
|
|||
|
|
// if (task_list.image_style != null) {
|
|||
|
|
// prompt += `((${task_list.image_style})),`;
|
|||
|
|
// }
|
|||
|
|
// if (task_list.lora != null) {
|
|||
|
|
// prompt += `${task_list.lora},`;
|
|||
|
|
// }
|
|||
|
|
// let image_styles = await ImageStyleDefine.getImageStyleStringByIds(task_list.image_style_list ? task_list.image_style_list : []);
|
|||
|
|
|
|||
|
|
// prompt = `${prompt}, ${image_styles}, ${imageJson.webui_config.prompt}`;
|
|||
|
|
let prompt = imageJson.webui_config.prompt;
|
|||
|
|
|
|||
|
|
// 判断当前是不是有开修脸修手
|
|||
|
|
let ADetailer = {
|
|||
|
|
args: sd_setting.adetailer
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
if (model == "img2img") {
|
|||
|
|
let web_api = this.global.config.webui_api_url + 'sdapi/v1/img2img'
|
|||
|
|
let sd_config = imageJson["webui_config"];
|
|||
|
|
sd_config.prompt = prompt;
|
|||
|
|
sd_config.seed = seed;
|
|||
|
|
let im = await fspromises.readFile(image, 'binary');
|
|||
|
|
sd_config.init_images = [new Buffer.from(im, 'binary').toString('base64')];
|
|||
|
|
|
|||
|
|
if (imageJson.adetailer) {
|
|||
|
|
let ta = {
|
|||
|
|
ADetailer: ADetailer
|
|||
|
|
}
|
|||
|
|
sd_config.alwayson_scripts = ta;
|
|||
|
|
}
|
|||
|
|
sd_config.height = sd_setting.webui.height;
|
|||
|
|
sd_config.width = sd_setting.webui.width;
|
|||
|
|
|
|||
|
|
const response = await axios.post(web_api, sd_config);
|
|||
|
|
let info = JSON.parse(response.data.info);
|
|||
|
|
if (seed == -1) {
|
|||
|
|
seed = info.seed;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 目前是单图出图
|
|||
|
|
let images = response.data.images;
|
|||
|
|
let imageData = Buffer.from(images[0].split(",", 1)[0], 'base64');
|
|||
|
|
await sharp(imageData)
|
|||
|
|
.toFile(image_path)
|
|||
|
|
.then(async () => {
|
|||
|
|
// console.log("图生图成功" + image_path);
|
|||
|
|
await this.tools.deletePngAndDeleteExifData(image_path, target_image_path);
|
|||
|
|
})
|
|||
|
|
.catch(err => {
|
|||
|
|
throw new Error(err);
|
|||
|
|
});
|
|||
|
|
return seed;
|
|||
|
|
|
|||
|
|
} else if (model == "txt2img") {
|
|||
|
|
let body = {
|
|||
|
|
"prompt": prompt,
|
|||
|
|
"negative_prompt": imageJson.webui_config.negative_prompt,
|
|||
|
|
"seed": seed,
|
|||
|
|
"sampler_name": imageJson.webui_config.sampler_name,
|
|||
|
|
// 提示词相关性
|
|||
|
|
"cfg_scale": imageJson.webui_config.cfg_scale,
|
|||
|
|
"width": sd_setting.webui.width,
|
|||
|
|
"height": sd_setting.webui.height,
|
|||
|
|
"batch_size": 1,
|
|||
|
|
"n_iter": 1,
|
|||
|
|
"steps": imageJson.webui_config.steps,
|
|||
|
|
"save_images": false,
|
|||
|
|
}
|
|||
|
|
let web_api = this.global.config.webui_api_url + 'sdapi/v1/txt2img';
|
|||
|
|
|
|||
|
|
if (imageJson.adetailer) {
|
|||
|
|
let ta = {
|
|||
|
|
ADetailer: ADetailer
|
|||
|
|
}
|
|||
|
|
body.alwayson_scripts = ta;
|
|||
|
|
}
|
|||
|
|
const response = await axios.post(web_api, body);
|
|||
|
|
let info = JSON.parse(response.data.info);
|
|||
|
|
if (seed == -1) {
|
|||
|
|
seed = info.seed;
|
|||
|
|
}
|
|||
|
|
// 目前是单图出图
|
|||
|
|
let images = response.data.images;
|
|||
|
|
let imageData = Buffer.from(images[0].split(",", 1)[0], 'base64');
|
|||
|
|
await sharp(imageData)
|
|||
|
|
.toFile(image_path)
|
|||
|
|
.then(async () => {
|
|||
|
|
// console.log("文生图成功" + image_path);
|
|||
|
|
await this.tools.deletePngAndDeleteExifData(image_path, target_image_path);
|
|||
|
|
})
|
|||
|
|
.catch(err => {
|
|||
|
|
// console.log(err)
|
|||
|
|
throw new Error(err);
|
|||
|
|
});
|
|||
|
|
return seed;
|
|||
|
|
} else {
|
|||
|
|
throw new Error("SD 模式错误");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
} catch (error) {
|
|||
|
|
// 当前队列执行失败移除整个批次的任务
|
|||
|
|
this.global.requestQuene.removeTask(task_list.out_folder, null)
|
|||
|
|
this.global.fileQueue.enqueue(async () => {
|
|||
|
|
// 记录失败状态
|
|||
|
|
let task_list_json = JSON.parse(await fspromises.readFile(taskPath, 'utf-8'));
|
|||
|
|
// 修改指定的列表的数据
|
|||
|
|
task_list_json.task_list.map(a => {
|
|||
|
|
if (a.id == task_list.id) {
|
|||
|
|
a.status = "error";
|
|||
|
|
a.errorMessage = error.toString();
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
// 写入
|
|||
|
|
await fspromises.writeFile(taskPath, JSON.stringify(task_list_json));
|
|||
|
|
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.IMAGE_TASK_STATUS_REFRESH, {
|
|||
|
|
out_folder: task_list.out_folder,
|
|||
|
|
status: "error"
|
|||
|
|
});
|
|||
|
|
})
|
|||
|
|
throw error;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
*文生图
|
|||
|
|
* @param {SD 请求的地址} url
|
|||
|
|
* @param {SD请求的body} body
|
|||
|
|
*/
|
|||
|
|
async txt2img(url, body) {
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
*图生图
|
|||
|
|
* @param {SD 请求的地址} url
|
|||
|
|
* @param {SD请求的body} body
|
|||
|
|
*/
|
|||
|
|
async img2img(url, body) {
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|