excel_tool/ai_trans.py

import os
from dotenv import load_dotenv
from mylib.pdfzh_translator import OpenAITranslator

# Load environment variables
load_dotenv()

# Initialize translator with environment variables
translator = OpenAITranslator(
    service="openai",
    lang_out="zh-CN",
    lang_in="ja",
    model=os.getenv('LLM_MODEL', 'gpt-3.5-turbo')  # Use env var or fallback to default
)

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)
        }