gpt-po 1.0.3 → 1.0.5

package/README.md

@@ -1,5 +1,8 @@
 # PO File Translation Tool for ChatGPT
 
+[![NPM version](https://img.shields.io/npm/v/gpt-po.svg)](https://npmjs.org/package/gpt-po)
+[![Downloads](https://img.shields.io/npm/dm/gpt-po.svg)](https://npmjs.org/package/gpt-po)
+
 Translation tool for gettext (po) files that supports custom system prompts and user dictionaries. It also supports translating specified po files to a designated target language and updating po files based on pot files.
 
 Read in other languages: English | [简体中文](./README_zh-CN.md)
@@ -18,6 +21,7 @@ Set `OPENAI_API_KEY` before using this tool.
 
 - `gpt-po sync --po <file> --pot <file>` Update the po file based on the pot file, while preserving the original translations.
 - `gpt-po --po <file>` Translate specified po files to a designated target language. By default, the target language is Simplified Chinese.
+- `gpt-po --dir .` Translate all po files in current directory to a designated target language.
 - `gpt-po userdict` Modify or view user dictionaries
 - `gpt-po systemprompt` Modify or view system prompts, if you are not sure how to use it, you can leave it alone
 
@@ -49,6 +53,9 @@ Options:
   --model <model>        openai model (choices: "gpt-4", "gpt-4-0314", "gpt-4-32k", "gpt-4-32k-0314", "gpt-3.5-turbo", "gpt-3.5-turbo-0301",
                          default: "gpt-3.5-turbo")
   --po <file>            po file path
+  --dir <dir>            po file directory
+  -src, --source <lang>  source language (default: "english")
+  --verbose              show verbose log
   -l, --lang <lang>      target language (default: "simplified chinese")
   -o, --output <file>    output file path, overwirte po file by default
   -h, --help             display help for command

package/README_zh-CN.md

@@ -1,8 +1,11 @@
 # PO文件CHATGPT翻译工具
 
+[![NPM version](https://img.shields.io/npm/v/gpt-po.svg)](https://npmjs.org/package/gpt-po)
+[![Downloads](https://img.shields.io/npm/dm/gpt-po.svg)](https://npmjs.org/package/gpt-po)
+
 gettext(po)文件翻译工具，支持自定义系统提示词和用户字典，支持翻译指定的po文件到指定的目标语言，支持根据pot文件更新po文件。
 
-使用其他语言阅读：[English](./Readme.md) | 简体中文
+使用其他语言阅读：[English](./README.md) | 简体中文
 
 ## 安装
 
@@ -18,6 +21,7 @@ npm install gpt-po
 
 - `gpt-po sync --po <file> --pot <file>` 根据pot文件更新po文件，保留原有翻译
 - `gpt-po --po <file>` 翻译指定的po文件到指定的目标语言，默认目标语言是简体中文
+- `gpt-po --dir .` 翻译当前目录下的所有po文件到指定的目标语言，默认目标语言是简体中文
 - `gpt-po userdict` 修改或查看用户字典
 - `gpt-po systemprompt` 修改或查看系统提示词，如果你不确定如何使用，可以不用修改
 
@@ -49,6 +53,9 @@ Options:
   --model <model>        openai model (choices: "gpt-4", "gpt-4-0314", "gpt-4-32k", "gpt-4-32k-0314", "gpt-3.5-turbo", "gpt-3.5-turbo-0301",
                          default: "gpt-3.5-turbo")
   --po <file>            po file path
+  --dir <dir>            po files directory
+  -src, --source <lang>  source language (default: "english")
+  --verbose              show verbose log
   -l, --lang <lang>      target language (default: "simplified chinese")
   -o, --output <file>    output file path, overwirte po file by default
   -h, --help             display help for command

package/lib/package.json

@@ -1,6 +1,6 @@
 {
     "name": "gpt-po",
-    "version": "1.0.3",
+    "version": "1.0.5",
     "description": "command tool for translate po files by gpt",
     "main": "lib/src/index.js",
     "bin": {

package/lib/src/index.js

@@ -34,18 +34,29 @@ program
     "gpt-3.5-turbo",
     "gpt-3.5-turbo-0301",
 ]))
-    .requiredOption("--po <file>", "po file path")
-    .option("-src, --source <lang>", "source language", "english")
-    .requiredOption("-l, --lang <lang>", "target language", "simplified chinese")
+    .addOption(new commander_1.Option("--po <file>", "po file path").conflicts("dir"))
+    .addOption(new commander_1.Option("--dir <dir>", "po file directory").conflicts("po"))
+    .option("-src, --source <lang>", "source language", "English")
+    .option("-l, --lang <lang>", "target language", "Simplified Chinese")
+    .option("--verbose", "print verbose log")
     .option("-o, --output <file>", "output file path, overwirte po file by default")
-    .action(({ key, host, model, po, source, lang, output }) => __awaiter(void 0, void 0, void 0, function* () {
+    .action(({ key, host, model, po, dir, source, lang, verbose, output }) => __awaiter(void 0, void 0, void 0, function* () {
     if (host) {
         process.env.OPENAI_API_HOST = host;
     }
     if (key) {
         process.env.OPENAI_API_KEY = key;
     }
-    yield (0, translate_1.translatePo)(model, po, source, lang, output);
+    if (po) {
+        yield (0, translate_1.translatePo)(model, po, source, lang, verbose, output);
+    }
+    else if (dir) {
+        yield (0, translate_1.translatePoDir)(model, dir, source, lang, verbose, output);
+    }
+    else {
+        console.error("po file or directory is required");
+        process.exit(1);
+    }
 }));
 program
     .command("sync")

package/lib/src/translate.d.ts

@@ -1,4 +1,5 @@
 import { OpenAIApi } from "openai";
 export declare function init(force?: boolean): OpenAIApi;
 export declare function translate(text: string, src: string, lang: string, model?: string): Promise<import("axios").AxiosResponse<import("openai").CreateChatCompletionResponse, any>>;
-export declare function translatePo(model: string | undefined, po: string, source: string, lang: string, output: string): Promise<void>;
+export declare function translatePo(model: string | undefined, po: string, source: string, lang: string, verbose: boolean, output: string): Promise<void>;
+export declare function translatePoDir(model: string | undefined, dir: string, source: string, lang: string, verbose: boolean, output: string): Promise<void>;

package/lib/src/translate.js

@@ -9,7 +9,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
     });
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.translatePo = exports.translate = exports.init = void 0;
+exports.translatePoDir = exports.translatePo = exports.translate = exports.init = void 0;
 const fs = require("fs");
 const openai_1 = require("openai");
 const os_1 = require("os");
@@ -54,13 +54,12 @@ function translate(text, src, lang, model = "gpt-3.5-turbo") {
         .flat();
     return openai.createChatCompletion({
         model,
-        temperature: 0.5,
-        n: 1,
+        temperature: 0.1,
         messages: [
             { role: "system", content: _systemprompt },
             {
                 role: "user",
-                content: `Translate incoming ${src} content into ${lang}, only the translated content can be returned, don't ask question.`,
+                content: `Translate the ${src} content I will post later into ${lang}, and keep the untranslated parts such as symbols in the result.`,
             },
             {
                 role: "assistant",
@@ -75,7 +74,7 @@ function translate(text, src, lang, model = "gpt-3.5-turbo") {
     });
 }
 exports.translate = translate;
-function translatePo(model = "gpt-3.5-turbo", po, source, lang, output) {
+function translatePo(model = "gpt-3.5-turbo", po, source, lang, verbose, output) {
     var _a;
     return __awaiter(this, void 0, void 0, function* () {
         const potrans = yield (0, utils_1.parsePo)(po);
@@ -88,9 +87,10 @@ function translatePo(model = "gpt-3.5-turbo", po, source, lang, output) {
             }
         }
         if (list.length == 0) {
-            console.log("nothing needs translate");
+            console.log("done.");
             return;
         }
+        potrans.headers["Last-Translator"] = `gpt-po v${pkg.version}`;
         let err429 = false;
         let modified = false;
         for (let i = 0; i < list.length; i++) {
@@ -105,7 +105,12 @@ function translatePo(model = "gpt-3.5-turbo", po, source, lang, output) {
                 const res = yield translate(trans.msgid, source, lang, model);
                 trans.msgstr[0] = ((_a = res.data.choices[0].message) === null || _a === void 0 ? void 0 : _a.content) || trans.msgstr[0];
                 modified = true;
+                if (verbose) {
+                    console.log(trans.msgid);
+                    console.log(trans.msgstr[0]);
+                }
                 (0, utils_1.printProgress)(i + 1, list.length);
+                yield (0, utils_1.compilePo)(potrans, output || po);
             }
             catch (error) {
                 if (error.response) {
@@ -124,11 +129,21 @@ function translatePo(model = "gpt-3.5-turbo", po, source, lang, output) {
                 }
             }
         }
-        if (modified) {
-            potrans.headers["Last-Translator"] = `gpt-po v${pkg.version}`;
-            yield (0, utils_1.compilePo)(potrans, output || po);
-        }
+        console.log("done.");
     });
 }
 exports.translatePo = translatePo;
+function translatePoDir(model = "gpt-3.5-turbo", dir, source, lang, verbose, output) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const files = fs.readdirSync(dir);
+        for (const file of files) {
+            if (file.endsWith(".po")) {
+                const po = (0, path_1.join)(dir, file);
+                console.log(`translating ${po}`);
+                yield translatePo(model, po, source, lang, verbose, output);
+            }
+        }
+    });
+}
+exports.translatePoDir = translatePoDir;
 //# sourceMappingURL=translate.js.map

package/lib/src/utils.js

@@ -36,7 +36,7 @@ function parsePo(poFile, defaultCharset) {
 }
 exports.parsePo = parsePo;
 function compilePo(data, poFile) {
-    const buffer = gettext_parser_1.po.compile(data);
+    const buffer = gettext_parser_1.po.compile(data, { foldLength: 120 });
     return new Promise((resolve, reject) => {
         fs.writeFile(poFile, buffer, (err) => {
             if (err)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "gpt-po",
-  "version": "1.0.3",
+  "version": "1.0.5",
   "description": "command tool for translate po files by gpt",
   "main": "lib/src/index.js",
   "bin": {