Update app.py

app.py

@@ -2,32 +2,29 @@ import gradio as gr
 import requests
 import os
 import json
-import fitz  # PyMuPDF
-import re
 
 # --- 核心配置 ---
 OPENROUTER_API_KEY = os.environ.get("OPENROUTER_API_KEY")
-# 默认使用 Gemini 2.0 Flash (OpenRouter 托管)，如遇繁忙可改为 "deepseek/deepseek-chat:free"
 MODEL_ID = "google/gemini-2.0-flash-001" 
 
 # --- 核心声明 HTML ---
 INFO_HTML = """
 <div style="text-align: left; border-left: 4px solid #2196F3; padding-left: 15px; margin-bottom: 20px;">
     <h3>MG TaxAI | 跨境财税合规实验室 (Beta)</h3>
-    <p>本系统依托 <b>MG 核心智库</b> 构建，旨在实现解析结果实时溯源至各国官方税收协定与法律文本。目前系统正处于<b>知识库全量装载阶段</b>，已优先上线核心业务国家的官方协定库。</p>
-    <p>我们正持续同步全球各主要经济体的国别投资税收指南及多税种年度税收报告。受限于测试版的数据填充进度，相关解析结果仅供专业参考。MG团队正加速完善每一条咨询建议的合规证据链，以确保交付专家级的数字化合规支持。</p>
+    <p>本系统依托 <b>MG 核心智库</b> 构建，旨在实现解析结果实时溯源至各国官方税收协定与法律文本。目前系统正处于<b>知识库全量装载阶段</b>。</p>
+    <p>受限于测试版的数据填充进度，相关解析结果仅供专业参考。MG团队正加速完善每一条咨询建议的合规证据链，以确保交付专家级的数字化合规支持。</p>
     <hr style="border: 0; border-top: 1px solid #eee; margin: 10px 0;">
     <p style="font-size: 0.85em; color: #666;">
         <b>⚠️ AI 免责声明：</b><br>
-        本系统生成的内容由人工智能根据现有库文件分析得出，不构成正式的法律或税务建议。在使用本系统结果进行任何商业决策前，请务必咨询 MG Consult 专业团队。
+        本系统生成的内容由人工智能根据现有库文件分析得出，不构成正式的法律或税务建议。
     </p>
 </div>
 """
 
 # --- API 调用逻辑 ---
-def ask_ai(user_query):
+def ask_ai(message, history):
     if not OPENROUTER_API_KEY:
-        return "错误：未检测到 OPENROUTER_API_KEY，请在环境变量中配置。"
+        return "错误：未检测到 API Key。"
 
     url = "https://openrouter.ai/api/v1/chat/completions"
     headers = {
@@ -37,36 +34,35 @@ def ask_ai(user_query):
     
     payload = {
         "model": MODEL_ID,
-        "messages": [
-            {"role": "system", "content": "你是一个专业的国际税务与贸易合规专家。请用专业、严谨的中文回答。"},
-            {"role": "user", "content": user_query}
-        ],
+        "messages": [{"role": "system", "content": "你是一个专业的国际税务专家。"}] + 
+                    [{"role": "user" if i%2==0 else "assistant", "content": h[0] if i%2==0 else h[1]} for i, h in enumerate(history)] +
+                    [{"role": "user", "content": message}],
         "temperature": 0.5
     }
 
     try:
-        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=40)
+        response = requests.post(url, headers=headers, data=json.dumps(payload), timeout=60)
         if response.status_code == 200:
-            result = response.json()
-            return result['choices'][0]['message']['content']
-        else:
-            return f"API 访问异常 (Code {response.status_code}): {response.text}"
+            return response.json()['choices'][0]['message']['content']
+        return f"接口异常: {response.status_code}"
     except Exception as e:
-        return f"系统连接失败: {str(e)}"
+        return f"连接失败: {str(e)}"
 
-# --- 构建 Gradio UI ---
-with gr.Blocks(title="MG Consulting TaxAI", css="footer {visibility: hidden}") as demo:
-    # 嵌入你提供的 HTML 声明
+# --- 构建界面 ---
+# 使用 fill_height=True 让容器占满浏览器高度
+with gr.Blocks(title="MG Consulting TaxAI", fill_height=True) as demo:
+    
     gr.HTML(INFO_HTML)
     
-    with gr.Row():
-        with gr.Column():
-            chatbot = gr.ChatInterface(
-                fn=lambda message, history: ask_ai(message),
-                fill_height=True
-            )
+    # 这里的 ChatInterface 设置 fill_height=True
+    # 它会自动根据屏幕大小调整高度，不再是短短的一截
+    chat_interface = gr.ChatInterface(
+        fn=ask_ai,
+        fill_height=True, 
+        retry_btn="🔄 重试",
+        undo_btn="↩️ 撤回",
+        clear_btn="🗑️ 清除记录",
+    )
 
-# --- 启动应用 ---
 if __name__ == "__main__":
-    # 确保 demo.launch() 缩进正确
     demo.launch()