update

app.py

@@ -20,7 +20,7 @@ import json
 import pytesseract
 import openpyxl
 import fitz  # PyMuPDF
-from zhipuai import ZhipuAI
+from zai import ZhipuAiClient
 from PIL import Image
 from mimetypes import guess_type
 from openpyxl.styles import numbers
@@ -37,7 +37,7 @@ import pytesseract
 import os
 
 # 硬编码 Hugging Face 环境的 Tesseract 路径
-pytesseract.pytesseract.tesseract_cmd = r'/usr/bin/tesseract'
+pytesseract.pytesseract.tesseract_cmd = r'/opt/homebrew/bin/tesseract'
 
 # 验证路径是否存在（调试用）
 if not os.path.exists(pytesseract.pytesseract.tesseract_cmd):
@@ -75,7 +75,7 @@ class PDFImageToMarkdown:
         prompt_txt = "Precisely identify the content within an image, specifically extracting all information and table content from an invoice. Convert the extracted information into Markdown format, and ensure the output is in Chinese."
 
         response = self.client.chat.completions.create(
-            model="glm-4v-plus-0111",
+            model="glm-4.5v",
             temperature=0,
             messages=[
                 {
@@ -170,7 +170,6 @@ class PDFImageToMarkdown:
 class MarkdownToJSON:
     DEFAULT_PROMPT = """
     请将以下 Markdown 格式的发票内容转换为 JSON 格式。要求如下：
-
     1. **提取以下字段**：
        - 发票代码（invoice_code）
        - 开票日期（invoice_date）
@@ -190,7 +189,6 @@ class MarkdownToJSON:
        - 价税合计（total_including_tax）：包含大写金额（capitalized）和小写金额（numeric）
        - 备注（remarks）：以列表形式存储
        - 开票人（issuer）
-
     2. **JSON 格式要求**：
        - 字段名称必须与上述要求一致。
        - 金额和税额字段的值应为字符串类型。
@@ -213,7 +211,7 @@ class MarkdownToJSON:
 
         try:
             response = self.client.chat.completions.create(
-                model="glm-4-plus",
+                model="glm-4.6",
                 temperature=0,
                 top_p=0.1,
                 max_tokens=4095,
@@ -447,10 +445,10 @@ async def process_invoices(api_key, files, progress=gr.Progress()):
         raise gr.Error("请输入智谱 API Key")
     
     try:
-        client = ZhipuAI(api_key=api_key)
+        client = ZhipuAiClient(api_key=api_key)
         # 测试API连接
         test_response = client.chat.completions.create(
-            model="glm-4",
+            model="glm-4.6",
             messages=[{"role": "user", "content": "test"}],
             temperature=0
         )
@@ -609,4 +607,5 @@ with gr.Blocks(title="发票识别系统") as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()
+