Update app.py

app.py

@@ -2,8 +2,17 @@ import gradio as gr
 import requests
 import json
 import os
+import warnings
 from huggingface_hub import InferenceClient
 
+# 抑制 asyncio 警告
+warnings.filterwarnings('ignore', category=DeprecationWarning)
+os.environ['PYTHONWARNINGS'] = 'ignore'
+
+# 如果在 GPU 环境但不需要 GPU，禁用 CUDA
+if 'CUDA_VISIBLE_DEVICES' not in os.environ:
+    os.environ['CUDA_VISIBLE_DEVICES'] = ''
+
 # ========== MCP 工具简化定义(符合MCP协议标准) ==========
 MCP_TOOLS = [
     {"type": "function", "function": {"name": "advanced_search_company", "description": "Search US companies", "parameters": {"type": "object", "properties": {"company_input": {"type": "string"}}, "required": ["company_input"]}}},
@@ -36,7 +45,14 @@ print(f"✅ LLM initialized: Qwen/Qwen2.5-72B-Instruct:novita")
 print(f"📊 MCP Services: {len(MCP_SERVICES)} services, {len(MCP_TOOLS)} tools")
 
 # ========== 系统提示词(简化) ==========
-SYSTEM_PROMPT = """You are a financial analysis assistant. Use tools to get data on company financials (past 5-year reports), current stock prices, market news, and company news. Provide data-driven insights."""
+from datetime import datetime
+
+def get_system_prompt():
+    """生成包含当前日期的系统提示词"""
+    current_date = datetime.now().strftime("%Y-%m-%d")
+    return f"""You are a financial analysis assistant. Use tools to get data on company financials (past 5-year reports), current stock prices, market news, and company news. Provide data-driven insights.
+
+IMPORTANT: Today's date is {current_date}. When querying news or time-sensitive data, use recent dates relative to today."""
 
 # ============================================================
 # MCP 服务调用核心代码区
@@ -141,7 +157,7 @@ def _call_gradio_api(service_url, tool_name, arguments):
 def chatbot_response(message, history):
     """AI 助手主函数(流式输出,性能优化)"""
     try:
-        messages = [{"role": "system", "content": SYSTEM_PROMPT}]
+        messages = [{"role": "system", "content": get_system_prompt()}]
         
         # 添加历史(最近5轮)
         if history:
@@ -204,12 +220,28 @@ def chatbot_response(message, history):
         # 构建响应前缀（简化版）
         response_prefix = ""
         
-        # 显示工具调用（简单文本）
+        # 显示工具调用（简单文本+可展开输出）
         if tool_calls_log:
             response_prefix += "**🛠️ Tools Used:**\n\n"
-            for tool_call in tool_calls_log:
-                response_prefix += f"- `{tool_call['name']}` {json.dumps(tool_call['arguments'], ensure_ascii=False)}\n"
-            response_prefix += "\n---\n\n"
+            for idx, tool_call in enumerate(tool_calls_log):
+                # 工具名称和输入
+                response_prefix += f"**{idx+1}. `{tool_call['name']}`**\n"
+                response_prefix += f"- 📥 Input: `{json.dumps(tool_call['arguments'], ensure_ascii=False)}`\n"
+                
+                # 输出（可展开）
+                if 'result' in tool_call:
+                    result_str = json.dumps(tool_call['result'], ensure_ascii=False, indent=2)
+                    # 截断过长结果
+                    if len(result_str) > 500:
+                        result_preview = result_str[:500] + "..."
+                    else:
+                        result_preview = result_str
+                    
+                    response_prefix += f"<details>\n<summary>📤 Output (click to expand)</summary>\n\n```json\n{result_preview}\n```\n</details>\n\n"
+                else:
+                    response_prefix += "\n"
+            
+            response_prefix += "---\n\n"
         
         # 流式输出最终答案
         yield response_prefix
@@ -255,9 +287,21 @@ with gr.Blocks(title="Financial AI Assistant") as demo:
 
 # 启动应用
 if __name__ == "__main__":
+    import sys
+    import asyncio
+    
+    # 修复 asyncio 事件循环问题
+    if sys.platform == 'linux':
+        try:
+            import asyncio
+            asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy())
+        except:
+            pass
+    
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
         show_error=True,
-        ssr_mode=False
+        ssr_mode=False,
+        quiet=False
     )