Update app.py

app.py

@@ -1,53 +1,84 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
+import os
 
-# 初始化客户端 (请确保你的环境变量已设置)
-client = InferenceClient("your-model-id")
+# 初始化
+client = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct")
 
-def respond(message, history, system_message, max_tokens, temperature, top_p):
-    messages = [{"role": "system", "content": system_message}]
-    for user_msg, assistant_msg in history:
-        if user_msg: messages.append({"role": "user", "content": user_msg})
-        if assistant_msg: messages.append({"role": "assistant", "content": assistant_msg})
+# 模拟 A：工具集成逻辑 (Vulnerability Scanner)
+def security_tool_scanner(target_type, content):
+    if target_type == "Code":
+        return f"🔍 [静态扫描] 发现 {len(content)//50} 个潜在注入点... 状态：待确认"
+    return "🌐 [资产识别] 目标服务响应正常，检测到 OpenSSH 8.2..."
 
-    messages.append({"role": "user", "content": message})
+# 核心聊天逻辑
+def chat_engine(message, history, system_message, file_content):
+    # C: RAG 逻辑处理 - 如果有上传文件，将其加入上下文
+    context = ""
+    if file_content is not None:
+        context = f"\n[参考私有知识库内容]: {file_content[:500]}..." # 简单模拟读取
 
-    response = ""
-    for message in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        token = message.choices[0].delta.content
-        response += token
-        yield response
-
-# --- 修复点 1: gr.Blocks 移除了 theme 和 css 参数 ---
-with gr.Blocks() as demo:
-    gr.Markdown("# 全能私有大脑 v4.0 (Ultra Stable)")
+    formatted_history = [{"role": "system", "content": system_message + context}]
+    for msg in history:
+        formatted_history.append(msg)
+    
+    formatted_history.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):
+        token = msg_chunk.choices[0].delta.content
+        if token:
+            response += token
+            yield history + [{"role": "user", "content": message}, {"role": "assistant", "content": response}]
+
+# --- B: 极客 UI 样式 (Kali Green 风格) ---
+terminal_css = """
+.gradio-container { background-color: #050505 !important; color: #00ff41 !important; font-family: 'Courier New', 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"})
+
     with gr.Row():
-        with gr.Column(scale=1):
-            system_input = gr.Textbox(value="你是一个资深安全审计专家...", label="系统指令")
-            with gr.Accordion("高级参数设置", open=False):
-                temp = gr.Slider(minimum=0.1, maximum=1.0, value=0.7, step=0.1, label="Temperature")
-                tokens = gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max Tokens")
-            
-            render_box = gr.HTML("✨ 实时安全分析渲染就绪...")
+        # 左栏：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)
+
+            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)
 
-        with gr.Column(scale=2):
-            # --- 修复点 2: 移除 show_copy_button 参数 ---
-            chat = gr.Chatbot(height=600, avatar_images=(None, "https://path-to-your-icon.png"))
-            msg = gr.Textbox(placeholder="输入指令进行分析...", container=False)
-            clear = gr.ClearButton([msg, chat])
+        # 右栏：核心交互区
+        with gr.Column(scale=3):
+            chatbot = gr.Chatbot(type="messages", 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")
+            
+            with gr.Accordion("B: 系统内核配置", open=False):
+                sys_msg = gr.Textbox(value="你是一个拥有最高权限的黑盒测试专家。", label="System Kernel")
 
-    msg.submit(respond, [msg, chat, system_input, tokens, temp], [chat])
+    # 绑定逻辑
+    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])
 
-# --- 修复点 3: 将 theme 和 css 移动到 launch 方法中 ---
 if __name__ == "__main__":
-    demo.launch(
-        theme=gr.themes.Soft(), 
-        css=".gradio-container {background-color: #0b0f19;}"
-    )
+    demo.launch(ssr=True)