From 9c735de71edc050897f655f160e64b045ed70168 Mon Sep 17 00:00:00 2001
From: zhiyu1998 <renzhiyu0416@qq.com>
Date: Sun, 9 Apr 2023 20:54:23 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8=20feat:=20=E5=A2=9E=E5=BC=BAOCR?=
 =?UTF-8?q?=E7=A8=8B=E5=BA=8F=E8=A7=A3=E9=87=8A/=E4=BC=98=E5=8C=96/?=
 =?UTF-8?q?=E8=A7=A3=E5=86=B3=E8=83=BD=E5=8A=9B=20&=20=F0=9F=8E=88=20perf:?=
 =?UTF-8?q?=20=E5=A2=9E=E5=BC=BAOCR=E7=9A=84=E5=81=A5=E5=A3=AE=E6=80=A7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/tools.js | 35 ++++++++++++++++++++++++++++++-----
 1 file changed, 30 insertions(+), 5 deletions(-)

diff --git a/apps/tools.js b/apps/tools.js
index 9ea31ce..5a74b82 100644
--- a/apps/tools.js
+++ b/apps/tools.js
@@ -34,7 +34,7 @@ export class tools extends plugin {
                     fnc: "trans",
                 },
                 {
-                    reg: "^#ocr$",
+                    reg: "^#(ocr|OCR)(解决|解释|优化)?$",
                     fnc: "ocr2anything",
                 },
                 {
@@ -156,19 +156,25 @@ export class tools extends plugin {
     async ocr2anythingContext() {
         // 当前消息
         const curMsg = this.e;
+        // 安全性检测
+        if (curMsg.img) {
+            curMsg.reply("请发送图片！")
+            return;
+        }
+        // 上一个消息
+        const preMsg = this.getContext().ocr2anythingContext;
         try {
             const defaultPath = `${this.defaultPath}${this.e.group_id || this.e.user_id}`
             await this.downloadImg(curMsg.img, defaultPath, "temp.jpg").then(async _ => {
                 const ocrRst = await Bot.imageOcr(`${defaultPath}/temp.jpg`);
                 const wordList = ocrRst.wordslist;
-                let prompt = wordList.map(item => item.words);
+                let prompt = wordList.map(item => item.words).join(" ");
                 if (this.openaiAccessToken) {
-                    prompt = "Summarize the key points of this article in Chinese and in a list of points. Choose an appropriate emoji for each bullet point. Each bullet point format is [emoji] - [text]." + prompt.join(" ")
+                    const func = preMsg.msg.replace("#ocr", "").trim();
+                    prompt = await this.ocrStrategy(func) + prompt;
                     const response = await this.chatGptClient.sendMessage(prompt);
                     // 暂时不设计上下文
                     prompt = response.response;
-                } else {
-                    prompt = prompt.join("\n")
                 }
                 curMsg.reply(prompt);
             });
@@ -180,6 +186,24 @@ export class tools extends plugin {
         this.finish("ocr2anythingContext")
     }
 
+    /**
+     * 图像识别调教
+     * @Link{ocr2anything} 的调教方法
+     * @return Promise{string}
+     **/
+    async ocrStrategy(keyword) {
+        switch (keyword) {
+            case "解决":
+               return "You are now a programmer, the following is an error, please tell me how to solve this error in Chinese. "
+            case "解释":
+                return "You are now a programming language expert, please tell me what the following code is doing in Chinese. "
+            case "优化":
+                return "You are now a Clean Code expert, I have the following code, please rewrite it in a cleaner and more concise way to make it easier for my colleagues to maintain the code. Also, delineation of why you want to refactor like this, so that I can add the description of the refactoring method to the Pull Request. Attach code "
+            default:
+                return "Summarize the key points of this article in Chinese and in a list of points. Choose an appropriate emoji for each bullet point. Each bullet point format is [emoji] - [text]."
+        }
+    }
+
     // 抖音解析
     async douyin(e) {
         const urlRex = /(http:|https:)\/\/v.douyin.com\/[A-Za-z\d._?%&+\-=\/#]*/g;
@@ -1079,6 +1103,7 @@ export class tools extends plugin {
             fileName = img.split("/").pop();
         }
         const filepath = `${dir}/${fileName}`;
+        await mkdirIfNotExists(dir)
         const writer = fs.createWriteStream(filepath);
         const axiosConfig = {
             headers: {