Update app.py

app.py

@@ -1,16 +1,22 @@
+import os
 import gradio as gr
 import spaces
 from huggingface_hub import InferenceClient
 
-# 初始化客户端
-client = InferenceClient("deepseek-ai/DeepSeek-R1-Distill-Qwen-32B")
+# 1. 自动读取你在 Settings 里面设置的保险箱密钥
+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:
-        # turn 可能是一个字典（新版）或元组（旧版）
+        # 兼容不同版本的 Gradio 历史记录格式
         if isinstance(turn, dict):
             messages.append({"role": turn["role"], "content": turn["content"]})
         else:
@@ -20,11 +26,11 @@ def chat_fn(message, history):
     messages.append({"role": "user", "content": message})
     
     response = ""
-    # 调用流式接口
+    # 开启流式输出 (Stream)，让字一个一个跳出来
     try:
         for msg in client.chat_completion(
             messages,
-            max_tokens=2048,
+            max_tokens=4096, # 调大 token，支持长文本专利分析
             stream=True,
         ):
             token = msg.choices[0].delta.content
@@ -32,17 +38,22 @@ def chat_fn(message, history):
                 response += token
                 yield response
     except Exception as e:
-        yield f"抱歉，算力节点暂时忙碌，请稍后再试。错误信息: {str(e)}"
+        yield f"⚠️ OpenIPOS 算力调度异常: {str(e)}。请确保已在 Settings 中设置 HF_TOKEN。"
 
-# 构建 OpenIPOS 品牌界面
-with gr.Blocks(fill_height=True) as demo:
+# 4. 构建 OpenIPOS 品牌演示界面
+with gr.Blocks(fill_height=True, theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🚀 OpenIPOS Global 算力节点")
-    gr.Markdown("### DeepSeek-R1 32B 满血版演示站")
+    gr.Markdown("### ⚖️ DeepSeek-R1 32B 法律级满血演示站")
+    gr.Markdown("本节点由 **OpenIPOS** 部署于全球高算力数据中心，支持完整 R1 思考链路，专为专利分析与法律逻辑设计。")
     
-    # 使用最简洁的参数配置
+    # 核心聊天界面
     gr.ChatInterface(
         chat_fn,
-        description="本节点由 OpenIPOS 提供法律级高逻辑推理算力支持。支持完整思考过程。",
+        examples=["请分析这篇专利要求书的逻辑漏洞...", "对比香港与内地的发明专利申请流程", "帮我写一段专利侵权判定的法律意见书"],
     )
 
+    gr.Markdown("---")
+    gr.Markdown("© 2026 OpenIPOS Global Limited. All Rights Reserved. 算力驱动：Hugging Face ZeroGPU")
+
+# 5. 启动程序
 demo.launch()