LaiTool V3.1.2

package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "laitool",
-  "version": "3.1.1",
+  "version": "3.1.2",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "laitool",
-      "version": "3.1.1",
+      "version": "3.1.2",
       "hasInstallScript": true,
       "dependencies": {
         "@alicloud/alimt20181012": "^1.2.0",

package.json

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

src/define/db/service/Book/bookTaskDetailService.ts

@@ -67,7 +67,7 @@ export class BookTaskDetailService extends BaseRealmService {
           videoPath: JoinPath(define.project_path, item.videoPath),
           audioPath: JoinPath(define.project_path, item.audioPath),
           oldImage: JoinPath(define.project_path, item.oldImage),
-          outImagePath: JoinPath(define.project_path, item.outImagePath),
+          outImagePath: JoinPath(define.project_path, item.outImagePath) , 
           subImagePath: (item.subImagePath as string[])?.map((subImage) => {
             return JoinPath(define.project_path, subImage)
           }),

src/define/define_string.ts

@@ -253,9 +253,20 @@ export const DEFINE_STRING = {
     REPLACE_BOOK_DATA: "REPLACE_BOOK_DATA",
     SAVE_COPYWRITING: 'SAVE_COPYWRITING',
 
-    //#region  原创推理提示词
+    //#region  提示词
+    /**
+     * 原创推理提示词
+     */
     ORIGINAL_GPT_PROMPT: "ORIGINAL_GPT_PROMPT",
+    /**
+     * 原创推理提示词返回
+     */
     ORIGINAL_GPT_PROMPT_RETURN: "ORIGINAL_GPT_PROMPT_RETURN",
+
+    /**
+     * 导入提示词，通用
+     */
+    IMPORT_GPT_PROMPT: "IMPORT_GPT_PROMPT",
     //#endregion
 
     //#region 生图返回相关
@@ -287,6 +298,15 @@ export const DEFINE_STRING = {
 
     //#endregion
 
+    //#region  图片相关
+
+    /**
+     * 获取MJ的消息ID，下载图片并分割
+     */
+    GET_IMAGE_URL_AND_DOWNLOAD : "GET_IMAGE_URL_AND_DOWNLOAD",
+
+    //#endregion
+
     COMPUTE_STORYBOARD: 'COMPUTE_STORYBOARD',
 
     GET_FRAME: 'GET_FRAME',

src/define/enum/mjEnum.ts

@@ -18,7 +18,11 @@ export enum MJImageType {
   FLUX_API = 'flux-api',
 
   // flxu-forge
-  FLUX_FORGE = 'flux-forge'
+  FLUX_FORGE = 'flux-forge',
+
+  // 导入 
+  IMPORT = 'import',
+
 }
 
 export enum MJRobotType {

src/main/IPCEvent/bookIpc.ts

@@ -149,6 +149,11 @@ export function BookIpc() {
       await bookPrompt.OriginalGetPrompt(id, operateBookType, coverData)
   )
 
+  ipcMain.handle(
+    DEFINE_STRING.BOOK.IMPORT_GPT_PROMPT,
+    async (event, bookTaskId: string, txtPath: string) => await bookPrompt.ImportGPTPrompt(bookTaskId, txtPath)
+  )
+
   //#endregion
 
   //#region 文案相关
@@ -252,6 +257,12 @@ export function BookIpc() {
     async (event, id, operateBookType) => await bookImage.ResetGenerateImage(id, operateBookType)
   )
 
+  ipcMain.handle(
+    DEFINE_STRING.BOOK.GET_IMAGE_URL_AND_DOWNLOAD,
+    async (event, id: string, operateBookType: OperateBookType, coverData: boolean) =>
+      await bookImage.GetImageUrlAndDownload(id, operateBookType, coverData)
+  )
+
   //#endregion
 
 

src/main/Service/Book/bookGeneral.ts

@@ -28,18 +28,37 @@ export class BookGeneral {
     this.bookServiceBasic.transaction((realm) => {
       for (let i = 0; i < bookTaskDetail.length; i++) {
         let element = bookTaskDetail[i];
+        // 这边新增判断，是不是有subValue，有点话替换subValue中的数据
         let afterGpt = element.afterGpt
-        // 判断是否存在before的数据
+        let subValue = element.subValue
-        if (!afterGpt.includes(replaceData.before)) {
+        if (subValue && subValue.length > 0) {
-          continue
+          subValue = subValue.map(item => {
+            return {
+              ...item,
+              srt_value: item.srt_value.replaceAll(replaceData.before, replaceData.after)
+            }
+          })
+          let btd = realm.objectForPrimaryKey('BookTaskDetail', element.id)
+          btd.subValue = JSON.stringify(subValue);
+          result.push({
+            bookTaskDetailId: element.id,
+            newData: subValue,
+            type: 'subValue'
+          } as Book.ReplaceDataRes)
+          // 保存数据
+        } else {
+          // 判断是否存在before的数据
+          if (!afterGpt.includes(replaceData.before)) {
+            continue
+          }
+          let newAfterGpt = afterGpt.replaceAll(replaceData.before, replaceData.after);
+          let btd = realm.objectForPrimaryKey('BookTaskDetail', element.id)
+          btd.afterGpt = newAfterGpt
+          result.push({
+            bookTaskDetailId: element.id,
+            newData: newAfterGpt
+          } as Book.ReplaceDataRes)
         }
-        let newAfterGpt = afterGpt.replaceAll(replaceData.before, replaceData.after);
-        let btd = realm.objectForPrimaryKey('BookTaskDetail', element.id)
-        btd.afterGpt = newAfterGpt
-        result.push({
-          bookTaskDetailId: element.id,
-          newData: newAfterGpt
-        } as Book.ReplaceDataRes)
       }
     })
     return result

src/main/Service/Book/bookImage.ts

@@ -1,4 +1,4 @@
-import { BookImageCategory, BookType, OperateBookType } from "../../../define/enum/bookEnum";
+import { BookImageCategory, BookType, MJAction, OperateBookType } from "../../../define/enum/bookEnum";
 import { GeneralResponse } from "../../../model/generalResponse";
 import { errorMessage, successMessage } from "../../Public/generalTools";
 import { Book } from "../../../model/book";
@@ -15,6 +15,8 @@ import { BookServiceBasic } from "../ServiceBasic/bookServiceBasic";
 import fs from 'fs'
 import util from 'util'
 import { exec } from 'child_process'
+import { MJImageType } from "../../../define/enum/mjEnum";
+import MJApi from '../MJ/mjApi'
 const execAsync = util.promisify(exec);
 
 /**
@@ -24,9 +26,11 @@ export class BookImage {
   bookServiceBasic: BookServiceBasic
   tools: Tools;
   mjOpt: MJOpt;
+  mjApi: MJApi
   constructor() {
     this.tools = new Tools()
     this.mjOpt = new MJOpt()
+    this.mjApi = new MJApi()
     this.bookServiceBasic = new BookServiceBasic()
   }
 
@@ -45,7 +49,7 @@ export class BookImage {
    * @param id 
    * @param operateBookType 
    */
-  async ResetGenerateImage(id: string, operateBookType: OperateBookType, coverData: boolean): Promise<GeneralResponse.SuccessItem | GeneralResponse.ErrorItem> {
+  async ResetGenerateImage(id: string, operateBookType: OperateBookType): Promise<GeneralResponse.SuccessItem | GeneralResponse.ErrorItem> {
     try {
       let bookTaskDetails = undefined as Book.SelectBookTaskDetail[]
       let bookTask = undefined as Book.SelectBookTask
@@ -58,9 +62,8 @@ export class BookImage {
         throw new Error('不支持的操作类型，请检查')
       }
       //这边过滤被锁定的数据
-      if (!coverData) {
+      bookTaskDetails = bookTaskDetails.filter(item => !item.imageLock)
-        bookTaskDetails = bookTaskDetails.filter(item => !item.imageLock)
+
-      }
 
       if (bookTaskDetails.length <= 0) {
         throw new Error('没有要删除的分镜数据，请检查')
@@ -69,15 +72,15 @@ export class BookImage {
       // 开始删除数据，要删除图片数据和出图的信息
       for (let i = 0; i < bookTaskDetails.length; i++) {
         const element = bookTaskDetails[i];
-        if (bookTask.imageCategory == BookImageCategory.MJ) {
+        await this.bookServiceBasic.DeleteBoookTaskDetailGenerateImage(element.id);
-          await this.bookServiceBasic.DeleteBoookTaskDetailGenerateImage(element.id);
+        // if (bookTask.imageCategory == BookImageCategory.MJ) {
-        } else if (bookTask.imageCategory == BookImageCategory.D3) {
+        // } else if (bookTask.imageCategory == BookImageCategory.D3) {
-          throw new Error('暂时不支持D3生成的图片删除')
+        //   throw new Error('暂时不支持D3生成的图片删除')
-        } else if (bookTask.imageCategory == BookImageCategory.SD) {
+        // } else if (bookTask.imageCategory == BookImageCategory.SD) {
-          await this.bookServiceBasic.DeleteBoookTaskDetailGenerateImage(element.id);
+        //   await this.bookServiceBasic.DeleteBoookTaskDetailGenerateImage(element.id);
-        } else {
+        // } else {
-          throw new Error('未知的小说类型，请检查')
+        //   throw new Error('未知的小说类型，请检查')
-        }
+        // }
         // 上面的信息重置完毕之后，开始删除文件信息
         let outImage = element.outImagePath;
         if (await CheckFileOrDirExist(outImage)) {
@@ -322,7 +325,7 @@ export class BookImage {
    * @param imageUrl 图片地址
    * @returns 
    */
-  async DownloadImageUrlAndSplit(bookTaskDetailId: string, imageUrl: string,) {
+  async DownloadImageUrlAndSplit(bookTaskDetailId: string, imageUrl: string,): Promise<GeneralResponse.ErrorItem | GeneralResponse.SuccessItem> {
     try {
       let bookTaskDetail = await this.bookServiceBasic.GetBookTaskDetailDataById(bookTaskDetailId)
       if (bookTaskDetail == null) {
@@ -348,19 +351,33 @@ export class BookImage {
         await this.tools.downloadFileUrl(imageUrl, imagePath)
       }
 
-      // 进行图片裁剪
+      let imageCategory = bookTask.imageCategory;
-      let imageRes = await ImageSplit(imagePath, bookTaskDetail.name, path.join(book.bookFolderPath, 'data\\MJOriginalImage'));
+      let out_file = undefined
-      if (imageRes && imageRes.length < 4) {
+      let imageRes = []
-        throw new Error("图片裁剪失败")
+      if (imageCategory == BookImageCategory.MJ) {
+        // 只有MJ需要裁剪，其他的不需要
+        // 进行图片裁剪
+        imageRes = await ImageSplit(imagePath, bookTaskDetail.name, path.join(book.bookFolderPath, 'data\\MJOriginalImage'));
+        if (imageRes && imageRes.length < 4) {
+          throw new Error("图片裁剪失败")
+        }
+        // 修改数据
+        // 修改数据库数据，将图片保存到对应的文件夹中
+        let firstImage = imageRes[0];
+        if (book.type == BookType.ORIGINAL) {
+          await CopyFileOrFolder(firstImage, path.join(book.bookFolderPath, `tmp\\input\\${bookTaskDetail.name}.png`));
+        }
+        out_file = path.join(bookTask.imageFolder, `${bookTaskDetail.name}.png`)
+        await CopyFileOrFolder(firstImage, out_file);
+      } else {
+        // 其他的导入，每次只能导入一张图
+        out_file = path.join(bookTask.imageFolder, `${bookTaskDetail.name}.png`);
+        if (book.type == BookType.ORIGINAL) {
+          await CopyFileOrFolder(imagePath, path.join(book.bookFolderPath, `tmp\\input\\${bookTaskDetail.name}.png`));
+        }
+        await CopyFileOrFolder(imagePath, out_file);
+        imageRes = [out_file]
       }
-      // 修改数据
-      // 修改数据库数据，将图片保存到对应的文件夹中
-      let firstImage = imageRes[0];
-      if (book.type == BookType.ORIGINAL) {
-        await CopyFileOrFolder(firstImage, path.join(book.bookFolderPath, `tmp\\input\\${bookTaskDetail.name}.png`));
-      }
-      let out_file = path.join(bookTask.imageFolder, `${bookTaskDetail.name}.png`)
-      await CopyFileOrFolder(firstImage, out_file);
 
       // 修改分镜的数据
       await this.bookServiceBasic.UpdateBookTaskDetail(bookTaskDetailId, {
@@ -368,9 +385,20 @@ export class BookImage {
         subImagePath: imageRes.map((item) => path.relative(define.project_path, item))
       })
 
+      let mjMessage = {
+        status: 'success',
+        progress: 100,
+        category: MJImageType.IMPORT,
+        messageId: '',
+        action: MJAction.IMAGINE,
+      };
+      await this.bookServiceBasic.UpdateBookTaskDetailMjMessage(bookTaskDetailId, mjMessage)
+
+
       return successMessage({
         outImagePath: out_file + '?time=' + new Date().getTime(),
-        subImagePath: imageRes.map((item) => item + '?time=' + new Date().getTime())
+        subImagePath: imageRes.map((item) => item + '?time=' + new Date().getTime()),
+        mjMessage: mjMessage
       }, "下载指定的图片地址并且分割成功", "BookImage_DownloadImageUrlAndSplit")
     } catch (error) {
       return {
@@ -379,4 +407,74 @@ export class BookImage {
       }
     }
   }
+
+
+  /**
+   * 获取指定的图片链接 ，然后下载图片
+   * @param id 
+   * @param operateBookType 
+   * @param coverData 
+   */
+  async GetImageUrlAndDownload(id: string, operateBookType: OperateBookType, coverData: boolean): Promise<GeneralResponse.SuccessItem | GeneralResponse.ErrorItem> {
+    try {
+      console.log("GetImageUrlAndDownload", id, operateBookType, coverData)
+      let bookTaskDetail = undefined as Book.SelectBookTaskDetail[]
+      let bookTask = undefined as Book.SelectBookTask
+
+      if (operateBookType == OperateBookType.BOOKTASK) {
+        bookTask = await this.bookServiceBasic.GetBookTaskDataById(id);
+        bookTaskDetail = await this.bookServiceBasic.GetBookTaskDetailData({
+          bookTaskId: bookTask.id
+        })
+        // 这边过滤出图成功的数据
+        if (!coverData) {
+          bookTaskDetail = bookTaskDetail.filter((item) => !item.outImagePath)
+        }
+      } else if (operateBookType == OperateBookType.BOOKTASKDETAIL) {
+        let currentBookTaskDetail = await this.bookServiceBasic.GetBookTaskDetailDataById(id);
+        bookTask = await this.bookServiceBasic.GetBookTaskDataById(currentBookTaskDetail.bookTaskId);
+        bookTaskDetail = [currentBookTaskDetail]
+      } else {
+        throw new Error('不支持的操作类型')
+      }
+
+      // 这边再做个详细的筛选
+
+      if (bookTaskDetail.length < 0) {
+        throw new Error("没有找到需要采集的数据")
+      }
+      if (bookTask.imageCategory != BookImageCategory.MJ) {
+        throw new Error("只有MJ模式下才能使用这个功能")
+      }
+      let result = []
+
+      for (let i = 0; i < bookTaskDetail.length; i++) {
+        const element = bookTaskDetail[i];
+        if (!element.mjMessage) continue;
+        if (element.mjMessage.status == 'error') continue;
+        if (isEmpty(element.mjMessage.messageId)) continue;
+        // 这边开始采集
+        let res = await this.mjApi.GetMJAPITaskById(element.mjMessage.messageId, undefined);
+        if (isEmpty(res.imagePath)) {
+          throw new Error("获取图片地址链接为空")
+        }
+        // 开始下载
+        let dr = await this.DownloadImageUrlAndSplit(element.id, res.imagePath);
+        if (dr.code == 0) {
+          throw new Error(dr.message)
+        }
+        result.push({
+          id: element.id,
+          data: dr.data
+        })
+      }
+
+      if (result.length <= 0) {
+        throw new Error("没有找到需要采集的数据")
+      }
+      return successMessage(result, "获取图片链接并且下载成功", "BookImage_GetImageUrlAndDownload")
+    } catch (error) {
+      return errorMessage('获取图片链接并且下载失败，错误信息如下：' + error.message, 'BookImage_GetImageUrlAndDownload')
+    }
+  }
 }

src/main/Service/Book/bookPrompt.ts

@@ -7,6 +7,11 @@ import { BookServiceBasic } from "../ServiceBasic/bookServiceBasic";
 import { GptService } from "../GPT/gpt";
 import { ExecuteConcurrently } from "../../../define/Tools/common";
 import { DEFINE_STRING } from "../../../define/define_string";
+import { CheckFileOrDirExist } from "../../../define/Tools/file";
+import fs from 'fs';
+import path from 'path'
+import readline from 'readline';
+
 
 export class BookPrompt {
   bookServiceBasic: BookServiceBasic
@@ -16,6 +21,74 @@ export class BookPrompt {
     this.gptService = new GptService()
   }
 
+  //#region 提示词通用
+
+  /**
+   * 导入提示功能的函数
+   * 该函数用于根据给定的小说任务ID和文本路径来处理一些导入逻辑
+   * 
+   * @param bookTaskId 小说任务的唯一标识符，用于跟踪和管理不同的导入任务
+   * @param txtPath 待导入的文本文件的路径，指定了要处理的具体资源
+   */
+  async ImportGPTPrompt(bookTaskId: string, txtPath: string): Promise<GeneralResponse.SuccessItem | GeneralResponse.ErrorItem> {
+    try {
+      console.log(bookTaskId, txtPath)
+      let bookTaskDetail = await this.bookServiceBasic.GetBookTaskDetailData({
+        bookTaskId: bookTaskId
+      })
+
+      if (isEmpty(txtPath)) {
+        throw new Error("传入的文件地址不能为空")
+      }
+      // 检查文件是不是存在
+      if (!await CheckFileOrDirExist(txtPath)) {
+        throw new Error("传入的文件地址对应的文件不存在");
+      }
+
+      async function processLineByLine() {
+        const fileStream = fs.createReadStream(path.normalize(txtPath));
+        const rl = readline.createInterface({
+          input: fileStream,
+          crlfDelay: Infinity
+        });
+        const lines = [];
+        for await (const line of rl) {
+          lines.push(line);
+        }
+        rl.close(); // 确保关闭 readline 流
+        return lines;
+      }
+      let result = []
+      // 这个就是txt文件里面所有的数据
+      let lines = await processLineByLine()
+      // 这边开始写入
+      for (let i = 0; i < bookTaskDetail.length; i++) {
+        const element = bookTaskDetail[i];
+        if (i >= lines.length) {
+          break
+        }
+        // 修改
+        this.bookServiceBasic.transaction((realm) => {
+          let btd = realm.objectForPrimaryKey("BookTaskDetail", element.id)
+          if (btd == null) {
+            throw new Error('未找到对应的分镜数据')
+          }
+          btd.gptPrompt = lines[i]
+        })
+        result.push({
+          id: element.id,
+          gptPrompt: lines[i]
+        })
+      }
+      return successMessage(result, "导入提示词数据成功", 'Book_ImportGPTPrompt');
+    } catch (error) {
+      return errorMessage("导入提示词失败，错误信息如下：" + error.message, "Book_ImportGPTPrompt")
+    }
+  }
+
+  //#endregion
+
+
   //#region 反推的提示词相关
   /**
    * 将MJ反推出来的数据，选择指定的写入到GPT提示词中

src/main/Service/Book/bookTask.ts

@@ -50,8 +50,8 @@ export class BookTask {
         name: element.name,
         bookId: element.bookId,
         bookTaskId: newBookTask.id,
-        videoPath: path.relative(define.project_path, element.videoPath),
+        videoPath: element.videoPath ? path.relative(define.project_path, element.videoPath) : undefined,
-        oldImage: path.relative(define.project_path, element.oldImage),
+        oldImage: element.oldImage ? path.relative(define.project_path, element.oldImage) : undefined,
         adetailer: element.adetailer,
         sdConifg: element.sdConifg,
         createTime: new Date(),
@@ -130,7 +130,7 @@ export class BookTask {
       generateVideoPath: undefined,
       srtPath: bookTask.srtPath,
       audioPath: bookTask.audioPath,
-      imageFolder: path.relative(define.project_path, imageFolder),
+      imageFolder: imageFolder ? path.relative(define.project_path, imageFolder) : undefined,
       status: BookTaskStatus.WAIT,
       errorMsg: undefined,
       updateTime: new Date(),
@@ -278,16 +278,16 @@ export class BookTask {
         throw new Error("没有找到对应的数小说任务，请检查数据")
       }
       // 获取所有的出图中最少的
-      let bookTaskDetail = (await this.bookServiceBasic.GetBookTaskData({
+      let bookTaskDetail = (await this.bookServiceBasic.GetBookTaskDetailData({
         bookTaskId: bookTaskId
-      })).bookTasks as Book.SelectBookTaskDetail[]
+      })) as Book.SelectBookTaskDetail[]
       if (bookTaskDetail == null || bookTaskDetail.length <= 0) {
         throw new Error("没有对应的小说分镜任务，请先添加分镜任务")
       }
 
       for (let i = 0; i < bookTaskDetail.length; i++) {
         const element = bookTaskDetail[i];
-        if (isEmpty(element.subImagePath)) {
+        if (!element.subImagePath) {
           throw new Error("检测到图片没有出完，请先检查出图")
         }
         if (element.subImagePath == null || element.subImagePath.length <= 0) {
@@ -324,7 +324,7 @@ export class BookTask {
 
       // 先处理文件夹的创建，包括小说任务的和小说任务分镜的
       for (let i = 0; i < copyCount; i++) {
-        let no = await this.bookServiceBasic.GetMaxBookTaskNo(sourceBookTask.bookId)
+        let no = await this.bookServiceBasic.GetMaxBookTaskNo(sourceBookTask.bookId) + i
         let name = 'output_0000' + no
         let imageFolder = path.join(define.project_path, `${sourceBookTask.bookId}/tmp/${name}`)
         await CheckFolderExistsOrCreate(imageFolder)
@@ -408,35 +408,35 @@ export class BookTask {
             }
           }
 
-          let addOneBookTaskDetail = {
+          let addOneBookTaskDetail = {} as Book.SelectBookTaskDetail
-            id: reverseId,
+          addOneBookTaskDetail.id = reverseId;
-            no: element.no,
+          addOneBookTaskDetail.no = element.no;
-            name: element.name,
+          addOneBookTaskDetail.name = element.name;
-            bookId: sourceBookTask.bookId,
+          addOneBookTaskDetail.bookId = sourceBookTask.bookId;
-            bookTaskId: addOneBookTask.id,
+          addOneBookTaskDetail.bookTaskId = addOneBookTask.id;
-            videoPath: path.relative(define.project_path, element.videoPath),
+          addOneBookTaskDetail.videoPath = element.videoPath ? path.relative(define.project_path, element.videoPath) : undefined;
-            word: element.word,
+          addOneBookTaskDetail.word = element.word;
-            oldImage: path.relative(define.project_path, element.oldImage),
+          addOneBookTaskDetail.oldImage = element.oldImage ? path.relative(define.project_path, element.oldImage) : undefined;
-            afterGpt: element.afterGpt,
+          addOneBookTaskDetail.afterGpt = element.afterGpt;
-            startTime: element.startTime,
+          addOneBookTaskDetail.startTime = element.startTime;
-            endTime: element.endTime,
+          addOneBookTaskDetail.endTime = element.endTime;
-            timeLimit: element.timeLimit,
+          addOneBookTaskDetail.timeLimit = element.timeLimit;
-            subValue: element.subValue && element.subValue.length > 0 ? JSON.stringify(element.subValue) : undefined,
+          addOneBookTaskDetail.subValue = (element.subValue && element.subValue.length > 0 ? JSON.stringify(element.subValue) : undefined) as string;
-            characterTags: element.characterTags && element.characterTags.length > 0 ? cloneDeep(element.characterTags) : [],
+          addOneBookTaskDetail.characterTags = element.characterTags && element.characterTags.length > 0 ? cloneDeep(element.characterTags) : [];
-            gptPrompt: element.gptPrompt,
+          addOneBookTaskDetail.gptPrompt = element.gptPrompt;
-            outImagePath: path.relative(define.project_path, outImagePath),
+          addOneBookTaskDetail.outImagePath = outImagePath ? path.relative(define.project_path, outImagePath) : undefined;
-            subImagePath: subImagePath || [],
+          addOneBookTaskDetail.subImagePath = subImagePath || [];
-            prompt: element.prompt,
+          addOneBookTaskDetail.prompt = element.prompt;
-            adetailer: element.adetailer,
+          addOneBookTaskDetail.adetailer = element.adetailer;
-            sdConifg: sdConifg,
+          addOneBookTaskDetail.sdConifg = sdConifg;
-            createTime: new Date(),
+          addOneBookTaskDetail.createTime = new Date();
-            updateTime: new Date(),
+          addOneBookTaskDetail.updateTime = new Date();
-            audioPath: element.audioPath,
+          addOneBookTaskDetail.audioPath = element.audioPath;
-            subtitlePosition: element.subtitlePosition,
+          addOneBookTaskDetail.subtitlePosition = element.subtitlePosition;
-            status: element.status,
+          addOneBookTaskDetail.status = element.status;
-            reversePrompt: reverseMessage,
+          addOneBookTaskDetail.reversePrompt = reverseMessage;
-            imageLock: element.imageLock
+          addOneBookTaskDetail.imageLock = element.imageLock
-          } as Book.SelectBookTaskDetail
+
           addBookTaskDetail.push(addOneBookTaskDetail)
         }
       }

src/main/Service/MJ/mj.ts

@@ -775,7 +775,7 @@ export class MJOpt {
           messageId: undefined,
           id: task.bookTaskDetailId,
           progress: 0,
-          message: error.toString(),
+          message: errorMsg,
           status: "error"
         }
       }, task.messageName)
@@ -788,7 +788,7 @@ export class MJOpt {
         messageId: "",
         action: MJAction.IMAGINE,
         status: "error",
-        message: error.toString()
+        message: errorMsg
       })
       return errorMessage(errorMsg, "MJReverse_MJImage2Text")
     }

src/main/Service/MJ/mjApi.ts

@@ -9,6 +9,7 @@ import { MJSetting } from "../../../model/Setting/mjSetting"
 import { GPT } from "../../Public/GPT"
 import { MJ } from "../../../model/mj"
 import { LaiAPIType } from "../../../define/enum/softwareEnum"
+import { isEmpty } from "lodash"
 
 /**
  * 调用MJ的API类
@@ -96,11 +97,13 @@ class MJApi {
       let _bookBackTaskListService = await BookBackTaskListService.getInstance()
       // 失败
       if (code == 0) {
-        _bookBackTaskListService.UpdateTaskStatus({
+        if (!isEmpty(backTaskId)) {
-          id: backTaskId,
+          _bookBackTaskListService.UpdateTaskStatus({
-          status: BookBackTaskStatus.FAIL,
+            id: backTaskId,
-          errorMessage: res.data.message
+            status: BookBackTaskStatus.FAIL,
-        })
+            errorMessage: res.data.message
+          })
+        }
       }
       let resObj = {
         type: MJRespoonseType.UPDATED,

src/model/book.d.ts

@@ -88,9 +88,9 @@ declare namespace Book {
 
   // 字幕相关
   type Subtitle = {
-    startTime: number
+    start_time: number
-    endTime: number
+    end_time: number
-    srtValue: string
+    srt_value: string
     id: string
   }
 
@@ -152,7 +152,7 @@ declare namespace Book {
     startTime?: number  // 开始时间
     endTime?: number  // 结束时间
     timeLimit?: string  // 事件实现（0 -- 3000）
-    subValue?: string  // 包含的字幕数据
+    subValue?: Subtitle[] | string // 包含的字幕数据
     characterTags?: string[]  // 角色标签
     sceneTags?: string[]  // 场景标签
     gptPrompt?: string // GPT提示词
@@ -279,8 +279,8 @@ declare namespace Book {
 
   type ReplaceDataRes = {
     bookTaskDetailId: string,
-    newData: string,
+    newData: string | Subtitle[],
-    type?: 'reversePrompt' | 'gptPrompt' | 'afterGpt' | 'prompt',
+    type?: 'reversePrompt' | 'gptPrompt' | 'afterGpt' | 'prompt' | 'subValue',
     reversePromptType?: 'prompt' | 'promptCN'
     reversePromptId?: string
   }

src/preload/book.ts

@@ -158,6 +158,8 @@ const book = {
   OriginalGetPrompt: async (id: string, operateBookType: OperateBookType, coverData: boolean) =>
     await ipcRenderer.invoke(DEFINE_STRING.BOOK.ORIGINAL_GPT_PROMPT, id, operateBookType, coverData),
 
+  // 导入提示词
+  ImportGPTPrompt: async (bookTaskId: string, txtPath: string) => await ipcRenderer.invoke(DEFINE_STRING.BOOK.IMPORT_GPT_PROMPT, bookTaskId, txtPath),
   //#endregion
 
   //#region 图片相关
@@ -166,6 +168,10 @@ const book = {
   ResetGenerateImage: async (id: string, operateBookType: OperateBookType, coverData: boolean) =>
     await ipcRenderer.invoke(DEFINE_STRING.BOOK.REMOVE_GENERATE_IMAGE, id, operateBookType, coverData),
 
+  // 获取图片的连接，然后下载图片
+  GetImageUrlAndDownload: async (id: string, operateBookType: OperateBookType, coverData: boolean) =>
+    await ipcRenderer.invoke(DEFINE_STRING.BOOK.GET_IMAGE_URL_AND_DOWNLOAD, id, operateBookType, coverData),
+
   //#endregion
 
   //#region  一键反推的单个任务

src/renderer/src/components/Book/Components/DatatableAfterGpt.vue

@@ -1,11 +1,11 @@
 <template>
-  <div style="height: 135px; width : 100%;overflow: auto; display: flex; margin: auto">
+  <div style="height: 135px; width: 100%; overflow: auto; display: flex; margin: auto">
-    <div v-if="data.subValue && data.subValue.length != 0" style="width : 100%">
+    <div v-if="data.subValue && data.subValue.length != 0" style="width: 100%">
-      <div v-for="(item, index) in data.subValue" style="width : 100%">
+      <div v-for="(item, index) in data.subValue" style="width: 100%">
         <n-input
           v-model:value="item.srt_value"
           type="textarea"
-          style="width : 100%"
+          style="width: 100%"
           size="tiny"
           :autosize="{ minRows: 1, maxRows: 5 }"
           @input="InputDebounced"
@@ -26,7 +26,7 @@
 </template>
 
 <script>
-import { ref, onMounted, defineComponent, onUnmounted, toRaw, watch, computed } from 'vue'
+import { ref, onMounted, defineComponent } from 'vue'
 import { useMessage, NInput } from 'naive-ui'
 import { debounce } from 'lodash'
 
@@ -35,6 +35,7 @@ export default defineComponent({
   props: ['initData', 'index'],
   setup(props) {
     let data = ref(props.initData)
+    console.log('data', props.initData, props.index)
     onMounted(async () => {})
     let InputDebounced = debounce(handleInput, 1000)
 

src/renderer/src/components/Book/Components/DatatableGenerateImage.vue

@@ -75,6 +75,7 @@
           type="info"
           style="font-size: 24px; position: absolute; left: 8px; top: -5px"
           @click="DownloadImage"
+          v-if="reverseManageStore.selectBookTask.imageCategory == 'mj'"
         >
           <n-icon>
             <Download />
@@ -116,7 +117,7 @@
 </template>
 
 <script setup>
-import { ref, onMounted, onUnmounted, watch } from 'vue'
+import { ref, onMounted, toRaw, watch } from 'vue'
 import {
   NImage,
   useDialog,
@@ -132,6 +133,8 @@ import { LockClosed, LockOpen, Download } from '@vicons/ionicons5'
 import { useSoftwareStore } from '../../../../../stores/software'
 import { useReverseManageStore } from '../../../../../stores/reverseManage'
 import { useImageStore } from '../../../../../stores/image'
+import { BookImageCategory, OperateBookType } from '../../../../../define/enum/bookEnum'
+import { isEmpty } from 'lodash'
 let props = defineProps({
   initData: undefined,
   index: undefined
@@ -268,53 +271,55 @@ async function ModifyLock() {
  */
 async function DownloadImage() {
   // 目前只能判断是不是有mj_message
-  dialog.warning({
+  let da = dialog.warning({
     title: '下载图片警告',
-    content: '单个图片的下载，只支持 MJ API 生图，当前有图片的话会被覆盖掉，是否继续？',
+    content:
+      '单个图片的下载，只支持 MJ API 生图，MJ图片有效时间大约为24小时，当前有图片的话会被覆盖掉，是否继续？',
     positiveText: '继续',
     negativeText: '取消',
     onPositiveClick: async () => {
-      if (data.value.mj_message) {
+      da?.destroy()
-        debugger
+      if (reverseManageStore.selectBookTask.imageCategory != BookImageCategory.MJ) {
-        // 加判断
+        message.error('当前图片不是MJ图片，不能下载')
-        // 只要状态不是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
+      }
+      // 只要状态不是success 和 error ，其他的都重新获取
+      if (data.value.mjMessage.status == 'error') {
+        message.error('失败状态不能采集图片')
+        return
+      }
+      if (isEmpty(data.value.mjMessage.messageId)) {
+        message.error('没有消息ID，不能采集图片')
+        return
+      }
+
+      // 开始下载图片
+      softwareStore.spin.spinning = true
+      softwareStore.spin.tip = '正在下载图片，请稍后...'
+      let res = await window.book.GetImageUrlAndDownload(
+        data.value.id,
+        OperateBookType.BOOKTASKDETAIL,
+        false
+      )
+      softwareStore.spin.spinning = false
+      if (res.code == 1) {
+        // 这边要修改下数据
+        for (let i = 0; i < res.data.length; i++) {
+          const element = res.data[i]
+          let findIndex = reverseManageStore.selectBookTaskDetail.findIndex(
+            (item) => item.id == element.id
+          )
+          if (findIndex != -1) {
+            reverseManageStore.selectBookTaskDetail[findIndex].outImagePath =
+              element.data.outImagePath
+            reverseManageStore.selectBookTaskDetail[findIndex].subImagePath =
+              element.data.subImagePath
+            reverseManageStore.selectBookTaskDetail[findIndex].mjMessage = element.data.mjMessage
+          }
+        }
+        message.success(res.message)
       } else {
-        message.error('没有生图数据')
+        message.error(res.message)
       }
     }
   })

src/renderer/src/components/Book/Components/DatatableHeaderAfterGptSelectAndReplace.vue

@@ -65,7 +65,11 @@ async function ReplaceAfterGpt() {
           (x) => x.id == element.bookTaskDetailId
         )
         if (index != -1) {
-          reverseManageStore.selectBookTaskDetail[index].afterGpt = element.newData
+          if (element.type == 'subValue') {
+            reverseManageStore.selectBookTaskDetail[index].subValue = element.newData
+          } else {
+            reverseManageStore.selectBookTaskDetail[index].afterGpt = element.newData
+          }
         }
       }
     } else if (props.type == BookRepalceDataType.GPT_PROMPT) {

src/renderer/src/components/Book/Components/DatatableHeaderGenerateImage.vue

@@ -1,5 +1,5 @@
 <template>
-  <div style="display: flex">
+  <div style="display: flex; align-items: center">
     <span>出图</span>
     <n-select
       style="width: 100px; margin-left: 5px"
@@ -9,20 +9,29 @@
       v-model:value="reverseManageStore.selectBookTask.imageCategory"
       placeholder="选择生图模式"
     ></n-select>
-    <n-button
+    <div v-if="reverseManageStore.selectBookTask.imageCategory == 'mj'">
-      v-if="reverseManageStore.selectBookTask.imageCategory == 'mj'"
+      <n-popover trigger="hover">
-      color="#7c461e"
+        <template #trigger>
-      size="tiny"
+          <n-button
-      style="margin-left: 5px"
+            text
-      @click="MJGetImage"
+            type="info"
-      >MJ采集图片</n-button
+            color="#e18a3b"
-    >
+            style="font-size: 24px; margin-left: 5px"
+            @click="DownloadAllImage"
+          >
+            <n-icon> <Download /> </n-icon>
+          </n-button>
+        </template>
+        <span>MJ采集全部图片</span>
+      </n-popover>
+    </div>
+
     <n-button
       v-if="reverseManageStore.selectBookTask.imageCategory == 'mj'"
       color="#e18a3b"
       size="tiny"
       style="margin-left: 5px"
-      @click="OneSplitFour"
+      @click="OneToFourBookTask"
       >1拆4</n-button
     >
   </div>
@@ -30,11 +39,14 @@
 
 <script setup>
 import { ref, onMounted } from 'vue'
-import { NSelect, NButton, useMessage } from 'naive-ui'
+import { NSelect, NButton, useMessage, useDialog, NPopover, NIcon } from 'naive-ui'
 import { useSoftwareStore } from '../../../../../stores/software'
 import { useReverseManageStore } from '../../../../../stores/reverseManage'
+import { Download } from '@vicons/ionicons5'
+import { OperateBookType } from '../../../../../define/enum/bookEnum'
 let softwareStore = useSoftwareStore()
 let message = useMessage()
+let dialog = useDialog()
 let reverseManageStore = useReverseManageStore()
 let image_options = ref([])
 
@@ -49,6 +61,7 @@ onMounted(async () => {
   })
 })
 
+// 更行生图方式
 async function UpdateImageGenerateCategory(value, options) {
   // 调用下保存
   let res = await window.db.UpdateBookTaskData(reverseManageStore.selectBookTask.id, {
@@ -56,4 +69,59 @@ async function UpdateImageGenerateCategory(value, options) {
   })
   window.api.showGlobalMessage(res)
 }
+
+async function DownloadAllImage() {
+  let da = dialog.warning({
+    title: '采集所有图片提示',
+    content:
+      '即将开始采集所有的MJ生图图片，满足一下条件的分镜才会被采集：状态不为 error，有生图信息，有消息ID，没有生成图片，图片的有效期为24小时，是否继续？',
+    positiveText: '继续',
+    negativeText: '取消',
+    onPositiveClick: async () => {
+      da?.destroy()
+      let res = await window.book.GetImageUrlAndDownload(
+        reverseManageStore.selectBookTask.id,
+        OperateBookType.BOOKTASK,
+        false
+      )
+      if (res.code == 1) {
+        // 这边要修改下数据
+        for (let i = 0; i < res.data.length; i++) {
+          const element = res.data[i]
+          let findIndex = reverseManageStore.selectBookTaskDetail.findIndex(
+            (item) => item.id == element.id
+          )
+          if (findIndex != -1) {
+            reverseManageStore.selectBookTaskDetail[findIndex].outImagePath =
+              element.data.outImagePath
+            reverseManageStore.selectBookTaskDetail[findIndex].subImagePath =
+              element.data.subImagePath
+            reverseManageStore.selectBookTaskDetail[findIndex].mjMessage = element.data.mjMessage
+          }
+        }
+        message.success(res.message)
+      } else {
+        message.error(res.message)
+      }
+    }
+  })
+}
+
+async function OneToFourBookTask() {
+  dialog.warning({
+    title: '一拆四提示',
+    content:
+      '是否一拆四（一拆四是个泛指，根据出的子图数量最少的），会自动生成多个批次任务并设置对应的图片！',
+    positiveText: '确定',
+    negativeText: '取消',
+    onPositiveClick: async () => {
+      let res = await window.book.OneToFourBookTask(reverseManageStore.selectBookTask.id)
+      if (res.code == 0) {
+        message.error(res.message)
+        return
+      }
+      message.success(res.message)
+    }
+  })
+}
 </script>

src/renderer/src/components/Book/Components/ManageBook/AddBook.vue

@@ -117,7 +117,7 @@ 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 { CloseSharp } from '@vicons/ionicons5'
-import { cloneDeep } from 'lodash'
+import { cloneDeep, isEmpty } from 'lodash'
 import { useSoftwareStore } from '../../../../../../stores/software.ts'
 import { BookType } from '../../../../../../define/enum/bookEnum'
 
@@ -227,9 +227,15 @@ export default defineComponent({
       oldVideoPath: {
         required: true,
         validator(rule, value) {
-          // 判断是不是原创，原创不能选择视频文件，反推必选
+          debugger
-          if(reverseManageStore.selectBook.type == BookType.SD_REVERSE || reverseManageStore.selectBook.type == BookType.MJ_REVERSE){
+          if (isEmpty(value)) {
-            return new Error("MJ反推和SD反推，视频文件必选")
+            // 判断是不是原创，原创不能选择视频文件，反推必选
+            if (
+              reverseManageStore.selectBook.type == BookType.SD_REVERSE ||
+              reverseManageStore.selectBook.type == BookType.MJ_REVERSE
+            ) {
+              return new Error('MJ反推和SD反推，视频文件必选')
+            }
           }
         },
         trigger: ['input', 'blur', 'change']

src/renderer/src/components/Book/Original/ODataTableHeaderGptPrompt.vue

@@ -1,66 +1,83 @@
 <template>
-  <div>
+  <div style="display: flex; align-items: center">
-    <span style="margin-right: 5px">推理提示词</span>
+    <div>
+      <span style="margin-right: 5px">推理提示词</span>
+      <n-popover trigger="hover">
+        <template #trigger>
+          <n-button
+            :color="softwareStore.SoftColor.ZHUYANTUO"
+            style="margin-left: 5px"
+            size="tiny"
+            @click="AddPrefix"
+            >通用前缀</n-button
+          >
+        </template>
+        <span>添加通用前缀，只能添加英文，并且通用前缀不为空，推理会自动添加前缀</span>
+      </n-popover>
+      <n-popover trigger="hover">
+        <template #trigger>
+          <n-button
+            :color="softwareStore.SoftColor.ZHUYANTUO"
+            style="margin-left: 5px"
+            size="tiny"
+            @click="AddSuffix"
+            >通用后缀</n-button
+          >
+        </template>
+        <span>添加通用后缀，只能添加英文，并且通用后缀不为空，推理会自动添加后缀</span>
+      </n-popover>
+      <n-popover trigger="hover">
+        <template #trigger>
+          <n-button
+            :color="softwareStore.SoftColor.ZHUYANTUO"
+            style="margin-left: 5px"
+            size="tiny"
+            @click="SelectStyle"
+            >风格</n-button
+          >
+        </template>
+        <span>选择生成图片的风格</span>
+      </n-popover>
+      <n-popover trigger="hover">
+        <template #trigger>
+          <span>
+            <n-dropdown trigger="hover" :options="translate_options" @select="TranslateAll">
+              <n-button
+                :color="softwareStore.SoftColor.ZHUYANTUO"
+                style="margin-left: 5px"
+                size="tiny"
+                @click="TranslateAll('english')"
+                >一键翻译</n-button
+              >
+            </n-dropdown>
+          </span>
+        </template>
+        <span
+          >将下面的推理提示词或者是手写的提示词翻译成英文（只会翻译存在中文的数据），下面的翻译中文会将所有的推理提示词翻译为中文（全部都会翻译）</span
+        >
+      </n-popover>
+    </div>
     <n-popover trigger="hover">
       <template #trigger>
         <n-button
-          :color="softwareStore.SoftColor.ZHUYANTUO"
+          color="#7c461e"
-          style="margin-left: 5px"
+          @click="SelectReversePromptAndReplace"
-          size="tiny"
+          text
-          @click="AddPrefix"
+          style="font-size: 24px; margin-left: 3px"
-          >通用前缀</n-button
         >
+          <n-icon>
+            <FindReplaceRound />
+          </n-icon>
+        </n-button>
       </template>
-      <span>添加通用前缀，只能添加英文，并且通用前缀不为空，推理会自动添加前缀</span>
+      <span>批量替换推理提示词</span>
-    </n-popover>
-    <n-popover trigger="hover">
-      <template #trigger>
-        <n-button
-          :color="softwareStore.SoftColor.ZHUYANTUO"
-          style="margin-left: 5px"
-          size="tiny"
-          @click="AddSuffix"
-          >通用后缀</n-button
-        >
-      </template>
-      <span>添加通用后缀，只能添加英文，并且通用后缀不为空，推理会自动添加后缀</span>
-    </n-popover>
-    <n-popover trigger="hover">
-      <template #trigger>
-        <n-button
-          :color="softwareStore.SoftColor.ZHUYANTUO"
-          style="margin-left: 5px"
-          size="tiny"
-          @click="SelectStyle"
-          >风格</n-button
-        >
-      </template>
-      <span>选择生成图片的风格</span>
-    </n-popover>
-    <n-popover trigger="hover">
-      <template #trigger>
-        <span>
-          <n-dropdown trigger="hover" :options="translate_options" @select="TranslateAll">
-            <n-button
-              :color="softwareStore.SoftColor.ZHUYANTUO"
-              style="margin-left: 5px"
-              size="tiny"
-              @click="TranslateAll('english')"
-              >一键翻译</n-button
-            >
-          </n-dropdown>
-        </span>
-      </template>
-      <span
-        >将下面的推理提示词或者是手写的提示词翻译成英文（只会翻译存在中文的数据），下面的翻译中文会将所有的推理提示词翻译为中文（全部都会翻译）</span
-      >
     </n-popover>
   </div>
 </template>
 
 <script setup>
 import { ref, h, toRaw } from 'vue'
-import { NPopover, NButton, NDropdown, useMessage, useDialog } from 'naive-ui'
+import { NPopover, NButton, NDropdown, useMessage, useDialog, NIcon } from 'naive-ui'
 import { useReverseManageStore } from '../../../../../stores/reverseManage'
 import { useSoftwareStore } from '../../../../../stores/software'
 import SelectImageStyle from '../../Components/SelectImageStyle.vue'
@@ -69,6 +86,9 @@ import { ContainsChineseOrPunctuation } from '../../../../../define/Tools/common
 import { isEmpty } from 'lodash'
 import { TranslateType } from '../../../../../define/enum/translate'
 import { DEFINE_STRING } from '../../../../../define/define_string'
+import FindReplaceRound from '../../Icon/FindReplaceRound.vue'
+import DatatableHeaderAfterGptSelectAndReplace from '../Components/DatatableHeaderAfterGptSelectAndReplace.vue'
+import { BookRepalceDataType } from '../../../../../define/enum/bookEnum'
 
 let softwareStore = useSoftwareStore()
 let reverseManageStore = useReverseManageStore()
@@ -288,4 +308,15 @@ async function SelectStyle() {
     }
   })
 }
+
+//
+async function SelectReversePromptAndReplace() {
+  dialog.create({
+    title: '批量查询替换反推提示词',
+    showIcon: false,
+    maskClosable: false,
+    content: () =>
+      h(DatatableHeaderAfterGptSelectAndReplace, { type: BookRepalceDataType.GPT_PROMPT })
+  })
+}
 </script>

src/renderer/src/components/Book/Original/ODatatableHeaderPrompt.vue

@@ -1,7 +1,6 @@
 <template>
   <div style="display: flex">
     <span style="margin-right: 5px">提示词命令</span>
-
     <n-popover trigger="hover">
       <template #trigger>
         <n-button
@@ -33,6 +32,22 @@
       @click="MJBadPromptCheck"
       >敏感词检查</n-button
     >
+
+    <n-popover trigger="hover">
+      <template #trigger>
+        <n-button
+          color="#7c461e"
+          @click="SelectPromptAndReplace"
+          text
+          style="font-size: 24px; margin-left: 10px"
+        >
+          <n-icon>
+            <FindReplaceRound />
+          </n-icon>
+        </n-button>
+      </template>
+      <span>批量查询替换文案</span>
+    </n-popover>
   </div>
 </template>
 
@@ -43,7 +58,9 @@ import { Construct } from '@vicons/ionicons5'
 import DynamicTagsSelect from '../../Components/DynamicTagsSelect.vue'
 import { useSoftwareStore } from '../../../../../stores/software'
 import { useReverseManageStore } from '../../../../../stores/reverseManage'
-import { OperateBookType } from '../../../../../define/enum/bookEnum'
+import { BookRepalceDataType, OperateBookType } from '../../../../../define/enum/bookEnum'
+import FindReplaceRound from '../../Icon/FindReplaceRound.vue'
+import DatatableHeaderAfterGptSelectAndReplace from '../Components/DatatableHeaderAfterGptSelectAndReplace.vue'
 let message = useMessage()
 let dialog = useDialog()
 let softwareStore = useSoftwareStore()
@@ -118,7 +135,7 @@ async function MergePrompt(type = null) {
     } else if (image_generate_category == 'mj') {
       mergeType = 'mj_merge'
     } else if (image_generate_category == 'd3') {
-      message.error('D3模式不支持')
+      mergeType = 'sd_merge'
     } else if (image_generate_category == 'flux-api' || image_generate_category == 'flux-forge') {
       mergeType = 'sd_merge'
     } else {
@@ -167,4 +184,13 @@ async function MergePromptSelect(key) {
 async function MJBadPromptCheck() {
   message.error('敏感词检测功能暂未开放')
 }
+
+async function SelectPromptAndReplace() {
+  dialog.create({
+    title: '批量查询替换生图提示词',
+    showIcon: false,
+    maskClosable: false,
+    content: () => h(DatatableHeaderAfterGptSelectAndReplace, { type: BookRepalceDataType.PROMPT })
+  })
+}
 </script>

src/renderer/src/components/Book/Original/ODatatablePrompt.vue

@@ -27,7 +27,7 @@
           :color="softwareStore.SoftColor.BROWN_YELLOW"
           style="margin-right: 5px"
           size="tiny"
-          @click="ImportMJImageUrl"
+          @click="ImportImageUrl"
           >导入图片</n-button
         >
       </template>
@@ -47,10 +47,11 @@
 </template>
 
 <script setup>
-import { ref } from 'vue'
+import { ref, h } from 'vue'
-import { NButton, NInput, NPopover, useMessage } from 'naive-ui'
+import { NButton, NInput, NPopover, useMessage, useDialog } from 'naive-ui'
 import { useSoftwareStore } from '../../../../../stores/software'
 import { useReverseManageStore } from '../../../../../stores/reverseManage'
+import InputDialogContent from '../../Original/Components/InputDialogContent.vue'
 import {
   BookBackTaskType,
   BookImageCategory,
@@ -66,7 +67,8 @@ let props = defineProps({
 let softwareStore = useSoftwareStore()
 let reverseManageStore = useReverseManageStore()
 let message = useMessage()
-
+let dialog = useDialog()
+let image_url_ref = ref(null)
 let row = ref(props.initData)
 
 let InputDebounced = debounce(handleInput, 500)
@@ -89,7 +91,7 @@ async function SingleMergePrompt() {
   } else if (image_generate_category == 'mj') {
     mergeType = 'mj_merge'
   } else if (image_generate_category == 'd3') {
-    message.error('D3模式不支持')
+    mergeType = 'sd_merge'
   } else if (image_generate_category == 'flux-api' || image_generate_category == 'flux-forge') {
     mergeType = 'sd_merge'
   } else {
@@ -262,4 +264,44 @@ async function NextGenerateImage() {
     message.error(res.message)
   }
 }
+
+async function ImportImageUrl() {
+  // 判断当前数据是不是存在
+  // 处理数据。获取当前的所有的数据
+  let dialogWidth = 400
+  let dialogHeight = 150
+  dialog.create({
+    title: '添加图片链接/本地地址',
+    showIcon: false,
+    closeOnEsc: false,
+    content: () =>
+      h(InputDialogContent, {
+        ref: image_url_ref,
+        initData: null,
+        placeholder: '请输入图片链接/本地地址'
+      }),
+    style: `width : ${dialogWidth}px; min-height : ${dialogHeight}px`,
+    maskClosable: false,
+    onClose: async () => {
+      let row_image_url = image_url_ref.value.data
+      softwareStore.spin.spinning = true
+      softwareStore.spin.tip = '正在下载图片/分割图片中。。。'
+      let res = await window.book.DownloadImageUrlAndSplit(row.value.id, row_image_url)
+      softwareStore.spin.spinning = false
+      if (res.code == 0) {
+        message.error(res.message)
+        return
+      }
+      let findIndex = reverseManageStore.selectBookTaskDetail.findIndex(
+        (item) => item.id == row.value.id
+      )
+      if (findIndex != -1) {
+        reverseManageStore.selectBookTaskDetail[findIndex].outImagePath = res.data.outImagePath
+        reverseManageStore.selectBookTaskDetail[findIndex].subImagePath = res.data.subImagePath
+        reverseManageStore.selectBookTaskDetail[findIndex].mjMessage = res.data.mjMessage
+      }
+      message.success(res.message)
+    }
+  })
+}
 </script>

src/renderer/src/components/Book/Original/OriginalMainButton.vue

@@ -77,6 +77,7 @@ import {
 } from '../../../../../define/enum/bookEnum'
 import { useRouter } from 'vue-router'
 import { DEFINE_STRING } from '../../../../../define/define_string'
+import { isEmpty } from 'lodash'
 let softwareStore = useSoftwareStore()
 let reverseManageStore = useReverseManageStore()
 let dialog = useDialog()
@@ -92,8 +93,8 @@ let GenerateImageOptions = ref([
   { label: '停止生成图片任务', key: 'stop_generate_image' }
 ])
 let ResetDataOptions = ref([
-  { label: '重置GPT提示词', key: 'reset_gpt_prompt' },
+  { label: '重置GPT提示词', key: 'reset_prompt' },
-  { label: '重置提示词', key: 'reset_prompt' },
+  { label: '重置提示词', key: 'reset_merge_prompt' },
   { label: '重置图片', key: 'reset_image' }
 ])
 
@@ -246,6 +247,39 @@ async function ResetGPTPrompt() {
   })
 }
 
+// 重置所有的合并提示词数据
+async function ResetMergePrompt() {
+  let da = dialog.warning({
+    title: '重置合并提示词提示',
+    content: '执行该操作会删除所有的合并提示词，并且此操作不可逆，是否继续？',
+    positiveText: '继续',
+    negativeText: '取消',
+    onPositiveClick: async () => {
+      da?.destroy()
+      softwareStore.spin.spinning = true
+      softwareStore.spin.tip = '正在重置推理提示词，请稍后。。。'
+      for (let i = 0; i < reverseManageStore.selectBookTaskDetail.length; i++) {
+        const element = reverseManageStore.selectBookTaskDetail[i]
+        let res = await window.db.UpdateBookTaskDetailData(element.id, {
+          prompt: ''
+        })
+        if (res.code == 1) {
+          let findIndex = reverseManageStore.selectBookTaskDetail.findIndex(
+            (item) => item.id == element.id
+          )
+          if (findIndex != -1) {
+            reverseManageStore.selectBookTaskDetail[findIndex].prompt = ''
+          }
+        } else {
+          message.error(res.message)
+        }
+      }
+      softwareStore.spin.spinning = false
+      message.success('重置提示词数据成功')
+    }
+  })
+}
+
 // 重置所有的生成图片信息
 async function ResetImage() {
   message.info('ResetImage')
@@ -295,6 +329,9 @@ async function ButtonSelect(key) {
   } else if (key == 'generate_space_image') {
     // 生成未出图分镜
     await GenerateImageAll(false)
+  } else if (key == 'reset_merge_prompt') {
+    // 重置合并提示词
+    await ResetMergePrompt()
   } else if (key == 'reset_image') {
     await ResetImage()
   } else {
@@ -454,47 +491,44 @@ async function OpenPromptSetting() {
 
 // 导入提示词
 async function ImportPrompt() {
-  message.error('导入提示词该方法暂未实现')
-  return
   // 弹窗选择文件
   window.api.SelectFile(['txt'], async (value) => {
     if (value.code == 0) {
       message.error(value.message)
       return
     }
-    // 拿到提示词文件地址，然后还是获取
+    if (isEmpty(value.value)) {
-    await window.pmpt.OpenPromptFileTxt(value.value, (res) => {
+      message.error('返回的文件地址为空')
-      debugger
+      return
-      console.log(res)
+    }
-      if (res.code == 0) {
+
-        message.error(res.message)
+    let da = dialog.warning({
-        return
+      title: '导入提示词提示',
-      }
+      content: `即将开始导入提示词，务必保证提示词文件的行数和分镜的行数相同，否则可能会导致数据丢失，当前的导入的文件地址为 ${value.value}，是否继续？`,
-      // 修改数据
+      positiveText: '继续',
-      // 判断获取的数据是不是小于当前的data的行
+      negativeText: '取消',
-      if (res.data.length > data.value.length) {
+      onPositiveClick: async () => {
-        dialog.warning({
+        da?.destroy()
-          title: '提示',
+        softwareStore.spin.spinning = true
-          content: '导入的数据行数大于当前的数据行数，多余的数据会被删除，是否继续导入？',
+        softwareStore.spin.tip = '正在导入提示词数据中。。。'
-          positiveText: '继续',
+        let res = await window.book.ImportGPTPrompt(
-          negativeText: '取消',
+          reverseManageStore.selectBookTask.id,
-          onPositiveClick: async () => {
+          value.value
-            debugger
+        )
-            // 开始导入
+        softwareStore.spin.spinning = false
-            for (let i = 0; i < data.value.length && i < res.data.length; i++) {
+        if (res.code == 1) {
-              const element = res.data[i]
+          // 这边还要修改下前端界面
-              // 开始修改
+          for (let i = 0; i < res.data.length; i++) {
-              data.value[i].gpt_prompt = element
+            const element = res.data[i]
+            let findIndex = reverseManageStore.selectBookTaskDetail.findIndex(
+              (item) => item.id == element.id
+            )
+            if (findIndex != -1) {
+              reverseManageStore.selectBookTaskDetail[findIndex].gptPrompt = element.gptPrompt
             }
           }
-        })
-      } else {
-        // 开始导入
-        for (let i = 0; i < data.value.length && i < res.data.length; i++) {
-          const element = res.data[i]
-          // 开始修改
-          data.value[i].gpt_prompt = element
         }
+        window.api.showGlobalMessage(res)
       }
     })
   })

src/renderer/src/components/Book/Original/OriginalMainDatatable.vue

@@ -107,6 +107,7 @@ const createColumns = ({}) => {
       className: 'empty-margin',
       fixed: 'left',
       render(row, index) {
+        console.log('row', row, index)
         return h(DatatableAfterGpt, { initData: row, index: index })
       }
     },
@@ -133,7 +134,7 @@ const createColumns = ({}) => {
       resizable: true,
       minWidth: 330,
       maxWidth: 700,
-      width: '350',
+      width: '360',
       render(row, index) {
         return h(ODataTableGptPrompt, { initData: row, index: index })
       }

src/stores/reverseManage.ts

@@ -230,10 +230,17 @@ export const useReverseManageStore = defineStore('reverseManage', {
      */
     async GetBookTaskDetail(bookTaskId: string, modifyProperty = null): Promise<GeneralResponse.ErrorItem | GeneralResponse.SuccessItem> {
       try {
-        debugger
         //@ts-ignore
         let detailRes = await window.book.GetBookTaskDetail(bookTaskId)
+        let bookTaskDetail = []
         if (detailRes.code == 1) {
+          bookTaskDetail = detailRes.data.map(item => {
+            return {
+              ...item,
+              outImagePath: item.outImagePath ? item.outImagePath + '?t=' + new Date().getTime() : undefined,
+              subImagePath: item.subImagePath ? item.subImagePath.map(item => item + '?t=' + new Date().getTime()) : []
+            }
+          })
           // 这边开始修改数据，
           if (this.selectBookTaskDetail && this.selectBookTaskDetail.length > 0) {
             this.selectBookTaskDetail.splice(0, this.selectBookTaskDetail.length, ...detailRes.data);
@@ -243,7 +250,7 @@ export const useReverseManageStore = defineStore('reverseManage', {
         } else {
           return errorMessage(detailRes.message)
         }
-        return successMessage(detailRes.data)
+        return successMessage(bookTaskDetail)
       } catch (error) {
         return errorMessage("获取小说任务详细数据失败，失败信息如下： " + error.toString())
       }