Update app/main.py

app/main.py

@@ -20,10 +20,11 @@ BAIDU_SECRET_KEY = os.getenv("BAIDU_SECRET_KEY", "")
 
 # —— 指定代理站 & API Key —— #
 openai.api_key  = "Magic-123456"
-openai.base_url = "https://freeapi.oaiopen.com"
+openai.base_url = "https://freeapi.oaiopen.com/v1/"
+openai.default_headers = {"x-foo": "true"}
 
 # —— 初始化 v1 客户端 —— #
-client = OpenAI(api_key=openai.api_key)
+#client = OpenAI(api_key=openai.api_key)
 # 同步 base_url
 #client.api_base = openai.api_base
 
@@ -71,7 +72,7 @@ def process(image_pil):
     fmt_prompt = (
         "请帮我整理下面的英语作文文本格式，只整理英文正文部分（忽略英文正文外所有部分），保证原汁原味（明显错误空格换行、乱码、非常用字符比如☰需要改正除外），出现的拼写错误也不要帮助改正：\n\n"+ essay_text
     )
-    fm_resp = client.chat.completions.create(
+    fm_resp = openai.chat.completions.create(
         model="gpt-4o",
         messages=[{"role": "user", "content": fmt_prompt}]
     )
@@ -82,7 +83,7 @@ def process(image_pil):
         "请帮我把下面的英语作文的语法错误改正，输出改正后的文章，"
         "原文错误用()括起来，修改部分用[]括起来：\n\n" + revised
     )
-    cm_resp = client.chat.completions.create(
+    cm_resp = openai.chat.completions.create(
         model="gpt-4o",
         messages=[{"role": "user", "content": corr_prompt}]
     )
@@ -92,7 +93,7 @@ def process(image_pil):
     review_prompt = (
         "下面是一份已经批改过的英语作文，请根据批注给出逐条批改意见：\n\n" + corrected
     )
-    rm_resp = client.chat.completions.create(
+    rm_resp = openai.chat.completions.create(
         model="gpt-4o",
         messages=[{"role": "user", "content": review_prompt}]
     )
@@ -102,7 +103,7 @@ def process(image_pil):
     rate_prompt = (
         "请按照 IELTS/CEFR 写作评价体系，从语言通顺度、连贯度、词汇与语法三维度打分并给出原因：\n\n" + revised
     )
-    rr_resp = client.chat.completions.create(
+    rr_resp = openai.chat.completions.create(
         model="gpt-4o",
         messages=[{"role": "user", "content": rate_prompt}]
     )
@@ -112,7 +113,7 @@ def process(image_pil):
     rewrite_prompt = (
         "请使用优秀表达重写下面这篇作文，加粗可供学习的部分：\n\n" + revised
     )
-    wm_resp = client.chat.completions.create(
+    wm_resp = openai.chat.completions.create(
         model="gpt-4o",
         messages=[{"role": "user", "content": rewrite_prompt}]
     )