Update app/main.py

app/main.py

@@ -17,15 +17,14 @@ from openai import OpenAI
 # —— 环境变量读取 —— #
 BAIDU_API_KEY    = os.getenv("BAIDU_API_KEY", "")
 BAIDU_SECRET_KEY = os.getenv("BAIDU_SECRET_KEY", "")
-OPENAI_API_KEY   = os.getenv("OPENAI_API_KEY", "")
 
-# —— 指定代理站 Base URL —— #
-openai.api_key  = OPENAI_API_KEY
-openai.api_base = "https://free.v36.cm/v1/"
+# —— 指定代理站 & API Key —— #
+openai.api_key  = "Magic-123456"
+openai.api_base = "https://freeapi.oaiopen.com/v1"
 
 # —— 初始化 v1 客户端 —— #
-client = OpenAI(api_key=OPENAI_API_KEY)
-# client.api_base 属性需要手动同步
+client = OpenAI(api_key=openai.api_key)
+# 同步 base_url
 client.api_base = openai.api_base
 
 # —— Baidu OCR —— #
@@ -43,14 +42,11 @@ def get_access_token(api_key, secret_key):
 
 def ocr_image(image_bytes: bytes, token: str):
     img_b64     = base64.b64encode(image_bytes).decode()
-    img_encoded = quote_plus(img_b64)  # URL 编码
+    img_encoded = quote_plus(img_b64)
     resp = requests.post(
         f"https://aip.baidubce.com/rest/2.0/ocr/v1/handwriting?access_token={token}",
         headers={"Content-Type": "application/x-www-form-urlencoded"},
-        data={
-            "image": img_encoded,
-            "language_type": "ENG"
-        }
+        data={"image": img_encoded, "language_type": "ENG"}
     )
     resp.raise_for_status()
     return resp.json().get("words_result", [])
@@ -63,17 +59,15 @@ def highlight_brackets(text: str) -> str:
 
 # —— 主处理函数 —— #
 def process(image_pil):
-    # —— PIL 转 bytes —— #
     buf = BytesIO()
     image_pil.save(buf, format="PNG")
     image_bytes = buf.getvalue()
 
-    # —— OCR 提取原文 —— #
     token      = get_access_token(BAIDU_API_KEY, BAIDU_SECRET_KEY)
     words      = ocr_image(image_bytes, token)
     essay_text = "\n".join([w["words"] for w in words])
 
-    # —— 1. 格式化原文 —— #
+    # 1. 格式化原文
     fmt_prompt = (
         "请帮我整理下面的英语作文文本格式，只整理英文正文部分，"
         "保证原汁原味（明显错误空格换行、乱码、非常用字符改正），"
@@ -85,7 +79,7 @@ def process(image_pil):
     )
     revised = fm_resp.choices[0].message.content
 
-    # —— 2. 批改 —— #
+    # 2. 批改
     corr_prompt = (
         "请帮我把下面的英语作文的语法错误改正，输出改正后的文章，"
         "原文错误用()括起来，修改部分用[]括起来：\n\n" + revised
@@ -96,7 +90,7 @@ def process(image_pil):
     )
     corrected = cm_resp.choices[0].message.content
 
-    # —— 3. 批改意见 —— #
+    # 3. 批改意见
     review_prompt = (
         "下面是一份已经批改过的英语作文，请根据批注给出逐条批改意见：\n\n" + corrected
     )
@@ -106,7 +100,7 @@ def process(image_pil):
     )
     review = rm_resp.choices[0].message.content
 
-    # —— 4. 评分 —— #
+    # 4. 评分
     rate_prompt = (
         "请按照 IELTS/CEFR 写作评价体系，从语言通顺度、连贯度、词汇与语法三维度打分并给出原因：\n\n" + revised
     )
@@ -116,7 +110,7 @@ def process(image_pil):
     )
     rating = rr_resp.choices[0].message.content
 
-    # —— 5. 优秀范文 —— #
+    # 5. 优秀范文
     rewrite_prompt = (
         "请使用优秀表达重写下面这篇作文，加粗可供学习的部分：\n\n" + revised
     )
@@ -126,7 +120,7 @@ def process(image_pil):
     )
     perfect = wm_resp.choices[0].message.content
 
-    # —— 6. 渲染 HTML —— #
+    # 6. 渲染 HTML
     code     = f"{random.randint(0,9999):04}-{datetime.now().strftime('%Y%m%d%H%M%S')}"
     tpl_path = os.path.join("app", "templates", "base.html")
     with open(tpl_path, encoding="utf-8") as f:
@@ -141,7 +135,7 @@ def process(image_pil):
     )
     full_html = tpl.render(code=code, content=html_content)
 
-    # —— 7. 写文件并返回 —— #
+    # 7. 写文件并返回
     output_dir = os.path.join("app", "output")
     os.makedirs(output_dir, exist_ok=True)
     html_path = os.path.join(output_dir, f"{code}.html")
@@ -149,11 +143,7 @@ def process(image_pil):
 
     with open(html_path, "w", encoding="utf-8") as f:
         f.write(full_html)
-
-    pdfkit.from_string(
-        full_html, pdf_path,
-        options={"enable-local-file-access": ""}
-    )
+    pdfkit.from_string(full_html, pdf_path, options={"enable-local-file-access": ""})
 
     return full_html, html_path, pdf_path