feat: add an option to display the reason for polishing or not

openai-translator · Mar 3, 2023 · 0d35445 · 0d35445
1 parent 584d48d
commit 0d35445
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 14 deletions.
diff --git a/src/info.json b/src/info.json
@@ -39,6 +39,22 @@
                     "value": "text-davinci-002"
                 }
             ]
+        },
+        {
+            "identifier": "polishing_mode",
+            "type": "menu",
+            "title": "润色模式",
+            "defaultValue": "simplicity",
+            "menuValues": [
+                {
+                    "title": "简洁（只输出润色后的文本）",
+                    "value": "simplicity"
+                },
+                {
+                    "title": "详尽（附加解释修改原因）",
+                    "value": "detailed"
+                },
+            ]
         }
     ]
 }
diff --git a/src/main.js b/src/main.js
@@ -12,23 +12,44 @@ function translate(query, completion) {
         "Content-Type": "application/json",
         Authorization: `Bearer ${api_key}`,
     };
-    let prompt = "Revise the following sentence to make it more clear, concise, and coherent. Please note that you need to list the changes and briefly explain why.";
+    const detailedPolishingMode = $option.polishing_mode === "detailed";
+    let prompt =
+        "Revise the following sentences to make them more clear, concise, and coherent";
+    if (detailedPolishingMode) {
+        prompt = `${prompt}. Please note that you need to list the changes and briefly explain why`;
+    }
     switch (query.detectFrom) {
         case "zh-Hant":
         case "zh-Hans":
-            prompt = "请润色一下这句话";
+            prompt = "润色此句";
+            if (detailedPolishingMode) {
+                prompt = `${prompt}。请注意要列出更改以及简要解释一下为什么这么修改`;
+            }
             break;
         case "ja":
-            prompt = "この文章を装飾してください";
+            prompt = "この文章を装飾する";
+            if (detailedPolishingMode) {
+                prompt = `${prompt}。変更点をリストアップし、なぜそのように変更したかを簡単に説明することに注意してください`;
+            }
             break;
         case "ru":
-            prompt = "Пожалуйста, приукрасьте это предложение";
+            prompt =
+                "Переформулируйте следующие предложения, чтобы они стали более ясными, краткими и связными";
+            if (detailedPolishingMode) {
+                prompt = `${prompt}. Пожалуйста, обратите внимание на необходимость перечисления изменений и краткого объяснения причин таких изменений`;
+            }
             break;
         case "wyw":
-            prompt = "请润色一下这句古文";
+            prompt = "润色此句古文";
+            if (detailedPolishingMode) {
+                prompt = `${prompt}。请注意要列出更改以及简要解释一下为什么这么修改`;
+            }
             break;
         case "yue":
-            prompt = "请润色一下这句粤语";
+            prompt = "潤色呢句粵語";
+            if (detailedPolishingMode) {
+                prompt = `${prompt}。記住要列出修改嘅內容同簡單解釋下點解要做呢啲更改`;
+            }
             break;
     }
     const body = {
@@ -43,10 +64,10 @@ function translate(query, completion) {
     if (isChatGPTModel) {
         body.messages = [
             { role: "system", content: prompt },
-            { role: "user", content: `"${query.text}"` },
+            { role: "user", content: query.text },
         ];
     } else {
-        body.prompt = `${prompt}:\n\n"${query.text}" =>`;
+        body.prompt = `${prompt}:\n\n${query.text} =>`;
     }
     (async () => {
         const resp = await $http.request({
@@ -89,12 +110,6 @@ function translate(query, completion) {
             } else {
                 targetTxt = choices[0].text.trim();
             }
-            if (targetTxt.startsWith('"') || targetTxt.startsWith("「")) {
-                targetTxt = targetTxt.slice(1);
-            }
-            if (targetTxt.endsWith('"') || targetTxt.endsWith("」")) {
-                targetTxt = targetTxt.slice(0, -1);
-            }
             completion({
                 result: {
                     from: query.detectFrom,