admin777/LaiTool
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

LaiTool 3.0.1

Browse Source
This commit is contained in:
lq1405 2024-08-20 10:37:38 +08:00
parent 223678f761
commit 07e8a5a657
28 changed files with 794 additions and 548 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
package-lock.json generated
View File

@ -1,12 +1,12 @@
{
"name": "laitool",
"version": "3.0.1-preview.7",
"version": "3.0.2",
"lockfileVersion": 3,
"requires": true,
"packages": {
"": {
"name": "laitool",
"version": "3.0.1-preview.7",
"version": "3.0.2",
"hasInstallScript": true,
"dependencies": {
"@alicloud/alimt20181012": "^1.2.0",

2
package.json
View File

@ -1,6 +1,6 @@
{
"name": "laitool",
"version": "3.0.1-preview.7",
"version": "3.0.2",
"description": "An AI tool for image processing, video processing, and other functions.",
"main": "./out/main/index.js",
"author": "laitool.cn",

BIN
resources/scripts/db/book.realm.lock
View File

Binary file not shown.

BIN
resources/scripts/db/software.realm.lock
View File

Binary file not shown.

28
src/define/Tools/common.ts
View File

@ -9,3 +9,31 @@ export function ContainsChineseOrPunctuation(str: string): boolean {
str
)
}
/**
* 通用的重试函数
* @param fn 要执行的函数
* @param retries 最大重试次数
* @param delay 每次重试之间的延迟（毫秒）
* @returns 返回函数的结果
*/
export async function RetryWithBackoff<T>(fn: () => Promise<T>, retries: number = 5, delay: number = 2000): Promise<T> {
let attempts = 0;
while (attempts < retries) {
try {
return await fn();
} catch (error) {
attempts++;
// 这边记下日志吧
global.logger.error(
fn.name + '_RetryWithBackoff',
`第 ${attempts} 请求失败，开始下一次重试，失败信息如下：` + error.toString()
)
if (attempts >= retries) {
throw new Error(`失败次数超过 ${retries} 错误信息如下: ${error.message}`);
}
await new Promise(resolve => setTimeout(resolve, delay));
}
}
throw new Error('所有重试失败'); // 理论上不会到达这里
}

2
src/main/IPCEvent/bookIpc.ts
View File

@ -120,7 +120,7 @@ export function BookIpc() {
// 开始执行获取小说文案的方法
ipcMain.handle(
DEFINE_STRING.BOOK.GET_COPYWRITING,
async (event, bookId, bookTaskId, operateBookType) => await subtitleService.GetCopywriting(bookId, bookTaskId, operateBookType)
async (event, bookId, bookTaskId, operateBookType, coverData) => await subtitleService.GetCopywriting(bookId, bookTaskId, operateBookType, coverData)
)
// 获取小说的文案数据，然后保存到对应的文件中

30
src/main/Original/MJOriginalImageGenerate.js
View File

@ -161,13 +161,13 @@ export class MJOriginalImageGenerate {
* 初始化MJ设置
*/
async InitMjSetting() {
let mjSetting_res = await ImageSetting.GetDefineConfigJsonByProperty(
JSON.stringify(['img_base', 'mj_config', false, null])
)
if (mjSetting_res.code == 0 || !mjSetting_res.data) {
throw new Error('请先添加MJ配置')
// 获取MJ配置，从数据库中
let _mjSettingService = await MJSettingService.getInstance()
let mjSettings = _mjSettingService.GetMJSettingTreeData()
if (mjSettings.code == 0) {
throw new Error(mjSettings.message)
}
let mjSetting = mjSetting_res.data
let mjSetting = mjSettings.data
return mjSetting
}
@ -247,6 +247,7 @@ export class MJOriginalImageGenerate {
let result = []
// 浏览器生图模式
if (request_model == 'browser_mj') {
throw new Error('该模式不支持获取已经生图完成的数据，并获取图片')
let param = []
// 循环数据，直传需要的数据
for (let i = 0; i < value.length; i++) {
@ -285,18 +286,18 @@ export class MJOriginalImageGenerate {
// 请求
for (let i = 0; i < value.length; i++) {
const element = value[i]
if (element.mj_message.progress == 100) {
continue
}
if (element.mj_message.progress.status == 'success') {
continue
}
// if (element.mj_message.progress == 100) {
// continue
// }
// if (element.mj_message.progress.status == 'success') {
// continue
// }
let task_res = await this.discordAPI.GetMJAPITaskByID(
element.mj_message.message_id,
once_get_task,
mjSetting.apiSetting.apiKey
)
) // 这边价格判断，是不是真的结束了
if (task_res.code == 0) {
task_res['id'] = element.id
task_res['mj_api_url'] = mjSetting.apiSetting.mjApiUrl
@ -306,7 +307,6 @@ export class MJOriginalImageGenerate {
if (task_res.progress != 100) {
continue
}
result.push({
id: element.id,
image_id: null,
@ -314,6 +314,8 @@ export class MJOriginalImageGenerate {
name: element.name
})
}
} else {
throw new Error('未知的生图模式，请检查配置')
}
let res = []

330
src/main/Public/GPT.js
View File

@ -1,34 +1,38 @@
import axios from "axios";
import path from "path";
import { DEFINE_STRING } from "../../define/define_string";
import { define } from "../../define/define";
let fspromises = require("fs").promises;
import { gptDefine } from "../../define/gptDefine";
import { apiUrl } from "../../define/api/apiUrlDefine";
import { successMessage } from "../Public/generalTools";
import axios from 'axios'
import path from 'path'
import { DEFINE_STRING } from '../../define/define_string'
import { define } from '../../define/define'
let fspromises = require('fs').promises
import { gptDefine } from '../../define/gptDefine'
import { apiUrl } from '../../define/api/apiUrlDefine'
import { successMessage } from '../Public/generalTools'
import { RetryWithBackoff } from '../../define/Tools/common'
export class GPT {
constructor(global) {
this.global = global;
this.global = global
}
/**
* 输出测试案例
* @param {*} value 传入的值（整个数据）
*/
async GenerateGptExampleOut(value) {
try {
let data = JSON.parse(value);
let message = gptDefine.CustomizeGptPrompt(data);
let content = await this.FetchGpt(message);
console.log(content);
let data = JSON.parse(value)
let message = gptDefine.CustomizeGptPrompt(data)
let content = await RetryWithBackoff(
async () => {
return await this.FetchGpt(message)
},
5,
2000
)
console.log(content)
return {
code: 1,
data: content
}
} catch (error) {
return {
code: 0,
@ -47,79 +51,102 @@ export class GPT {
async GPTPromptGenerate(element, gpt_count, auto_analyze_character) {
try {
// 获取当前的推理模式
let gpt_auto_inference = this.global.config.gpt_auto_inference;
let message = null;
if (gpt_auto_inference == "customize") {
let gpt_auto_inference = this.global.config.gpt_auto_inference
let message = null
if (gpt_auto_inference == 'customize') {
// 自定义模式
// 获取当前自定义的推理提示词
let customize_gpt_prompt = (await gptDefine.getGptDataByTypeAndProperty("dynamic", "customize_gpt_prompt", [])).data;
let index = customize_gpt_prompt.findIndex(item => item.id == this.global.config.customize_gpt_prompt);
let customize_gpt_prompt = (
await gptDefine.getGptDataByTypeAndProperty('dynamic', 'customize_gpt_prompt', [])
).data
let index = customize_gpt_prompt.findIndex(
(item) => item.id == this.global.config.customize_gpt_prompt
)
if (this.global.config.customize_gpt_prompt && index < 0) {
throw new Error("自定义推理默认要选择对应的自定义推理词");
throw new Error('自定义推理默认要选择对应的自定义推理词')
}
message = gptDefine.CustomizeGptPrompt(customize_gpt_prompt[index], element.after_gpt);
message = gptDefine.CustomizeGptPrompt(customize_gpt_prompt[index], element.after_gpt)
message.push({
"role": "user",
"content": element.after_gpt
role: 'user',
content: element.after_gpt
})
} else {
// 内置模式
// 获取
let prefix_word = "";
let prefix_word = ''
// 拼接一个word
let i = element.no - 1;
let i = element.no - 1
if (i <= gpt_count) {
prefix_word = this.all_data.filter((item, index) => index < i).map(item => item.after_gpt).join('\r\n');
prefix_word = this.all_data
.filter((item, index) => index < i)
.map((item) => item.after_gpt)
.join('\r\n')
} else if (i > gpt_count) {
prefix_word = this.all_data.filter((item, index) => i - index <= gpt_count && i - index > 0).map(item => item.after_gpt).join('\r\n');
prefix_word = this.all_data
.filter((item, index) => i - index <= gpt_count && i - index > 0)
.map((item) => item.after_gpt)
.join('\r\n')
}
let suffix_word = "";
let o_i = this.all_data.length - i;
let suffix_word = ''
let o_i = this.all_data.length - i
if (o_i <= gpt_count) {
suffix_word = this.all_data.filter((item, index) => index > i).map(item => item.after_gpt).join('\r\n');
suffix_word = this.all_data
.filter((item, index) => index > i)
.map((item) => item.after_gpt)
.join('\r\n')
} else if (o_i > gpt_count) {
suffix_word = this.all_data.filter((item, index) => index - i <= gpt_count && index - i > 0).map(item => item.after_gpt).join('\r\n');
suffix_word = this.all_data
.filter((item, index) => index - i <= gpt_count && index - i > 0)
.map((item) => item.after_gpt)
.join('\r\n')
}
let word = `${prefix_word}\r\n${element.after_gpt}\r\n${suffix_word}`;
let single_word = element.after_gpt;
let word = `${prefix_word}\r\n${element.after_gpt}\r\n${suffix_word}`
let single_word = element.after_gpt
// 判断当前的格式
if (["superSinglePrompt", 'onlyPromptMJ'].includes(this.global.config.gpt_auto_inference)) {
if (['superSinglePrompt', 'onlyPromptMJ'].includes(this.global.config.gpt_auto_inference)) {
// 有返回案例的
message = gptDefine.GetExamplePromptMessage(this.global.config.gpt_auto_inference);
message = gptDefine.GetExamplePromptMessage(this.global.config.gpt_auto_inference)
// 加当前提问的
message.push({
"role": "user",
"content": single_word
role: 'user',
content: single_word
})
} else {
// 直接返回，没有案例的
message = [
{
"role": "system",
"content": gptDefine.getSystemContentByType(this.global.config.gpt_auto_inference, {
role: 'system',
content: gptDefine.getSystemContentByType(this.global.config.gpt_auto_inference, {
textContent: word,
characterContent: auto_analyze_character
})
},
{
"role": "user",
"content": gptDefine.getUserContentByType(this.global.config.gpt_auto_inference, {
role: 'user',
content: gptDefine.getUserContentByType(this.global.config.gpt_auto_inference, {
textContent: single_word,
wordCount: this.global.config.gpt_model && this.global.config.gpt_model.includes("gpt-4") ? '20' : '40'
wordCount:
this.global.config.gpt_model && this.global.config.gpt_model.includes('gpt-4')
? '20'
: '40'
})
}
]
}
}
let res = await this.FetchGpt(message);
return res;
let res = await RetryWithBackoff(
async () => {
return await this.FetchGpt(message)
},
5,
2000
)
return res
} catch (error) {
throw error;
throw error
}
}
@ -129,44 +156,60 @@ export class GPT {
async GPTPrompt(data) {
try {
console.log(data)
let value = JSON.parse(data[0]);
let show_global_message = data[1];
this.all_data = JSON.parse(data[2]);
let value = JSON.parse(data[0])
let show_global_message = data[1]
this.all_data = JSON.parse(data[2])
// 获取data中的after_gpt，然后使用换行符拼接成一个字符串
// let word = value.map(item => item.after_gpt).join('\r\n');
let batch = DEFINE_STRING.QUEUE_BATCH.SD_ORIGINAL_GPT_PROMPT;
let batch = DEFINE_STRING.QUEUE_BATCH.SD_ORIGINAL_GPT_PROMPT
// 获取人物角色数据
let config_json = JSON.parse(await fspromises.readFile(path.join(this.global.config.project_path, "scripts/config.json"), 'utf-8'));
let auto_analyze_character = config_json.auto_analyze_character;
let gpt_count = this.global.config.gpt_count ? this.global.config.gpt_count : 10;
let config_json = JSON.parse(
await fspromises.readFile(
path.join(this.global.config.project_path, 'scripts/config.json'),
'utf-8'
)
)
let auto_analyze_character = config_json.auto_analyze_character
let gpt_count = this.global.config.gpt_count ? this.global.config.gpt_count : 10
for (let i = 0; i < value.length; i++) {
const element = value[i];
this.global.requestQuene.enqueue(async () => {
const element = value[i]
this.global.requestQuene.enqueue(
async () => {
try {
let content = await this.GPTPromptGenerate(element, gpt_count, auto_analyze_character);
let content = await this.GPTPromptGenerate(element, gpt_count, auto_analyze_character)
if (content) {
content = content.replace(/\)\s*\(/g, ", ").replace(/^\(/, "").replace(/\)$/, "")
content = content
.replace(/\)\s*\(/g, ', ')
.replace(/^\(/, '')
.replace(/\)$/, '')
}
// 获取对应的数据，将数据返回前端事件
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.GPT_GENERATE_PROMPT_RETURN, {
this.global.newWindow[0].win.webContents.send(
DEFINE_STRING.GPT_GENERATE_PROMPT_RETURN,
{
id: element.id,
gpt_prompt: content
})
}
)
this.global.fileQueue.enqueue(async () => {
// 将推理出来的数据写入执行的文件中
let json_config = JSON.parse(await fspromises.readFile(element.prompt_json, 'utf-8'));
let json_config = JSON.parse(
await fspromises.readFile(element.prompt_json, 'utf-8')
)
// 写入
json_config.gpt_prompt = content;
await fspromises.writeFile(element.prompt_json, JSON.stringify(json_config));
json_config.gpt_prompt = content
await fspromises.writeFile(element.prompt_json, JSON.stringify(json_config))
})
} catch (error) {
throw error;
throw error
}
}, `${batch}_${element.id}`, batch);
},
`${batch}_${element.id}`,
batch
)
}
this.global.requestQuene.setBatchCompletionCallback(batch, (failedTasks) => {
@ -176,8 +219,8 @@ export class GPT {
但是以下任务执行失败：
`
failedTasks.forEach(({ taskId, error }) => {
message += `${taskId}-, \n 错误信息: ${error}` + '\n';
});
message += `${taskId}-, \n 错误信息: ${error}` + '\n'
})
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
code: 0,
@ -187,17 +230,15 @@ export class GPT {
if (show_global_message) {
this.global.newWindow[0].win.webContents.send(DEFINE_STRING.SHOW_MESSAGE_DIALOG, {
code: 1,
message: "所有推理任务完成"
message: '所有推理任务完成'
})
}
}
});
})
return {
code: 1,
code: 1
}
} catch (error) {
return {
code: 0,
@ -212,19 +253,19 @@ export class GPT {
* @returns
*/
ModifyData(gpt_url, data) {
let res = data;
if (gpt_url.includes("dashscope.aliyuncs.com")) {
let res = data
if (gpt_url.includes('dashscope.aliyuncs.com')) {
res = {
"model": data.model,
"input": {
"messages": data.messages,
model: data.model,
input: {
messages: data.messages
},
"parameters": {
"result_format": "message"
parameters: {
result_format: 'message'
}
}
}
return res;
return res
}
/**
@ -234,14 +275,13 @@ export class GPT {
* @returns
*/
GetResponseContent(gpt_url, res) {
let content = "";
if (gpt_url.includes("dashscope.aliyuncs.com")) {
content = res.data.output.choices[0].message.content;
let content = ''
if (gpt_url.includes('dashscope.aliyuncs.com')) {
content = res.data.output.choices[0].message.content
} else {
content = res.data.choices[0].message.content;
content = res.data.choices[0].message.content
}
return content;
return content
}
/**
@ -252,46 +292,47 @@ export class GPT {
* @param {*} gpt_model gpt的model，默认在global中取
* @returns
*/
async FetchGpt(message,
async FetchGpt(
message,
gpt_url = this.global.config.gpt_business,
gpt_key = this.global.config.gpt_key,
gpt_model = this.global.config.gpt_model) {
gpt_model = this.global.config.gpt_model
) {
try {
// 还有自定义的
let all_options = (await this.GetGPTBusinessOption("all", (value) => value.gpt_url)).data;
let all_options = (await this.GetGPTBusinessOption('all', (value) => value.gpt_url)).data
// 判断gpt_business 是不是一个http开头的
if (!gpt_url.includes("http")) {
if (!gpt_url.includes('http')) {
// 获取对应Id的gpt_url
let index = all_options.findIndex(item => item.value == gpt_url && item.gpt_url);
let index = all_options.findIndex((item) => item.value == gpt_url && item.gpt_url)
if (index < 0) {
throw new Error("获取GPT的服务商配置失败");
throw new Error('获取GPT的服务商配置失败')
}
gpt_url = all_options[index].gpt_url;
gpt_url = all_options[index].gpt_url
}
let data = {
"model": gpt_model,
"messages": message
};
model: gpt_model,
messages: message
}
data = this.ModifyData(gpt_url, data);
data = this.ModifyData(gpt_url, data)
let config = {
method: 'post',
maxBodyLength: Infinity,
url: gpt_url,
headers: {
'Authorization': `Bearer ${gpt_key}`,
Authorization: `Bearer ${gpt_key}`,
'Content-Type': 'application/json'
},
data: JSON.stringify(data)
};
}
let res = await axios.request(config);
let content = this.GetResponseContent(gpt_url, res);
return content;
let res = await axios.request(config)
let content = this.GetResponseContent(gpt_url, res)
return content
} catch (error) {
throw error;
throw error
}
}
@ -304,15 +345,21 @@ export class GPT {
try {
let message = [
{
"role": "system",
"content": gptDefine.getSystemContentByType("character", { textContent: value })
role: 'system',
content: gptDefine.getSystemContentByType('character', { textContent: value })
},
{
"role": "user",
"content": gptDefine.getUserContentByType("character", {})
role: 'user',
content: gptDefine.getUserContentByType('character', {})
}
]
let content = await this.FetchGpt(message);
let content = await RetryWithBackoff(
async () => {
return await this.FetchGpt(message)
},
5,
2000
)
return {
code: 1,
@ -326,15 +373,14 @@ export class GPT {
}
}
/**
* 获取GPT的服务商配置，默认的和自定义的
* @returns
*/
async GetGPTBusinessOption(value, callback = null) {
let res = await gptDefine.getGptDataByTypeAndProperty(value, "gpt_options", []);
let res = await gptDefine.getGptDataByTypeAndProperty(value, 'gpt_options', [])
if (res.code == 0) {
return res;
return res
} else {
if (callback) {
callback(res.data)
@ -348,7 +394,7 @@ export class GPT {
* @returns
*/
async GetGPTModelOption(value) {
return await gptDefine.getGptDataByTypeAndProperty(value, "gpt_model_options", []);
return await gptDefine.getGptDataByTypeAndProperty(value, 'gpt_model_options', [])
}
/**
@ -356,7 +402,7 @@ export class GPT {
* @returns
*/
async GetGptAutoInferenceOptions(value) {
return await gptDefine.getGptDataByTypeAndProperty(value, "gpt_auto_inference", []);
return await gptDefine.getGptDataByTypeAndProperty(value, 'gpt_auto_inference', [])
}
/**
@ -364,7 +410,7 @@ export class GPT {
* @returns
*/
async GetCustomizeGptPrompt(value) {
return await gptDefine.getGptDataByTypeAndProperty(value, "customize_gpt_prompt", []);
return await gptDefine.getGptDataByTypeAndProperty(value, 'customize_gpt_prompt', [])
}
/**
@ -374,9 +420,9 @@ export class GPT {
*/
async SaveDynamicGPTOption(value) {
try {
let res = await gptDefine.saveDynamicGPTOption(value);
let res = await gptDefine.saveDynamicGPTOption(value)
return {
code: 1,
code: 1
}
} catch (error) {
return {
@ -393,7 +439,7 @@ export class GPT {
*/
async DeleteDynamicGPTOption(value) {
try {
let res = await gptDefine.deleteDynamicGPTOption(value);
let res = await gptDefine.deleteDynamicGPTOption(value)
return {
code: 1,
data: res
@ -412,21 +458,26 @@ export class GPT {
*/
async TestGPTConnection(value) {
try {
value = JSON.parse(value);
value = JSON.parse(value)
let message = [
{
"role": "system",
"content": "你好"
role: 'system',
content: '你好'
},
{
"role": "user",
"content": "你好"
role: 'user',
content: '你好'
}
];
let content = await this.FetchGpt(message, value.gpt_business, value.gpt_key, value.gpt_model);
]
let content = await RetryWithBackoff(
async () => {
return await this.FetchGpt(message, value.gpt_business, value.gpt_key, value.gpt_model)
},
5,
2000
)
return {
code: 1,
code: 1
}
} catch (error) {
return {
@ -442,18 +493,24 @@ export class GPT {
*/
async AIModifyOneWord(value) {
try {
let message = [
{
"role": "system",
"content": "You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible."
role: 'system',
content:
'You are ChatGPT, a large language model trained by OpenAI. Answer as concisely as possible.'
},
{
"role": "user",
"content": `请您扮演一个抖音网文改写专家，我会给你一句文案，请你不要改变文案的结构，不改变原来的意思，仅对文案进行同义转换改写，不要有奇怪的写法，说法通俗一点，不要其他的标点符号，每一小句话之间都是以句号连接，参考抖音网文解说，以下是文案：${value[1]}。`
role: 'user',
content: `请您扮演一个抖音网文改写专家，我会给你一句文案，请你不要改变文案的结构，不改变原来的意思，仅对文案进行同义转换改写，不要有奇怪的写法，说法通俗一点，不要其他的标点符号，每一小句话之间都是以句号连接，参考抖音网文解说，以下是文案：${value[1]}。`
}
]
let content = await this.FetchGpt(message);
let content = await RetryWithBackoff(
async () => {
return await this.FetchGpt(message)
},
5,
2000
)
return {
code: 1,
@ -466,5 +523,4 @@ export class GPT {
}
}
}
}

86
src/main/Public/Translate.js
View File

@ -14,6 +14,7 @@ import { isEmpty } from 'lodash'
import { ValidateJson } from '../../define/Tools/validate'
import { GetOpenAISuccessResponse } from '../../define/response/openAIResponse'
import { successMessage } from './generalTools'
import { RetryWithBackoff } from '../../define/Tools/common'
let { Signer } = require('@volcengine/openapi')
@ -267,68 +268,82 @@ export class Translate {
messages: []
}
// 开始调用GPT进行翻译
if (from == 'en' && to == "zh") {
if (from == 'en' && to == 'zh') {
data.messages.push({
"role": "system",
"content": '我想让你充当英译中专家，用中文100%还原描述，不要加其他的联想，只翻译字面意思，请检查所有信息是否准确，并在回答时保持简活，不需要任何其他反馈。'
role: 'system',
content:
'我想让你充当英译中专家，用中文100%还原描述，不要加其他的联想，只翻译字面意思，请检查所有信息是否准确，并在回答时保持简活，不需要任何其他反馈。'
})
// 添加示例
data.messages.push({
"role": "user",
"content": 'A woman in her twenties with messy hair and a look of shock and despair. She is wearing a simple white shirt and jeans. Her face shows mixed emotions. The background is a dim and quiet room. The historical background is modern. The screen content is a close-up of the woman’s face with tear marks.'
role: 'user',
content:
'A woman in her twenties with messy hair and a look of shock and despair. She is wearing a simple white shirt and jeans. Her face shows mixed emotions. The background is a dim and quiet room. The historical background is modern. The screen content is a close-up of the woman’s face with tear marks.'
})
data.messages.push({
"role": "assistant",
"content": '一位二十多岁的女人，头发凌乱，表情震惊和绝望。她穿着一件简单的白色衬衫和牛仔裤。她的脸上显示出复杂的情感。背景是一个昏暗安静的房间。历史背景是现代的。屏幕内容是女人脸部的特写，带有泪痕。'
role: 'assistant',
content:
'一位二十多岁的女人，头发凌乱，表情震惊和绝望。她穿着一件简单的白色衬衫和牛仔裤。她的脸上显示出复杂的情感。背景是一个昏暗安静的房间。历史背景是现代的。屏幕内容是女人脸部的特写，带有泪痕。'
})
data.messages.push({
"role": "user",
"content": 'A twenty-something woman with short curly hair, smiling, wearing a casual white shirt and jeans, sitting on a comfortable sofa with a cup of coffee in her hand. She is in a small and cozy living room with a few books on the bookshelf, modern interior design, and natural light pouring into the room in the late afternoon. Screen content: Part of the sofa cushions.'
role: 'user',
content:
'A twenty-something woman with short curly hair, smiling, wearing a casual white shirt and jeans, sitting on a comfortable sofa with a cup of coffee in her hand. She is in a small and cozy living room with a few books on the bookshelf, modern interior design, and natural light pouring into the room in the late afternoon. Screen content: Part of the sofa cushions.'
})
data.messages.push({
"role": "assistant",
"content": '一位二十多岁的女性，留着短卷发，面带微笑，穿着休闲的白色衬衫和牛仔裤，手拿一杯咖啡，坐在一个舒适的沙发上。她所在的是一个小而温馨的客厅，书架上有几本书，现代室内设计，下午晚些时候，自然光线洒进房间。屏幕内容：沙发垫的一部分。'
role: 'assistant',
content:
'一位二十多岁的女性，留着短卷发，面带微笑，穿着休闲的白色衬衫和牛仔裤，手拿一杯咖啡，坐在一个舒适的沙发上。她所在的是一个小而温馨的客厅，书架上有几本书，现代室内设计，下午晚些时候，自然光线洒进房间。屏幕内容：沙发垫的一部分。'
})
data.messages.push({
"role": "user",
"content": 'In a modern city, a streamlined car is parked on the street. A man in his thirties, with short brown hair combed back, a calm, confident look, tall and thin in a clean white shirt and black pants, sits in the car. The interior of the car is clean and modern, and the background is blurred to highlight the man\'s calm demeanor. The man\'s cell phone is ringing. The scene is set in the present.'
role: 'user',
content:
"In a modern city, a streamlined car is parked on the street. A man in his thirties, with short brown hair combed back, a calm, confident look, tall and thin in a clean white shirt and black pants, sits in the car. The interior of the car is clean and modern, and the background is blurred to highlight the man's calm demeanor. The man's cell phone is ringing. The scene is set in the present."
})
data.messages.push({
"role": "assistant",
"content": '在现代城市，一辆流线型轿车停在街上。一个三十多岁的男人，短短的棕色头发梳向后，神情冷静，自信，穿着干净的白衬衫和黑裤子，身材高挑瘦长，坐在车里。车内干净而现代，背景模糊，以突出男人平静的神态。男人的手机正在响。场景设定在现在。'
role: 'assistant',
content:
'在现代城市，一辆流线型轿车停在街上。一个三十多岁的男人，短短的棕色头发梳向后，神情冷静，自信，穿着干净的白衬衫和黑裤子，身材高挑瘦长，坐在车里。车内干净而现代，背景模糊，以突出男人平静的神态。男人的手机正在响。场景设定在现在。'
})
} else if (from == 'zh' && to == "en") {
} else if (from == 'zh' && to == 'en') {
data.messages.push({
"role": "system",
"content": '我想让你充当中译英专家，用中文100%还原描述，不要加其他的联想，只翻译字面意思，请检查所有信息是否准确，并在回答时保持简活，不需要任何其他反馈。'
role: 'system',
content:
'我想让你充当中译英专家，用中文100%还原描述，不要加其他的联想，只翻译字面意思，请检查所有信息是否准确，并在回答时保持简活，不需要任何其他反馈。'
})
// 添加示例
data.messages.push({
"role": "user",
"content": '一位二十多岁的女人，头发凌乱，表情震惊和绝望。她穿着一件简单的白色衬衫和牛仔裤。她的脸上显示出复杂的情感。背景是一个昏暗安静的房间。历史背景是现代的。屏幕内容是女人脸部的特写，带有泪痕。'
role: 'user',
content:
'一位二十多岁的女人，头发凌乱，表情震惊和绝望。她穿着一件简单的白色衬衫和牛仔裤。她的脸上显示出复杂的情感。背景是一个昏暗安静的房间。历史背景是现代的。屏幕内容是女人脸部的特写，带有泪痕。'
})
data.messages.push({
"role": "assistant",
"content": 'A woman in her twenties with messy hair and a look of shock and despair. She is wearing a simple white shirt and jeans. Her face shows mixed emotions. The background is a dim and quiet room. The historical background is modern. The screen content is a close-up of the woman’s face with tear marks.'
role: 'assistant',
content:
'A woman in her twenties with messy hair and a look of shock and despair. She is wearing a simple white shirt and jeans. Her face shows mixed emotions. The background is a dim and quiet room. The historical background is modern. The screen content is a close-up of the woman’s face with tear marks.'
})
data.messages.push({
"role": "user",
"content": '一位二十多岁的女性，留着短卷发，面带微笑，穿着休闲的白色衬衫和牛仔裤，手拿一杯咖啡，坐在一个舒适的沙发上。她所在的是一个小而温馨的客厅，书架上有几本书，现代室内设计，下午晚些时候，自然光线洒进房间。屏幕内容：沙发垫的一部分。'
role: 'user',
content:
'一位二十多岁的女性，留着短卷发，面带微笑，穿着休闲的白色衬衫和牛仔裤，手拿一杯咖啡，坐在一个舒适的沙发上。她所在的是一个小而温馨的客厅，书架上有几本书，现代室内设计，下午晚些时候，自然光线洒进房间。屏幕内容：沙发垫的一部分。'
})
data.messages.push({
"role": "assistant",
"content": 'A twenty-something woman with short curly hair, smiling, wearing a casual white shirt and jeans, sitting on a comfortable sofa with a cup of coffee in her hand. She is in a small and cozy living room with a few books on the bookshelf, modern interior design, and natural light pouring into the room in the late afternoon. Screen content: Part of the sofa cushions.'
role: 'assistant',
content:
'A twenty-something woman with short curly hair, smiling, wearing a casual white shirt and jeans, sitting on a comfortable sofa with a cup of coffee in her hand. She is in a small and cozy living room with a few books on the bookshelf, modern interior design, and natural light pouring into the room in the late afternoon. Screen content: Part of the sofa cushions.'
})
data.messages.push({
"role": "user",
"content": '在现代城市，一辆流线型轿车停在街上。一个三十多岁的男人，短短的棕色头发梳向后，神情冷静，自信，穿着干净的白衬衫和黑裤子，身材高挑瘦长，坐在车里。车内干净而现代，背景模糊，以突出男人平静的神态。男人的手机正在响。场景设定在现在。'
role: 'user',
content:
'在现代城市，一辆流线型轿车停在街上。一个三十多岁的男人，短短的棕色头发梳向后，神情冷静，自信，穿着干净的白衬衫和黑裤子，身材高挑瘦长，坐在车里。车内干净而现代，背景模糊，以突出男人平静的神态。男人的手机正在响。场景设定在现在。'
})
data.messages.push({
"role": "assistant",
"content": 'In a modern city, a streamlined car is parked on the street. A man in his thirties, with short brown hair combed back, a calm, confident look, tall and thin in a clean white shirt and black pants, sits in the car. The interior of the car is clean and modern, and the background is blurred to highlight the man\'s calm demeanor. The man\'s cell phone is ringing. The scene is set in the present.'
role: 'assistant',
content:
"In a modern city, a streamlined car is parked on the street. A man in his thirties, with short brown hair combed back, a calm, confident look, tall and thin in a clean white shirt and black pants, sits in the car. The interior of the car is clean and modern, and the background is blurred to highlight the man's calm demeanor. The man's cell phone is ringing. The scene is set in the present."
})
} else {
throw new Error("GPT翻译只支持中英互译")
throw new Error('GPT翻译只支持中英互译')
}
data.messages.push({
@ -346,7 +361,13 @@ export class Translate {
},
data: JSON.stringify(data)
}
let res = await axios.request(config)
let res = await RetryWithBackoff(
async () => {
return await axios.request(config)
},
5,
2000
)
// 将返回的数据进行拼接数据处理
let res_data = []
@ -416,7 +437,6 @@ export class Translate {
if (arr_data.length <= 0) {
return
}
let req_data = {}
for (let j = 0; j < arr_data.length; j++) {
const element = arr_data[j]

2
src/main/Service/Book/bookTask.ts
View File

@ -243,7 +243,7 @@ export class BookTask {
startTime: element.startTime,
endTime: element.endTime,
timeLimit: element.timeLimit,
subValue: element.subValue,
subValue: element.subValue && element.subValue.length > 0 ? JSON.stringify(element.subValue) : undefined,
characterTags: element.characterTags && element.characterTags.length > 0 ? cloneDeep(element.characterTags) : [],
gptPrompt: element.gptPrompt,
outImagePath: path.relative(define.project_path, outImagePath),

6
src/main/Service/GPT/gpt.ts
View File

@ -1,6 +1,7 @@
import { isEmpty } from "lodash";
import { gptDefine } from "../../../define/gptDefine";
import axios from "axios";
import { RetryWithBackoff } from "../../../define/Tools/common";
/**
* 一些GPT相关的服务都在这边
@ -168,7 +169,10 @@ export class GptService {
let baseSubUrl = baseUrl ? (baseUrl.endsWith('/') ? baseUrl + 'v1/chat/completions' : baseUrl + '/v1/chat/completions') : null;
let url = baseSubUrl ? baseSubUrl : "https://api.laitool.cc/v1/chat/completions"
// 开始请求，这个默认是使用的是LAI API的gpt-4o-mini
let content = await this.FetchGpt(message, 'gpt-4o-mini', apiKey, url)
let content = await RetryWithBackoff<string>(async () => {
return await this.FetchGpt(message, 'gpt-4o-mini', apiKey, url);
}, 5, 2000)
return content
} catch (error) {
throw error

12
src/main/Service/Subtitle/subtitle.ts
View File

@ -22,6 +22,7 @@ import { BookTaskStatus, OperateBookType } from '../../../define/enum/bookEnum'
import axios from 'axios'
import { GptService } from '../GPT/gpt'
import FormData from 'form-data'
import { RetryWithBackoff } from '../../../define/Tools/common'
const util = require('util')
const { exec } = require('child_process')
const execAsync = util.promisify(exec)
@ -596,17 +597,22 @@ export class Subtitle {
headers: {
'Accept': 'application/json',
'Authorization': subtitleSetting.laiWhisper.apiKey,
'User-Agent': 'Apifox/1.0.0 (https://apifox.com)',
'Content-Type': 'multipart/form-data',
...formdata.getHeaders() // 在Node.js环境中需要添加这一行
},
data: formdata
};
let res = await axios(config)
// laiwhisper 要做重试机制
let res = await RetryWithBackoff(async () => {
return await axios(config)
}, 5, 2000)
let text = res.data.text;
// 但是这边是繁体，需要转化为简体
let simpleText = await this.gptService.ChineseTraditionalToSimplified(text, subtitleSetting.laiWhisper.apiKey, url);
// 请求也要做重试
let simpleText = await RetryWithBackoff(async () => {
return await this.gptService.ChineseTraditionalToSimplified(text, subtitleSetting.laiWhisper.apiKey, url);
}, 5, 2000);
console.log(res.data)
return simpleText;

18
src/main/Service/Subtitle/subtitleService.ts
View File

@ -11,9 +11,7 @@ import fs from 'fs'
import { CheckFileOrDirExist } from "../../../define/Tools/file";
import { BookServiceBasic } from "../ServiceBasic/bookServiceBasic";
import { Subtitle } from "./subtitle";
import { LoggerStatus, ResponseMessageType } from "../../../define/enum/softwareEnum";
import { TaskScheduler } from "../taskScheduler";
import { OperationType } from "realm/dist/public-types/internal";
import { OperateBookType } from "../../../define/enum/bookEnum";
import { Book } from "../../../model/book";
@ -126,9 +124,14 @@ export class SubtitleService {
//#region 语音转文案或者是字幕识别
/**
* 反推提取文案
* 反推提取文案的入口方法
* @param bookId 小说ID
* @param bookTaskId 小说批次任务ID
* @param operateBookType 操作的小说类型
* @param coverData 是不是要覆盖旧的数据
* @returns
*/
async GetCopywriting(bookId: string, bookTaskId: string, operateBookType: OperateBookType): Promise<GeneralResponse.ErrorItem | GeneralResponse.SuccessItem> {
async GetCopywriting(bookId: string, bookTaskId: string, operateBookType: OperateBookType, coverData: boolean): Promise<GeneralResponse.ErrorItem | GeneralResponse.SuccessItem> {
try {
let subtitleSettingRes = await this.GetSubtitleSetting();
if (subtitleSettingRes.code == 0) {
@ -152,8 +155,13 @@ export class SubtitleService {
} else {
throw new Error("未知的操作类型")
}
if (!coverData) { // 不覆盖数据，将已经有的数据过滤掉
bookTaskDetails = bookTaskDetails.filter(item => isEmpty(item.afterGpt) && isEmpty(item.word))
}
if (bookTaskDetails.length <= 0) {
throw new Error("分镜信息不存在");
throw new Error("分镜信息不存在 / 已经有文案，无需提取");
}
let { book, bookTask } = await this.bookServiceBasic.GetBookAndTask(bookId, tempBookTaskId)

6
src/main/Service/Translate/Translate.ts
View File

@ -16,6 +16,7 @@ import { SoftwareService } from "../../../define/db/service/SoftWare/softwareSer
import { isEmpty } from "lodash";
import { ValidateJson } from "../../../define/Tools/validate";
import { GetOpenAISuccessResponse } from "../../../define/response/openAIResponse";
import { RetryWithBackoff } from "../../../define/Tools/common";
let {
Signer
@ -202,7 +203,10 @@ export class Translate {
},
data: JSON.stringify(data)
};
let res = await axios.request(config);
let res = await RetryWithBackoff(async () => {
return await axios.request(config);
}, 5, 2000)
// let res = await axios.request(config);
// 将返回的数据进行拼接数据处理
let res_data = [];

5
src/preload/book.js
View File

@ -72,12 +72,13 @@ const book = {
//#region 文案相关信息
// 获取文案信息
GetCopywriting: async (bookId, bookTaskId, operateBookType) =>
GetCopywriting: async (bookId, bookTaskId, operateBookType, coverData) =>
await ipcRenderer.invoke(
DEFINE_STRING.BOOK.GET_COPYWRITING,
bookId,
bookTaskId,
operateBookType
operateBookType,
coverData
),
// 将文案信息导出，方便修改

4
src/renderer/src/components/Book/Components/AddBook.vue → src/renderer/src/components/Book/Components/ManageBook/AddBook.vue
View File

@ -115,10 +115,10 @@
<script>
import { ref, onMounted, defineComponent, onUnmounted, toRaw, watch } from 'vue'
import { useMessage, NButton, NForm, NFormItem, NInput, NSelect, NIcon } from 'naive-ui'
import { useReverseManageStore } from '../../../../../stores/reverseManage.ts'
import { useReverseManageStore } from '../../../../../../stores/reverseManage.ts'
import { CloseSharp } from '@vicons/ionicons5'
import { cloneDeep } from 'lodash'
import { useSoftwareStore } from '../../../../../stores/software'
import { useSoftwareStore } from '../../../../../../stores/software.ts'
export default defineComponent({
components: {

6
src/renderer/src/components/Book/Components/BookListAction.vue → src/renderer/src/components/Book/Components/ManageBook/BookListAction.vue
View File

@ -30,9 +30,9 @@
<script setup>
import { ref, h, toRaw } from 'vue'
import { useMessage, useDialog, NButton } from 'naive-ui'
import { useReverseManageStore } from '../../../../../stores/reverseManage.ts'
import { useSoftwareStore } from '../../../../../stores/software'
import AddBook from '../Components/AddBook.vue'
import { useReverseManageStore } from '../../../../../../stores/reverseManage.ts'
import { useSoftwareStore } from '../../../../../../stores/software.ts'
import AddBook from './AddBook.vue'
let props = defineProps({
book: undefined
})

6
src/renderer/src/components/Book/Components/BookTaskListAction.vue → src/renderer/src/components/Book/Components/ManageBook/BookTaskListAction.vue
View File

@ -36,9 +36,9 @@
<script>
import { ref, onMounted, defineComponent, h } from 'vue'
import { useMessage, NButton, useDialog } from 'naive-ui'
import { useReverseManageStore } from '../../../../../stores/reverseManage.ts'
import { useSoftwareStore } from '../../../../../stores/software'
import { BookType, OperateBookType } from '../../../../../define/enum/bookEnum'
import { useReverseManageStore } from '../../../../../../stores/reverseManage.ts'
import { useSoftwareStore } from '../../../../../../stores/software.ts'
import { BookType, OperateBookType } from '../../../../../../define/enum/bookEnum.ts'
import ManageBookTaskGenerateInformation from './ManageBookTaskGenerateInformation.vue'
export default defineComponent({

8
src/renderer/src/components/Book/Components/ManageBookButton.vue → src/renderer/src/components/Book/Components/ManageBook/ManageBookButton.vue
View File

@ -41,10 +41,10 @@
import { ref, onMounted, defineComponent, onUnmounted, h, toRaw, computed } from 'vue'
import { useMessage, NButton, NIcon, useDialog, NPopover, NSwitch, NDropdown } from 'naive-ui'
import { AddSharp, ReloadSharp, ResizeSharp, SwapHorizontalSharp } from '@vicons/ionicons5'
import AddBook from '../Components/AddBook.vue'
import { useSoftwareStore } from '../../../../../stores/software'
import { useReverseManageStore } from '../../../../../stores/reverseManage.ts'
import { BookType } from '../../../../../define/enum/bookEnum'
import AddBook from './AddBook.vue'
import { useSoftwareStore } from '../../../../../../stores/software'
import { useReverseManageStore } from '../../../../../../stores/reverseManage'
import { BookType } from '../../../../../../define/enum/bookEnum'
export default defineComponent({
components: {

6
src/renderer/src/components/Book/Components/ManageBookShowLogger.vue → src/renderer/src/components/Book/Components/ManageBook/ManageBookShowLogger.vue
View File

@ -14,9 +14,9 @@
<script>
import { ref, onMounted, defineComponent, onUnmounted, toRaw, nextTick, watchEffect } from 'vue'
import { useMessage, NLog } from 'naive-ui'
import Artplayer from '../../Components/Artplayer.vue'
import { useReverseManageStore } from '../../../../../stores/reverseManage.ts'
import { DEFINE_STRING } from '../../../../../define/define_string'
import Artplayer from '../../../Components/Artplayer.vue'
import { useReverseManageStore } from '../../../../../../stores/reverseManage.ts'
import { DEFINE_STRING } from '../../../../../../define/define_string.ts'
export default defineComponent({
components: { Artplayer, NLog },

2
src/renderer/src/components/Book/Components/ManageBookTaskGenerateInformation.vue → src/renderer/src/components/Book/Components/ManageBook/ManageBookTaskGenerateInformation.vue
View File

@ -66,7 +66,7 @@
import { ref, onMounted, defineComponent, onUnmounted, toRaw, watch } from 'vue'
import { useMessage, NForm, NFormItem, NButton, NIcon, NInput, NSelect } from 'naive-ui'
import { FolderOpen } from '@vicons/ionicons5'
import { OperateBookType } from '../../../../../define/enum/bookEnum'
import { OperateBookType } from '../../../../../../define/enum/bookEnum'
export default defineComponent({
components: { NForm, NFormItem, NButton, NIcon, NInput, NSelect, FolderOpen },

35
src/renderer/src/components/Book/Components/ManageBookDetailButton.vue
View File

@ -15,9 +15,9 @@
<n-button
:color="softwareStore.SoftColor.ORANGE"
style="margin-left: 5px"
@click="GetCopywriting"
@click="GetCopywriting('all')"
>
开始提取文案
开始提取全部文案
</n-button>
</n-dropdown>
@ -244,7 +244,7 @@ export default defineComponent({
}
// 获取文案信息
async function GetCopywriting() {
async function GetCopywriting(type) {
if (isEmpty(reverseManageStore.selectBookTask.id)) {
window.api.showGlobalMessageDialog({
code: 0,
@ -261,7 +261,9 @@ export default defineComponent({
let da = dialog.warning({
title: '开始提取文案提示',
content: `即将进行文案提取，当前的文案提取模式为 ${subtitleSettingRes.data.selectModel} ，是否继续？`,
content: `即将进行 ${type == 'blank' ? '空白' : '全部'} 文案提取${
type == 'blank' ? '' : '，会重新提取并覆盖旧的文案数据'
}，当前的文案提取模式为 ${subtitleSettingRes.data.selectModel} ，是否继续？`,
positiveText: '继续',
negativeText: '取消',
onPositiveClick: async () => {
@ -271,7 +273,8 @@ export default defineComponent({
let copywriting_res = await window.book.GetCopywriting(
reverseManageStore.selectBook.id,
reverseManageStore.selectBookTask.id,
OperateBookType.BOOKTASK
OperateBookType.BOOKTASK,
type == 'blank' ? false : true
)
softwareStore.spin.spinning = false
if (copywriting_res.code == 0) {
@ -382,6 +385,9 @@ export default defineComponent({
case 'framing': // 开始分镜
await Framing()
break
case 'recognizing_blank': // 提取空白文案
await GetCopywriting('blank')
break
case 'recognizing_setting': // 文案位置设置
await GetCopywritingSetting()
break
@ -561,9 +567,20 @@ export default defineComponent({
* @param to 目标语言
*/
async function TranslateService(from, to) {
debugger
let translateData = []
for (let i = 0; i < reverseManageStore.selectBookTaskDetail.length; i++) {
const element = reverseManageStore.selectBookTaskDetail[i]
if (!isEmpty(element.gptPrompt)) {
translateData.push({
text: element.gptPrompt,
from: from,
to: to,
type: TranslateType.GPT_PROMPT_TRANSLATE,
isSplit: false, // 该参数设置为true时，会将文本按照标点符号分割成多个句子进行翻译
bookTaskDetailId: element.id
})
} else {
let reversePrompt = element.reversePrompt
if (reversePrompt) {
reversePrompt.forEach((item) => {
@ -573,9 +590,8 @@ export default defineComponent({
return
}
}
let temp_obj = {
text: item.prompt,
text: item.promptCN ? item.promptCN : item.prompt,
from: from,
to: to,
type: TranslateType.REVERSE_PROMPT_TRANSLATE,
@ -587,6 +603,7 @@ export default defineComponent({
})
}
}
}
if (translateData.length == 0) {
message.error('没有需要翻译的数据')
return
@ -704,6 +721,10 @@ export default defineComponent({
{ label: '分镜', key: 'framing' }
],
copywritingOptions: [
{
label: '提取空白文案',
key: 'recognizing_blank'
},
{
label: '提取文案位置',
key: 'recognizing_setting'

3
src/renderer/src/components/Book/Components/ManageBookOldImage.vue
View File

@ -71,7 +71,8 @@ async function GetVideoSubtitle() {
let res = await window.book.GetCopywriting(
reverseManageStore.selectBook.id,
data.value.id,
OperateBookType.BOOKTASKDETAIL
OperateBookType.BOOKTASKDETAIL,
true
)
softwareStore.spin.spinning = false
if (res.code == 0) {

4
src/renderer/src/components/Book/ManageBook.vue
View File

@ -21,11 +21,11 @@
<script>
import { ref, onMounted, defineComponent, onUnmounted, toRaw, reactive, computed, h } from 'vue'
import { useMessage, NButton, NDataTable, NTag, NEllipsis } from 'naive-ui'
import ManageBookButton from './Components/ManageBookButton.vue'
import ManageBookButton from './Components/ManageBook/ManageBookButton.vue'
import { useSoftwareStore } from '../../../../stores/software'
import { useReverseManageStore } from '../../../../stores/reverseManage.ts'
import { BookType } from '../../../../define/enum/bookEnum'
import BookListAction from './Components/BookListAction.vue'
import BookListAction from './Components/ManageBook/BookListAction.vue'
export default defineComponent({
components: {

2
src/renderer/src/components/Book/ManageBookDetail.vue
View File

@ -18,7 +18,7 @@ import { ref, onMounted, defineComponent, onUnmounted, toRaw, watch } from 'vue'
import { useMessage, NDivider } from 'naive-ui'
import ManageBookDetailButton from './Components/ManageBookDetailButton.vue'
import ManageBookReverseTable from './MJReverse/ManageBookReverseTable.vue'
import ManageBookShowLogger from './Components/ManageBookShowLogger.vue'
import ManageBookShowLogger from './Components/ManageBook/ManageBookShowLogger.vue'
import { useSoftwareStore } from '../../../../stores/software'
import { DEFINE_STRING } from '../../../../define/define_string'
import { DialogType } from '../../../../define/enum/bookEnum'

2
src/renderer/src/components/Book/ManageBookTask.vue
View File

@ -36,7 +36,7 @@ import { useMessage, useDialog, NButton, NDataTable, NIcon, NWatermark } from 'n
import { useReverseManageStore } from '../../../../stores/reverseManage'
import { useSoftwareStore } from '../../../../stores/software'
import { AddSharp } from '@vicons/ionicons5'
import BookTaskListAction from './Components/BookTaskListAction.vue'
import BookTaskListAction from './Components/ManageBook/BookTaskListAction.vue'
import { useRouter } from 'vue-router'
import { OperateBookType } from '../../../../define/enum/bookEnum'
import { DEFINE_STRING } from '../../../../define/define_string'

89
src/renderer/src/components/Original/Components/DataTableShowGenerateImage.vue
View File

@ -88,7 +88,7 @@
<n-button
text
:type="data.imageLock ? 'error' : 'primary'"
style="font-size: 24px; position: absolute; left: -12px; top: -0"
style="font-size: 24px; position: absolute; left: -12px; top: -5px"
@click="ModifyLock"
>
<n-icon>
@ -96,6 +96,16 @@
<LockOpen v-else />
</n-icon>
</n-button>
<n-button
text
type="info"
style="font-size: 24px; position: absolute; left: 8px; top: -5px"
@click="DownloadImage"
>
<n-icon>
<Download />
</n-icon>
</n-button>
</div>
<div>
<n-divider vertical style="height: 100%" />
@ -128,6 +138,7 @@ import { ref, h, onMounted, defineComponent, onUnmounted, toRaw, watch } from 'v
import {
NImage,
useMessage,
useDialog,
NDivider,
NImageGroup,
NButton,
@ -138,8 +149,9 @@ import {
} from 'naive-ui'
import { usePromptStore } from '../../../../../stores/prompt'
import { useImageStore } from '../../../../../stores/image'
import { LockClosed, LockOpen } from '@vicons/ionicons5'
import { LockClosed, LockOpen, Download } from '@vicons/ionicons5'
import { useReverseManageStore } from '../../../../../stores/reverseManage'
import { isEmpty } from 'lodash'
export default defineComponent({
components: {
@ -152,13 +164,15 @@ export default defineComponent({
NTag,
NIcon,
LockClosed,
LockOpen
LockOpen,
Download
},
props: ['initData', 'type', 'image_generate_category', 'func', 'index'],
setup(props) {
let imageStore = useImageStore()
let promptStore = usePromptStore()
let message = useMessage()
let dialog = useDialog()
let data = ref(props.initData)
let type = ref(props.type)
let images = ref(props.initData.subImagePath)
@ -313,7 +327,16 @@ export default defineComponent({
* MJ采集图片
*/
async function MJGetImage() {
props.func.mJGetImage()
dialog.warning({
title: 'MJ采集图片警告',
content:
'只有API出图才能采集，其他模式不支持，且API出图图片链接一般只有二十四个小时有效，过期失败，是否继续？',
positiveText: '继续',
negativeText: '取消',
onPositiveClick: async () => {
await props.func.mJGetImage()
}
})
}
/**
@ -351,12 +374,70 @@ export default defineComponent({
}
}
/**
* 下载对应的分镜图片
*/
async function DownloadImage() {
// 目前只能判断是不是有mj_message
dialog.warning({
title: '下载图片警告',
content: '单个图片的下载，只支持 MJ API 生图，当前有图片的话会被覆盖掉，是否继续？',
positiveText: '继续',
negativeText: '取消',
onPositiveClick: async () => {
if (data.value.mj_message) {
debugger
// 加判断
// 只要状态不是success 和 error ，其他的都重新获取
if (data.value.mj_message.status == 'error') {
message.error('失败状态不能采集图片')
return
}
if (isEmpty(data.value.mj_message.message_id)) {
message.error('没有消息ID，不能采集图片')
return
}
window.mj.GetGeneratedMJImageAndSplit(JSON.stringify([toRaw(data.value)]), (value) => {
if (value.code == 0) {
message.error(value.message)
return
}
// 修改数据
for (let i = 0; i < value.data.length; i++) {
const element = value.data[i]
// 修改对应的数据
data.value.outImagePath =
'file://' +
element.outImagePath.replaceAll('\\', '/') +
'?time=' +
new Date().getTime()
data.value.subImagePath = element.subImagePath.map(
(item) => 'file://' + item.replaceAll('\\', '/') + '?time=' + new Date().getTime()
)
data.value.mj_message.image_click = element.result
data.value.mj_message.image_path = element.image_path
data.value.mj_message.progress = 100
data.value.mj_message.status = 'success'
}
})
} else if (data.value.mjMessage) {
message.error('暂时不支持')
return
} else {
message.error('没有生图数据')
}
}
})
}
return {
data,
type,
image_options,
sd_batch_size,
ModifyLock,
DownloadImage,
width,
images,
space_image,

16
src/renderer/src/components/Original/DataTable.vue
View File

@ -933,12 +933,26 @@ export default defineComponent({
if (!item.mj_message) {
continue
}
if (item.mj_message.status && item.mj_message.status == 'error') {
// 只要状态不是success 和 error ，其他的都重新获取
if (item.mj_message.status == 'error' || item.mj_message.status == 'success') {
continue
}
// 判断message_id是不是有值，有值重新抓
if (item.mj_message.message_id && !isEmpty(item.mj_message.message_id)) {
pa.push(item)
}
// // 判断是不是又图片，没有的话，重新抓
// if (
// !(isEmpty(item.outImagePath) || (!item.subImagePath && item.subImagePath.length <= 0))
// ) {
// continue
// }
// if (item.mj_message.message_id && !isEmpty(item.mj_message.message_id)) {
// pa.push(item)
// }
}
if (pa.length > 0) {