Update app.py

app.py

@@ -1,84 +1,79 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-import os
 
 # 初始化
 client = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct")
 
-# 模拟 A：工具集成逻辑 (Vulnerability Scanner)
-def security_tool_scanner(target_type, content):
+# 模拟 A：工具集成逻辑
+def security_tool_scanner(target_type):
     if target_type == "Code":
-        return f"🔍 [静态扫描] 发现 {len(content)//50} 个潜在注入点... 状态：待确认"
-    return "🌐 [资产识别] 目标服务响应正常，检测到 OpenSSH 8.2..."
+        return "🔍 [静态扫描] 发现 3 个潜在注入点... 状态：待确认"
+    return "🌐 [资产识别] 目标服务响应正常。"
 
-# 核心聊天逻辑
-def chat_engine(message, history, system_message, file_content):
-    # C: RAG 逻辑处理 - 如果有上传文件，将其加入上下文
-    context = ""
-    if file_content is not None:
-        context = f"\n[参考私有知识库内容]: {file_content[:500]}..." # 简单模拟读取
-
-    formatted_history = [{"role": "system", "content": system_message + context}]
-    for msg in history:
-        formatted_history.append(msg)
+# 核心聊天逻辑 (适配旧版 Chatbot 格式)
+def chat_engine(message, history, system_message):
+    messages = [{"role": "system", "content": system_message}]
+    # 兼容处理：将历史记录转为 API 需要的格式
+    for user_msg, assistant_msg in history:
+        messages.append({"role": "user", "content": user_msg})
+        messages.append({"role": "assistant", "content": assistant_msg})
     
-    formatted_history.append({"role": "user", "content": message})
+    messages.append({"role": "user", "content": message})
 
-    # B: 视觉反馈 - 模拟黑客终端的“思考”过程
-    yield history + [{"role": "assistant", "content": "SYSTEM: Accessing Kernel... Analyzing...", "metadata": {"title": "⚡ 核心注入中..."}}]
-    
     response = ""
-    for msg_chunk in client.chat_completion(formatted_history, stream=True, max_tokens=2048):
+    # 流式输出
+    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 + [{"role": "user", "content": message}, {"role": "assistant", "content": response}]
+            yield history + [[message, response]]
 
-# --- B: 极客 UI 样式 (Kali Green 风格) ---
+# B: 极客 UI 样式
 terminal_css = """
-.gradio-container { background-color: #050505 !important; color: #00ff41 !important; font-family: 'Courier New', monospace !important; }
+.gradio-container { background-color: #050505 !important; color: #00ff41 !important; font-family: 'monospace' !important; }
 .message.user { border-left: 3px solid #00ff41 !important; background: #0a1a0a !important; }
-footer { visibility: hidden; }
 #terminal-header { border-bottom: 2px solid #00ff41; padding-bottom: 10px; margin-bottom: 20px; }
-.gr-button-primary { background: #00ff41 !important; color: black !important; border: none !important; }
 """
 
-with gr.Blocks(fill_height=True, css=terminal_css) as demo:
-    # 头部状态栏
-    with gr.Row(elem_id="terminal-header"):
-        gr.Markdown("### 💀 BLACK-CORE OS v6.0 | SESSION: ACTIVE | ENCRYPTION: AES-256")
-        status_light = gr.HighlightedText(value=[("SYSTEM", "ONLINE"), ("VPN", "CONNECTED")], color_map={"ONLINE": "green", "CONNECTED": "blue"})
+# --- 核心 UI 布局 ---
+with gr.Blocks(fill_height=True) as demo:
+    # 顶部状态栏
+    gr.Markdown("### 💀 BLACK-CORE OS v6.0 | SESSION: ACTIVE", elem_id="terminal-header")
 
     with gr.Row():
-        # 左栏：A+C 工具区
+        # 左栏：工具与知识库
         with gr.Column(scale=1, min_width=300):
             with gr.Group():
                 gr.Markdown("#### [A] 自动化审计工具")
-                tool_type = gr.Dropdown(["Code", "Network", "Social"], label="扫描类型")
-                scan_btn = gr.Button("RUN SCAN", variant="primary")
-                scan_output = gr.Textbox(label="扫描日志", interactive=False, placeholder="等待指令...")
-                scan_btn.click(security_tool_scanner, [tool_type, scan_output], scan_output)
+                tool_type = gr.Dropdown(["Code", "Network"], label="扫描类型")
+                scan_btn = gr.Button("RUN SCAN")
+                scan_output = gr.Textbox(label="扫描日志", interactive=False)
+                scan_btn.click(security_tool_scanner, [tool_type], scan_output)
 
             with gr.Group():
                 gr.Markdown("#### [C] 私有知识库 (RAG)")
-                file_input = gr.File(label="上传审计报告/文档", file_types=[".txt", ".pdf", ".py"])
-                vector_status = gr.Markdown("状态: 知识库未挂载")
-                file_input.change(lambda: "状态: 实时向量索引已建立 ✅", None, vector_status)
+                file_input = gr.File(label="上传审计报告")
+                vector_status = gr.Markdown("状态: 待命")
 
-        # 右栏：核心交互区
+        # 右栏：对话区
         with gr.Column(scale=3):
-            chatbot = gr.Chatbot(type="messages", height="65vh", show_label=False)
+            # 修复点：移除了 type="messages"，回归最稳定的 Chatbot 定义
+            chatbot = gr.Chatbot(height="65vh", show_label=False)
             
             with gr.Row():
-                input_text = gr.Textbox(placeholder="root@kali:~# 输入渗透指令...", scale=8, container=False)
-                submit_btn = gr.Button("EXE", scale=1, variant="primary")
+                input_text = gr.Textbox(placeholder="root@kali:~# 输入指令...", scale=8, container=False)
+                submit_btn = gr.Button("EXE", scale=1)
             
-            with gr.Accordion("B: 系统内核配置", open=False):
-                sys_msg = gr.Textbox(value="你是一个拥有最高权限的黑盒测试专家。", label="System Kernel")
+            with gr.Accordion("内核配置", open=False):
+                sys_msg = gr.Textbox(value="你是一个渗透测试专家。", label="System Kernel")
 
-    # 绑定逻辑
-    input_text.submit(chat_engine, [input_text, chatbot, sys_msg, file_input], [chatbot])
-    submit_btn.click(chat_engine, [input_text, chatbot, sys_msg, file_input], [chatbot])
+    # 逻辑绑定
+    input_text.submit(chat_engine, [input_text, chatbot, sys_msg], [chatbot])
+    submit_btn.click(chat_engine, [input_text, chatbot, sys_msg], [chatbot])
 
+# --- 修复点：所有样式参数必须在 launch 中 ---
 if __name__ == "__main__":
-    demo.launch(ssr=True)
+    demo.launch(
+        theme=gr.themes.Soft(primary_hue="green"),
+        css=terminal_css
+    )