V 2.2.9 自动匹配,手动排序
This commit is contained in:
@@ -3,7 +3,7 @@ import path, { resolve } from "path";
|
||||
import { Tools } from "../tools";
|
||||
import fs from "fs";
|
||||
import { ImageSetting } from "../../define/setting/imageSetting";
|
||||
import { isEmpty, reject } from "lodash";
|
||||
import { isEmpty } from "lodash";
|
||||
import { basicApi } from "../../api/apiBasic";
|
||||
import sharp from 'sharp';
|
||||
import { file, image } from "../../define/Tools";
|
||||
|
||||
@@ -0,0 +1,82 @@
|
||||
import { errorMessage, successMessage } from "../generalTools";
|
||||
import { LOGGER_DEFINE } from '../../define/logger_define'
|
||||
import { Setting } from "../setting/setting";
|
||||
|
||||
export class Prompt {
|
||||
constructor() {
|
||||
this.setting = new Setting(global);
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取所有的排序选项
|
||||
*/
|
||||
GetPromptSortOptions() {
|
||||
return successMessage([
|
||||
{
|
||||
label: '人物',
|
||||
value: 'character'
|
||||
}, {
|
||||
label: '场景',
|
||||
value: 'scene'
|
||||
}, {
|
||||
label: '提示词',
|
||||
value: 'prompt'
|
||||
}, {
|
||||
label: '风格',
|
||||
value: 'style'
|
||||
}
|
||||
], null, LOGGER_DEFINE.PROMPT.GET_PROMPT_SORT_OPTIONS);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* 保存SD排序的选项
|
||||
* @param {*} value
|
||||
*/
|
||||
async SavePromptSort(value) {
|
||||
try {
|
||||
// 将当前数据保存
|
||||
await this.setting.SaveDefineConfigJsonByProperty(JSON.stringify(['img_base', 'prompt_sort', value]));
|
||||
return successMessage(null, "保存提示词排序顺序成功", LOGGER_DEFINE.PROMPT.SAVE_PROMPT_SORT_DATA);
|
||||
|
||||
} catch (error) {
|
||||
return errorMessage(error.toString(), LOGGER_DEFINE.PROMPT.SAVE_PROMPT_SORT_DATA);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取当前的
|
||||
*/
|
||||
async GetPromptSort() {
|
||||
try {
|
||||
let res = await this.setting.GetDefineConfigJsonByProperty(JSON.stringify(['img_base', 'prompt_sort', false, []]))
|
||||
if (res.code == 0) {
|
||||
throw new Error(res.message);
|
||||
}
|
||||
res = res.data;
|
||||
if (res && res.length == 0) {
|
||||
res = [
|
||||
{
|
||||
label: '风格',
|
||||
value: 'style'
|
||||
},
|
||||
{
|
||||
label: '人物',
|
||||
value: 'character'
|
||||
},
|
||||
{
|
||||
label: '场景',
|
||||
value: 'scene'
|
||||
},
|
||||
{
|
||||
label: '提示词',
|
||||
value: 'prompt'
|
||||
}
|
||||
]
|
||||
}
|
||||
return successMessage(res, "获取数据成功", LOGGER_DEFINE.PROMPT.GET_PROMPT_SORT_DATA)
|
||||
} catch (error) {
|
||||
return errorMessage(error.toString(), LOGGER_DEFINE.PROMPT.GET_PROMPT_SORT_DATA)
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -110,7 +110,7 @@ export class SD {
|
||||
return successMessage(data);
|
||||
|
||||
} catch (error) {
|
||||
return errorMessage("加载数据失败,错误信息如下:" +error.toString());
|
||||
return errorMessage("加载数据失败,错误信息如下:" + error.toString());
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user