Update app.py

app.py

@@ -1,19 +1,18 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
-# 初始化客户端
+# 初始化客户端 (请确保你的环境变量已设置)
 client = InferenceClient("your-model-id")
 
 def respond(message, history, system_message, max_tokens, temperature, top_p):
     messages = [{"role": "system", "content": system_message}]
-    for val in history:
-        if val[0]: messages.append({"role": "user", "content": val[0]})
-        if val[1]: messages.append({"role": "assistant", "content": val[1]})
+    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})
 
     messages.append({"role": "user", "content": message})
 
     response = ""
-    # 修复了旧版 InferenceClient 可能存在的迭代问题，确保流式输出顺滑
     for message in client.chat_completion(
         messages,
         max_tokens=max_tokens,
@@ -25,28 +24,30 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         response += token
         yield response
 
-# 升级后的专业 UI 布局
-with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {background-color: #0b0f19;}") as demo:
+# --- 修复点 1: gr.Blocks 移除了 theme 和 css 参数 ---
+with gr.Blocks() as demo:
     gr.Markdown("# 全能私有大脑 v4.0 (Ultra Stable)")
     
     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("✨ 实时安全分析渲染就绪...")
 
-        # 右侧聊天区
         with gr.Column(scale=2):
-            chat = gr.Chatbot(height=600, show_copy_button=True, avatar_images=(None, "https://path-to-your-icon.png"))
+            # --- 修复点 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])
 
     msg.submit(respond, [msg, chat, system_input, tokens, temp], [chat])
 
+# --- 修复点 3: 将 theme 和 css 移动到 launch 方法中 ---
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(
+        theme=gr.themes.Soft(), 
+        css=".gradio-container {background-color: #0b0f19;}"
+    )