Update app.py

app.py

@@ -3,57 +3,44 @@ import gradio as gr
 import spaces
 from huggingface_hub import InferenceClient
 
-# 1. 自动读取你在 Settings 里面设置的保险箱密钥
+# 1. 读取密钥
 hf_token = os.getenv("HF_TOKEN")
-
-# 2. 初始化客户端，指向 DeepSeek-R1-32B 满血版
-# 如果没有设置 HF_TOKEN，这里会报错，所以我们在下面加了异常处理
 client = InferenceClient("deepseek-ai/DeepSeek-R1-Distill-Qwen-32B", token=hf_token)
 
-# 3. 定义核心推理逻辑
 @spaces.GPU
 def chat_fn(message, history):
-    # 构建对话历史上下文
     messages = []
     for turn in history:
-        # 兼容不同版本的 Gradio 历史记录格式
         if isinstance(turn, dict):
             messages.append({"role": turn["role"], "content": turn["content"]})
         else:
             messages.append({"role": "user", "content": turn[0]})
             messages.append({"role": "assistant", "content": turn[1]})
-            
     messages.append({"role": "user", "content": message})
     
     response = ""
-    # 开启流式输出 (Stream)，让字一个一个跳出来
     try:
-        for msg in client.chat_completion(
-            messages,
-            max_tokens=4096, # 调大 token，支持长文本专利分析
-            stream=True,
-        ):
-            token = msg.choices[0].delta.content
-            if token:
-                response += token
-                yield response
+        # 这里是核心优化：增加对异常块的过滤
+        for msg in client.chat_completion(messages, max_tokens=4096, stream=True):
+            # 只有当返回结构完整且有内容时才处理
+            if len(msg.choices) > 0:
+                token = msg.choices[0].delta.content
+                if token:
+                    response += token
+                    yield response
     except Exception as e:
-        yield f"⚠️ OpenIPOS 算力调度异常: {str(e)}。请确保已在 Settings 中设置 HF_TOKEN。"
+        yield f"⚠️ OpenIPOS 调度提醒: {str(e)}"
 
-# 4. 构建 OpenIPOS 品牌演示界面
+# UI 部分保持你的 OpenIPOS 品牌风格
 with gr.Blocks(fill_height=True, theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🚀 OpenIPOS Global 算力节点")
     gr.Markdown("### ⚖️ DeepSeek-R1 32B 法律级满血演示站")
-    gr.Markdown("本节点由 **OpenIPOS** 部署于全球高算力数据中心，支持完整 R1 思考链路，专为专利分析与法律逻辑设计。")
     
-    # 核心聊天界面
     gr.ChatInterface(
         chat_fn,
-        examples=["请分析这篇专利要求书的逻辑漏洞...", "对比香港与内地的发明专利申请流程", "帮我写一段专利侵权判定的法律意见书"],
+        examples=["请分析这篇专利要求书的逻辑漏洞...", "对比香港与内地的发明专利申请流程"],
     )
-
     gr.Markdown("---")
-    gr.Markdown("© 2026 OpenIPOS Global Limited. All Rights Reserved. 算力驱动：Hugging Face ZeroGPU")
+    gr.Markdown("© 2026 OpenIPOS Global Limited. 算力驱动：Hugging Face ZeroGPU")
 
-# 5. 启动程序
 demo.launch()