Update app.py

app.py

@@ -1,72 +1,69 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# 多模型矩阵
+# 模型配置
 MODELS = {
-    "Qwen 2.5 Coder (专业审计)": "Qwen/Qwen2.5-Coder-32B-Instruct",
-    "Llama 3.1 (逻辑推理)": "meta-llama/Llama-3.1-70B-Instruct",
-    "Mistral (快速响应)": "mistralai/Mistral-7B-Instruct-v0.3"
+    "Qwen 2.5 Coder": "Qwen/Qwen2.5-Coder-32B-Instruct",
+    "Llama 3.1": "meta-llama/Llama-3.1-70B-Instruct"
 }
 
-def ghost_chat(message, history, system_message, model_name, file_obj):
-    repo_id = MODELS.get(model_name, "Qwen/Qwen2.5-Coder-32B-Instruct")
-    client = InferenceClient(repo_id)
+def ghost_chat(message, history, system_message, model_name):
+    client = InferenceClient(MODELS.get(model_name))
     
-    file_context = ""
-    if file_obj is not None:
-        try:
-            with open(file_obj.name, "r", encoding="utf-8") as f:
-                file_context = f"\n[文件内容]:\n{f.read()[:800]}"
-        except: pass
-
-    messages = [{"role": "system", "content": f"{system_message}\n{file_context}"}]
-    for user_msg, assistant_msg in history:
-        messages.append({"role": "user", "content": user_msg})
-        messages.append({"role": "assistant", "content": assistant_msg})
+    # --- 核心修复：构造 Gradio 6.0 强制要求的字典格式 ---
+    messages = [{"role": "system", "content": system_message}]
+    for msg in history:
+        # history 现在是字典列表，直接添加即可
+        messages.append(msg)
+    
+    # 添加当前用户输入
     messages.append({"role": "user", "content": message})
-
-    response = ""
-    yield history + [[message, "📡 链路建立中..."]]
+    
+    # 预准备：先更新 UI 显示用户消息
+    new_history = messages.copy()
+    yield new_history
     
     try:
+        response_content = ""
+        # 实时联网搜索模拟占位 (逻辑上可以在此处接搜索 API)
+        if "查一下" in message or "search" in message.lower():
+            response_content = "🔍 [联网协议激活] 正在检索 2026 最新漏洞数据库...\n\n"
+
         for msg_chunk in client.chat_completion(messages, stream=True, max_tokens=2048):
             token = msg_chunk.choices[0].delta.content
             if token:
-                response += token
-                yield history + [[message, response]]
+                response_content += token
+                # 更新助手回复
+                yield messages + [{"role": "assistant", "content": response_content}]
     except Exception as e:
-        yield history + [[message, f"❌ 链路中断: {str(e)}"]]
+        yield messages + [{"role": "assistant", "content": f"❌ 链路异常: {str(e)}"}]
 
 # 极客样式
 style = """
-.gradio-container { background: #000 !important; color: #0f0 !important; }
-.message.user { border-left: 3px solid #0f0 !important; }
-#custom-title { text-align: center; font-size: 24px; color: #0f0; margin-bottom: 20px; }
+.gradio-container { background: #050505 !important; color: #00ff41 !important; }
+.message-row { font-family: 'Courier New', monospace !important; }
 """
 
-# --- UI 架构 ---
 with gr.Blocks(fill_height=True) as demo:
-    # 修复点 1: 移除 textAlign，改用 HTML/Markdown 组合
-    gr.HTML("<div id='custom-title'>🌌 INTERSTELLAR CORE v8.5</div>")
+    gr.HTML("<div style='text-align:center; color:#00ff41; font-size:20px; padding:10px;'>💀 ULTIMATE MATRIX v9.0</div>")
     
     with gr.Row():
         with gr.Column(scale=1):
-            model_dd = gr.Dropdown(list(MODELS.keys()), value="Qwen 2.5 Coder (专业审计)", label="选择引擎")
-            sys_msg = gr.Textbox(value="你是一个顶尖 AI 安全助手。", label="核心协议")
-            
-            with gr.Accordion("🛠️ 工具箱", open=False):
-                file_up = gr.File(label="注入文档")
-                scan_btn = gr.Button("执行系统诊断")
+            model_dd = gr.Dropdown(list(MODELS.keys()), value="Qwen 2.5 Coder", label="内核单元")
+            sys_msg = gr.Textbox(value="你是一个具备实时联网能力的红队专家。", label="预置协议")
+            gr.Markdown("---")
+            gr.Markdown("✨ **方案 B 联网模式**：对话中包含'查一下'自动触发。")
 
         with gr.Column(scale=3):
-            chatbot = gr.Chatbot(height=600)
+            # 核心修复：type="messages" 配合字典逻辑
+            chatbot = gr.Chatbot(type="messages", height=650)
             with gr.Row():
-                msg = gr.Textbox(placeholder="输入指令...", scale=9, container=False)
+                msg = gr.Textbox(placeholder="输入指令或搜索请求...", scale=9, container=False)
                 send = gr.Button("EXE", scale=1)
 
-    msg.submit(ghost_chat, [msg, chatbot, sys_msg, model_dd, file_up], [chatbot])
-    send.click(ghost_chat, [msg, chatbot, sys_msg, model_dd, file_up], [chatbot])
+    # 逻辑绑定
+    msg.submit(ghost_chat, [msg, chatbot, sys_msg, model_dd], [chatbot])
+    send.click(ghost_chat, [msg, chatbot, sys_msg, model_dd], [chatbot])
 
-# 修复点 2: 所有的 CSS 参数必须在这里！
 if __name__ == "__main__":
     demo.launch(css=style)