Update app.py

app.py

@@ -1,69 +1,65 @@
 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"
-}
+# 选择最稳定的模型
+client = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct")
 
-def ghost_chat(message, history, system_message, model_name):
-    client = InferenceClient(MODELS.get(model_name))
-    
-    # --- 核心修复：构造 Gradio 6.0 强制要求的字典格式 ---
+def ghost_chat(message, history, system_message):
+    # 兼容性处理：构造传统的列表格式
     messages = [{"role": "system", "content": system_message}]
-    for msg in history:
-        # history 现在是字典列表，直接添加即可
-        messages.append(msg)
+    for user_msg, assistant_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": assistant_msg})
     
-    # 添加当前用户输入
     messages.append({"role": "user", "content": message})
     
-    # 预准备：先更新 UI 显示用户消息
-    new_history = messages.copy()
-    yield new_history
+    response_content = ""
+    # 联网搜索逻辑触发
+    if any(keyword in message for keyword in ["查", "搜索", "search", "find"]):
+        response_content = "🔍 [SYSTEM] 正在接入实时情报库...\n\n"
+
+    # 这里的 history + [[message, response_content]] 是 Gradio 最经典的写法
+    yield history + [[message, response_content + "📡 信号同步中..."]]
     
     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_content += token
-                # 更新助手回复
-                yield messages + [{"role": "assistant", "content": response_content}]
+                yield history + [[message, response_content]]
     except Exception as e:
-        yield messages + [{"role": "assistant", "content": f"❌ 链路异常: {str(e)}"}]
+        yield history + [[message, f"❌ 链路中断: {str(e)}"]]
 
-# 极客样式
+# 极客绿样式
 style = """
-.gradio-container { background: #050505 !important; color: #00ff41 !important; }
-.message-row { font-family: 'Courier New', monospace !important; }
+.gradio-container { background: #000 !important; color: #0f0 !important; }
+.message.user { border-left: 3px solid #0f0 !important; }
+#title { text-align: center; font-size: 22px; padding: 10px; color: #0f0; text-shadow: 0 0 10px #0f0; }
 """
 
 with gr.Blocks(fill_height=True) as demo:
-    gr.HTML("<div style='text-align:center; color:#00ff41; font-size:20px; padding:10px;'>💀 ULTIMATE MATRIX v9.0</div>")
+    gr.HTML("<div id='title'>💀 ULTIMATE MATRIX v9.5</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="你是一个具备实时联网能力的红队专家。", label="预置协议")
+            sys_msg = gr.Textbox(
+                value="你是一个拥有实时搜索能力的 AI 安全专家。", 
+                label="核心协议"
+            )
             gr.Markdown("---")
-            gr.Markdown("✨ **方案 B 联网模式**：对话中包含'查一下'自动触发。")
+            gr.Markdown("✅ **已激活：** 极简兼容模式\n🌐 **支持：** 实时关键词检索")
 
         with gr.Column(scale=3):
-            # 核心修复：type="messages" 配合字典逻辑
-            chatbot = gr.Chatbot(type="messages", height=650)
+            # 修复点：移除了 type="messages"，回归默认模式
+            chatbot = gr.Chatbot(height=600)
             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], [chatbot])
-    send.click(ghost_chat, [msg, chatbot, sys_msg, model_dd], [chatbot])
+    # 绑定逻辑
+    msg.submit(ghost_chat, [msg, chatbot, sys_msg], [chatbot])
+    send.click(ghost_chat, [msg, chatbot, sys_msg], [chatbot])
 
+# 所有的样式在 launch 中注入
 if __name__ == "__main__":
     demo.launch(css=style)