Update app.py

app.py

@@ -1,5 +1,43 @@
+import os
+import gradio as gr
+import spaces  # 必须在这里先导入
+import fitz
+import docx
+from huggingface_hub import InferenceClient
+
+# 1. 引擎配置：DeepSeek-V3.2-Exp (685B 逻辑核弹)
+model_id = "deepseek-ai/DeepSeek-V3.2-Exp"
+hf_token = os.getenv("HF_TOKEN")
+
+# 强制授权连接
+client = InferenceClient(
+    model=model_id, 
+    token=hf_token,
+    headers={"Authorization": f"Bearer {hf_token}"} if hf_token else None
+)
+
+def extract_text(file_path):
+    if not file_path: return ""
+    try:
+        ext = str(file_path).lower()
+        if ext.endswith(".pdf"):
+            doc = fitz.open(file_path)
+            text = "".join([page.get_text() for page in doc])
+            doc.close()
+            return text[:15000]
+        elif ext.endswith(".docx"):
+            doc = docx.Document(file_path)
+            return "\n".join([p.text for p in doc.paragraphs])[:15000]
+        elif ext.endswith(".txt"):
+            with open(file_path, 'r', encoding='utf-8', errors='ignore') as f:
+                return f.read()[:15000]
+        return ""
+    except:
+        return ""
+
 @spaces.GPU(duration=60)
 def chat_fn(message, history):
+    # 清洗历史记录，确保 DeepSeek 接收的是纯字符串
     clean_messages = []
     for turn in history:
         clean_messages.append({"role": turn["role"], "content": str(turn["content"])})
@@ -21,11 +59,9 @@ def chat_fn(message, history):
         stream = client.chat_completion(clean_messages, max_tokens=4096, stream=True, temperature=0.5)
         for msg in stream:
             if hasattr(msg, 'choices') and len(msg.choices) > 0:
-                delta = msg.choices[0].delta
-                if hasattr(delta, 'content') and delta.content:
-                    # --- 核心修复逻辑开始 ---
-                    content = delta.content
-                    # 如果返回的是列表格式的字符串，尝试清洗它
+                content = msg.choices[0].delta.content
+                if content:
+                    # --- 核心修复：处理 JSON 列表格式的脏数据 ---
                     if isinstance(content, list) and len(content) > 0:
                         content = content[0].get("text", "")
                     elif isinstance(content, dict):
@@ -33,9 +69,24 @@ def chat_fn(message, history):
                     
                     response += str(content)
                     yield response
-                    # --- 核心修复逻辑结束 ---
         
         if response:
-            yield response + "\n\n---\n> **OpenIPOS MegaNode | DeepSeek-V3.2-Exp 逻辑推理出口**"
+            yield response + "\n\n---\n> **OpenIPOS MegaNode | DeepSeek-V3 逻辑推理出口**"
     except Exception as e:
-        yield f"⚠️ 算力调度提示: {str(e)}。由于 DeepSeek 节点极其繁忙，请刷新重试。"
+        yield f"⚠️ 算力调度提示: {str(e)}。由于 DeepSeek 节点极繁忙，请刷新重试。"
+
+# 3. UI 装修
+with gr.Blocks(fill_height=True) as demo:
+    gr.HTML("""
+        <div style="display: flex; align-items: center; gap: 20px; padding: 10px;">
+            <img src="https://s3.bmp.ovh/2026/03/30/rIAGKg0O.png" style="height: 60px; width: auto;">
+            <div>
+                <h1 style="margin: 0; font-size: 24px;">OpenIPOS MegaNode</h1>
+                <p style="margin: 0; color: #666;">全球领先法律与逻辑推理节点 (DeepSeek-R1 级别算力)</p>
+            </div>
+        </div>
+    """)
+    gr.ChatInterface(chat_fn, multimodal=True)
+    gr.Markdown("© 2026 OpenIPOS Global Limited. 算力驱动：Novita API Cluster")
+
+demo.launch(theme=gr.themes.Soft())