Update app.py

app.py

@@ -2,40 +2,47 @@ import gradio as gr
 import spaces
 from huggingface_hub import InferenceClient
 
-# 初始化客户端，指向 DeepSeek-R1-32B
+# 初始化客户端
 client = InferenceClient("deepseek-ai/DeepSeek-R1-Distill-Qwen-32B")
 
 @spaces.GPU
 def chat_fn(message, history):
-    # 构建符合 OpenIPOS 标准的对话上下文
+    # 构建对话消息列表
     messages = []
     for turn in history:
-        messages.append({"role": "user", "content": turn[0]})
-        messages.append({"role": "assistant", "content": turn[1]})
+        # turn 可能是一个字典（新版）或元组（旧版）
+        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})
     
-    generate_kwargs = dict(
-        max_tokens=2048,
-        stream=True,
-    )
-
     response = ""
-    # 流式获取，确保文字一个一个跳出来，更有算力质感
-    for msg in client.chat_completion(messages, **generate_kwargs):
-        token = msg.choices[0].delta.content
-        if token:
-            response += token
-            yield response
+    # 调用流式接口
+    try:
+        for msg in client.chat_completion(
+            messages,
+            max_tokens=2048,
+            stream=True,
+        ):
+            token = msg.choices[0].delta.content
+            if token:
+                response += token
+                yield response
+    except Exception as e:
+        yield f"抱歉，算力节点暂时忙碌，请稍后再试。错误信息: {str(e)}"
 
 # 构建 OpenIPOS 品牌界面
 with gr.Blocks(fill_height=True) as demo:
     gr.Markdown("# 🚀 OpenIPOS Global 算力节点")
     gr.Markdown("### DeepSeek-R1 32B 满血版演示站")
     
-    # 核心聊天组件（彻底删除了报错的 type 参数）
+    # 使用最简洁的参数配置
     gr.ChatInterface(
-        fn=chat_fn,
-        description="本节点由 OpenIPOS 提供法律级高逻辑推理算力支持。支持 DeepSeek-R1 完整思考链路。",
+        chat_fn,
+        description="本节点由 OpenIPOS 提供法律级高逻辑推理算力支持。支持完整思考过程。",
     )
 
 demo.launch()