Update app.py

app.py

@@ -1,8 +1,7 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
-import re
 
-# 备选模型列表，增加稳定性
+# 多模型矩阵
 MODELS = {
     "Qwen 2.5 Coder (专业审计)": "Qwen/Qwen2.5-Coder-32B-Instruct",
     "Llama 3.1 (逻辑推理)": "meta-llama/Llama-3.1-70B-Instruct",
@@ -10,16 +9,14 @@ MODELS = {
 }
 
 def ghost_chat(message, history, system_message, model_name, file_obj):
-    # 自动选择模型
-    repo_id = MODELS.get(model_name, MODELS["Qwen 2.5 Coder (专业审计)"])
+    repo_id = MODELS.get(model_name, "Qwen/Qwen2.5-Coder-32B-Instruct")
     client = InferenceClient(repo_id)
     
-    # 文件处理
     file_context = ""
     if file_obj is not None:
         try:
             with open(file_obj.name, "r", encoding="utf-8") as f:
-                file_context = f"\n[File Context]:\n{f.read()[:1000]}"
+                file_context = f"\n[文件内容]:\n{f.read()[:800]}"
         except: pass
 
     messages = [{"role": "system", "content": f"{system_message}\n{file_context}"}]
@@ -29,7 +26,7 @@ def ghost_chat(message, history, system_message, model_name, file_obj):
     messages.append({"role": "user", "content": message})
 
     response = ""
-    yield history + [[message, "🛰️ 正在链接卫星链路..."]]
+    yield history + [[message, "📡 链路建立中..."]]
     
     try:
         for msg_chunk in client.chat_completion(messages, stream=True, max_tokens=2048):
@@ -38,41 +35,38 @@ def ghost_chat(message, history, system_message, model_name, file_obj):
                 response += token
                 yield history + [[message, response]]
     except Exception as e:
-        yield history + [[message, f"❌ 系统崩溃: {str(e)}"]]
+        yield history + [[message, f"❌ 链路中断: {str(e)}"]]
 
-# --- 极简黑盒样式 ---
+# 极客样式
 style = """
 .gradio-container { background: #000 !important; color: #0f0 !important; }
-.message.user { border: 1px solid #0f0 !important; }
-.tabs { border: none !important; }
+.message.user { border-left: 3px solid #0f0 !important; }
+#custom-title { text-align: center; font-size: 24px; color: #0f0; margin-bottom: 20px; }
 """
 
-with gr.Blocks(fill_height=True, css=style) as demo:
-    gr.Markdown("## 🌌 INTERSTELLAR CORE v8.0", textAlign="center")
+# --- UI 架构 ---
+with gr.Blocks(fill_height=True) as demo:
+    # 修复点 1: 移除 textAlign，改用 HTML/Markdown 组合
+    gr.HTML("<div id='custom-title'>🌌 INTERSTELLAR CORE v8.5</div>")
     
     with gr.Row():
-        # 左侧控制面板
         with gr.Column(scale=1):
-            with gr.Group():
-                model_dd = gr.Dropdown(list(MODELS.keys()), value="Qwen 2.5 Coder (专业审计)", label="选择核心")
-                sys_msg = gr.Textbox(value="你是一个顶尖的 AI 助手。", label="系统架构")
+            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_input = gr.Textbox(label="扫描器输入")
-                scan_btn = gr.Button("RUN ANALYSIS")
-                scan_res = gr.Textbox(label="分析结果")
+            with gr.Accordion("🛠️ 工具箱", open=False):
+                file_up = gr.File(label="注入文档")
+                scan_btn = gr.Button("执行系统诊断")
 
-        # 右侧主控制台
         with gr.Column(scale=3):
-            chatbot = gr.Chatbot(height=650)
+            chatbot = gr.Chatbot(height=600)
             with gr.Row():
-                msg = gr.Textbox(placeholder="输入指令...", scale=8, container=False)
-                send = gr.Button("EXE", scale=2)
+                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])
 
+# 修复点 2: 所有的 CSS 参数必须在这里！
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(css=style)