Răsfoiți Sursa

feat: add ai_trans.py with pdfzh_translator integration for Japanese to Chinese translation

Your Name (aider) 1 an în urmă
părinte
comite
c7e57baa37
1 a modificat fișierele cu 34 adăugiri și 0 ștergeri
  1. 34 0
      ai_trans.py

+ 34 - 0
ai_trans.py

@@ -0,0 +1,34 @@
+import os
+from pdfzh_translator import OpenAITranslator
+
+# Initialize translator
+translator = OpenAITranslator(service="openai", lang_out="zh-CN", lang_in="ja", model="gpt-3.5-turbo")
+
+def translate_sentences(sentences, target_language='zh'):
+    """
+    翻译句子列表
+    :param sentences: 要翻译的句子列表
+    :param target_language: 目标语言,默认为中文
+    :return: 包含翻译结果的字典
+    """
+    try:
+        # 将句子列表连接成一个字符串进行批量翻译
+        combined_text = "\n".join(sentences)
+        
+        # 使用pdfzh_translator进行翻译
+        translated_text = translator.translate(combined_text)
+        
+        # 将翻译结果分割回单独的句子
+        translations = translated_text.split("\n")
+        
+        return {
+            "translations": translations,
+            "status": "success",
+            "message": f"Translated {len(sentences)} sentences"
+        }
+    except Exception as e:
+        return {
+            "translations": [],
+            "status": "error",
+            "message": str(e)
+        }