Update app.py

app.py

@@ -174,11 +174,13 @@ def chatbot_response(message, history):
                 for tool_call in choice.message.tool_calls:
                     tool_name = tool_call.function.name
                     tool_args = json.loads(tool_call.function.arguments)
-                    tool_calls_log.append({"name": tool_name, "arguments": tool_args})
                     
                     # 调用 MCP 工具
                     tool_result = call_mcp_tool(tool_name, tool_args)
                     
+                    # 记录工具调用(包含结果)
+                    tool_calls_log.append({"name": tool_name, "arguments": tool_args, "result": tool_result})
+                    
                     messages.append({
                         "role": "tool",
                         "name": tool_name,
@@ -195,18 +197,56 @@ def chatbot_response(message, history):
         response_prefix += "<span style='color: white; font-size: 0.9em;'>🤖 <strong>Qwen2.5-72B</strong></span>\n"
         response_prefix += "</div>\n\n"
         
-        # 显示工具调用
+        # 显示工具调用(可展开/折叠)
         if tool_calls_log:
+            # 添加 JavaScript 脚本用于切换展开/折叠
+            response_prefix += """<script>
+function toggleToolDetails(id) {
+    var details = document.getElementById('tool-details-' + id);
+    var arrow = document.getElementById('tool-arrow-' + id);
+    if (details.style.display === 'none' || details.style.display === '') {
+        details.style.display = 'block';
+        arrow.innerHTML = '▼';
+    } else {
+        details.style.display = 'none';
+        arrow.innerHTML = '▶';
+    }
+}
+</script>\n"""
+            
             response_prefix += "<div style='background: #f8f9fa; border-radius: 6px; padding: 10px; margin-bottom: 12px;'>\n"
             response_prefix += "<div style='font-weight: 600; color: #495057; margin-bottom: 8px;'>🛠️ Tools Used</div>\n"
             
             tool_icons = {"advanced_search_company": "🔍", "get_latest_financial_data": "📊", "extract_financial_metrics": "📈", 
                          "get_quote": "💹", "get_market_news": "📰", "get_company_news": "📢"}
             
-            for tool_call in tool_calls_log:
+            for idx, tool_call in enumerate(tool_calls_log):
                 icon = tool_icons.get(tool_call['name'], "⚙️")
-                response_prefix += f"<div style='background: white; padding: 6px 10px; margin: 4px 0; border-radius: 4px; border-left: 3px solid #28a745;'>\n"
+                tool_id = f"tool_{idx}"
+                
+                # 工具主体(可点击)
+                response_prefix += f"<div style='background: white; padding: 8px 10px; margin: 6px 0; border-radius: 4px; border-left: 3px solid #28a745; cursor: pointer;' onclick='toggleToolDetails(\"{tool_id}\")'>\n"
+                response_prefix += f"<span id='tool-arrow-{tool_id}' style='color: #666; margin-right: 6px; font-size: 0.8em;'>▶</span>"
                 response_prefix += f"<span style='color: #28a745; font-weight: 600;'>{icon} {tool_call['name']}</span>\n"
+                response_prefix += f"</div>\n"
+                
+                # 工具详情(默认隐藏)
+                response_prefix += f"<div id='tool-details-{tool_id}' style='display: none; background: #fff; padding: 10px; margin: 0 0 6px 20px; border-radius: 4px; border: 1px solid #e0e0e0; font-size: 0.85em;'>\n"
+                
+                # 输入参数
+                response_prefix += "<div style='margin-bottom: 8px;'>\n"
+                response_prefix += "<strong style='color: #2196f3;'>📝 Input:</strong>\n"
+                response_prefix += f"<pre style='background: #f5f5f5; padding: 8px; border-radius: 4px; margin: 4px 0; overflow-x: auto;'>{json.dumps(tool_call['arguments'], indent=2, ensure_ascii=False)}</pre>\n"
+                response_prefix += "</div>\n"
+                
+                # 输出结果
+                if 'result' in tool_call:
+                    response_prefix += "<div>\n"
+                    response_prefix += "<strong style='color: #4caf50;'>✅ Output:</strong>\n"
+                    result_preview = str(tool_call['result'])[:200] + '...' if len(str(tool_call['result'])) > 200 else str(tool_call['result'])
+                    response_prefix += f"<pre style='background: #f5f5f5; padding: 8px; border-radius: 4px; margin: 4px 0; overflow-x: auto;'>{result_preview}</pre>\n"
+                    response_prefix += "</div>\n"
+                
                 response_prefix += "</div>\n"
             
             response_prefix += "</div>\n\n"
@@ -233,17 +273,13 @@ def chatbot_response(message, history):
         yield f"❌ Error: {str(e)}"
 
 # ========== Gradio 界面 ==========
-with gr.Blocks(title="Financial & Market AI Assistant") as demo:
+with gr.Blocks(title="Financial & Market AI Assistant", css="""
+    .contain { max-width: 100% !important; padding: 0 !important; }
+    #component-0 { height: 100vh !important; }
+    .chatbot { height: calc(100vh - 200px) !important; }
+""") as demo:
     gr.Markdown("# 🤖 Financial & Market AI Assistant")
     
-    gr.Markdown("""
-    <div style='padding: 15px; background: #d4edda; border-left: 4px solid #28a745; margin: 10px 0; border-radius: 4px;'>
-        <strong>✅ AI Powered by:</strong> Qwen/Qwen2.5-72B-Instruct:novita
-        <br>
-        <strong>📊 Data Sources:</strong> SEC Financial Reports + Market & Stock Data
-    </div>
-    """)
-    
     chat = gr.ChatInterface(
         fn=chatbot_response,
         examples=[
@@ -253,9 +289,16 @@ with gr.Blocks(title="Financial & Market AI Assistant") as demo:
             "Get the latest market news about crypto",
             "Compare Microsoft's latest earnings with its current stock price",
         ],
-        title="💬 AI Assistant",
-        description="Ask me about company financials, stock prices, or market trends. I'll automatically fetch the data you need!"
+        title="💬 Chat with AI",
+        chatbot=gr.Chatbot(height=700, show_copy_button=True),
     )
+    
+    # 底部信息栏(缩小)
+    gr.Markdown("""
+    <div style='padding: 8px; background: #f8f9fa; border-top: 1px solid #dee2e6; margin-top: 10px; font-size: 0.75em; color: #6c757d; text-align: center;'>
+        ✅ <strong>AI:</strong> Qwen/Qwen2.5-72B-Instruct:novita &nbsp;|&nbsp; 📊 <strong>Data:</strong> SEC Financial Reports + Market & Stock Data
+    </div>
+    """)
 
 # 启动应用
 if __name__ == "__main__":