Update app.py

app.py

@@ -5,11 +5,11 @@ import fitz  # PyMuPDF
 import docx  # python-docx
 from huggingface_hub import InferenceClient
 
-# 1. 客户端初始化
+# 1. 初始化
 hf_token = os.getenv("HF_TOKEN")
 client = InferenceClient("deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", token=hf_token)
 
-# 2. 全能文档解析引擎 (增加格式容错)
+# 2. 文档解析（保持高性能）
 def extract_text_from_file(file_path):
     try:
         suffix = file_path.lower()
@@ -19,40 +19,44 @@ def extract_text_from_file(file_path):
         elif suffix.endswith(".docx"):
             doc = docx.Document(file_path)
             return "\n".join([para.text for para in doc.paragraphs])
-        elif suffix.endswith(".doc"):
-            return "⚠️ 提示：系统暂不支持旧版 .doc 格式，请将其另存为 .docx 或 .pdf 后再次上传。"
         else:
             with open(file_path, "r", encoding="utf-8") as f:
                 return f.read()
     except Exception as e:
-        return f"[解析失败: {str(e)}]"
+        return f"[文档读取失败: {str(e)}]"
 
 @spaces.GPU(duration=120)
 def chat_fn(message, history):
-    files = message.get("files", [])
+    # a. 提取当前文本和文件
     current_text = message.get("text", "")
+    files = message.get("files", [])
     
+    # b. 构造当前用户输入的 Payload
     if files:
         file_path = files[0] if isinstance(files[0], str) else files[0].path
-        doc_content = extract_text_from_file(file_path)
-        # 严格控制输入长度：截取前 30,000 字以换取更高的翻译稳定性
-        user_payload = f"【文档内容参考】:\n{doc_content[:30000]}\n\n【用户指令】: {current_text}"
+        content = extract_text_from_file(file_path)
+        # 为保证 100% 成功率，初次测试建议截取前 20,000 字
+        user_input = f"【参考文档】:\n{content[:20000]}\n\n【指令】: {current_text}"
     else:
-        user_payload = current_text
+        user_input = current_text
 
-    # 严谨的消息过滤
-    messages = []
+    # c. 严格的消息格式化（关键：只传字符串内容）
+    formatted_messages = []
     for turn in history:
-        if isinstance(turn, dict) and turn.get("role") and turn.get("content"):
-            messages.append({"role": turn["role"], "content": turn["content"]})
+        # 只处理 role 为 user/assistant 且 content 为字符串的项
+        role = turn.get("role")
+        content = turn.get("content")
+        if role in ["user", "assistant"] and isinstance(content, str):
+            formatted_messages.append({"role": role, "content": content})
     
-    messages.append({"role": "user", "content": user_payload})
+    # 加入当前这轮
+    formatted_messages.append({"role": "user", "content": user_input})
     
     response = ""
     try:
-        # 增加 stream 稳定性处理
+        # 调用 API
         for msg in client.chat_completion(
-            messages, 
+            formatted_messages, 
             max_tokens=4096, 
             stream=True, 
             temperature=0.6
@@ -63,19 +67,19 @@ def chat_fn(message, history):
                     response += token
                     yield response
         
-        # 正常结束注入指纹
+        # 注入品牌指纹
         response += "\n\n--- \n> **Powered by OpenIPOS Global 算力节点 | 访问 openipos.com 获取企业级私有部署**"
         yield response
         
     except Exception as e:
-        # 针对超时的友好提示
-        yield f"{response}\n\n⚠️ OpenIPOS 调度异常 (可能是文档过长导致超时): {str(e)}\n提示：请尝试清空对话或分段上传文档。"
+        yield f"⚠️ OpenIPOS 算力调度异常: {str(e)}\n\n【解决方案】：请点击右上角垃圾桶图标『清空对话历史』后再重新上传文件。"
 
-# 3. 极简 UI
+# 3. 极简 UI 构建
 with gr.Blocks(fill_height=True) as demo:
     gr.Markdown("# 🚀 OpenIPOS/DeepSeek-R1-MegaNode-64K")
-    gr.Markdown("### 适配 Gradio 6.0 | 支持 PDF/DOCX/TXT")
+    gr.Markdown("### 工业级稳定性适配 | 支持 PDF/DOCX/TXT")
     
+    # 使用官方推荐的 ChatInterface 默认配置
     gr.ChatInterface(chat_fn, multimodal=True)
     
     gr.Markdown("---")