From ebd25b9cdb939f899a8119ace535d4b4e77eec4b Mon Sep 17 00:00:00 2001
From: Begild <yucang_bao@126.com>
Date: Mon, 4 Nov 2024 22:36:01 +0800
Subject: [PATCH] =?UTF-8?q?=E8=A7=A3=E5=86=B3=E9=80=80=E5=87=BA=E6=8A=A5?=
 =?UTF-8?q?=E9=94=99event=20loop=20closed=E7=9A=84=E9=94=99=E8=AF=AF=20?=
 =?UTF-8?q?=E5=B0=9D=E8=AF=95=E4=BA=86=E5=87=A0=E4=B8=AA=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E5=8F=91=E7=8E=B0=E8=BF=98=E6=98=AFgpt-4o-mini=E6=9B=B4?=
 =?UTF-8?q?=E5=A5=BD=EF=BC=8C=E6=9B=B4=E5=87=86=E7=A1=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md          |  8 +++++++-
 excel_translate.py | 12 ++++++++----
 2 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/README.md b/README.md
index 055f4eb..8b0e61b 100755
--- a/README.md
+++ b/README.md
@@ -1,3 +1,9 @@
 # excel_translate
 
-Emma工作需要，将excel中的所有中文转换为英文
\ No newline at end of file
+Emma工作需要，将excel中的所有中文转换为英文
+
+如果出现以下错误，请关闭代理尝试
+```bash
+Error: Unknown scheme for proxy URL URL('socks4://127.0.0.1:1080')
+```
+
diff --git a/excel_translate.py b/excel_translate.py
index 2c58df9..4120f9e 100755
--- a/excel_translate.py
+++ b/excel_translate.py
@@ -3,6 +3,7 @@ import shutil
 from typing import List, Tuple
 import os
 import sys
+import platform
 try:
     import openpyxl
 except ImportError:
@@ -35,6 +36,8 @@ API_URL = "https://api.gpt.ge/v1"
 # exit(0)
 
 class Model:
+    default = "gpt-4o-mini"
+    doubao_lite_32k = "doubao-lite-32k"
     gpt_4o_mini = "gpt-4o-mini"
     text_davinci_002 = "text-davinci-002"
     gpt_4o = "gpt-4o"
@@ -49,7 +52,7 @@ class Model:
 # 注意仅输出翻译后的内容即可，不要保留原文任何内容！！！
 # 请翻译：\n"""
 PROMT = "将如下内容翻译为英文，仅输出翻译后的内容，不输出任何原文:"
-async def chinese2english(text, model=Model.gpt_4o_mini):
+async def chinese2english(text, model=Model.default):
     # print("start translate")
     #Translate the text using OpenAI
     client = AsyncOpenAI(api_key=API_KEY, base_url=API_URL)
@@ -66,13 +69,12 @@ async def chinese2english(text, model=Model.gpt_4o_mini):
     )
     # print("translate done")
     translated_text = response.choices[0].message.content
-    client.close()
     return translated_text
 
-# def chinese2english_stream(text, model=Model.gpt_4o_mini):
+# def chinese2english_stream(text, model=Model.default):
 #     client = OpenAI(api_key=API_KEY, base_url=API_URL)
 #     with client.chat.completions.with_streaming_response.create(
-#         model = Model.gpt_4o_mini,
+#         model = model,
 #         messages = [
 #             {
 #                 "role": "user",
@@ -212,5 +214,7 @@ async def translate_excel_process(input_file_path="input.xlsx", output_file_path
     output_workbook.save(output_file_path)
     pass
 if __name__ == "__main__":
+    if platform.system()=='Windows':
+        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy()) #解决windows下退出报错 event loop is closed
     asyncio.run(main())
     exit(0)
\ No newline at end of file