Upload app.py

app.py

@@ -2,6 +2,9 @@ import gradio as gr
 import requests
 import json
 import os
+import time
+from requests.adapters import HTTPAdapter
+from urllib3.util.retry import Retry
 
 MCP_SPACE = "JC321/EasyReportsMCPServer"
 MCP_URL = "https://jc321-easyreportsmcpserver.hf.space"
@@ -12,6 +15,23 @@ HEADERS = {
     "User-Agent": "SEC-Query-Assistant/1.0 (jtyxabc@gmail.com)"
 }
 
+# 创建带重试的 requests session
+def create_session_with_retry():
+    """创建带重试机制的 requests session"""
+    session = requests.Session()
+    retry = Retry(
+        total=3,  # 最多重试3次
+        backoff_factor=1,  # 重试间隔：1秒, 2秒, 4秒
+        status_forcelist=[500, 502, 503, 504],  # 这些状态码会触发重试
+    )
+    adapter = HTTPAdapter(max_retries=retry)
+    session.mount('http://', adapter)
+    session.mount('https://', adapter)
+    return session
+
+# 创建全局 session
+session = create_session_with_retry()
+
 # 格式化数值显示
 def format_value(value, value_type="money"):
     """
@@ -141,18 +161,21 @@ def query_financial_data(company_name, query_type):
     try:
         # 使用 MCP 协议调用工具
         # 先搜索公司（使用 advanced_search_company）
-        search_resp = requests.post(
-            f"{MCP_URL}/message",
-            json={
-                "method": "tools/call",
-                "params": {
-                    "name": "advanced_search_company",
-                    "arguments": {"company_input": company_name}
-                }
-            },
-            headers=HEADERS,
-            timeout=60  # 增加到60秒
-        )
+        try:
+            search_resp = session.post(
+                f"{MCP_URL}/message",
+                json={
+                    "method": "tools/call",
+                    "params": {
+                        "name": "advanced_search_company",
+                        "arguments": {"company_input": company_name}
+                    }
+                },
+                headers=HEADERS,
+                timeout=60  # 增加到60秒
+            )
+        except requests.exceptions.Timeout:
+            return f"❌ MCP Server Timeout: The server took too long to respond (>60s).\n\n**Possible reasons**:\n1. MCP Server is cold starting (first request after idle)\n2. Server is overloaded\n3. Network issues\n\n**Suggestion**: Please try again in a few moments. If the problem persists, the MCP Server at {MCP_URL} may be down."
         
         if search_resp.status_code != 200:
             return f"❌ Server Error: HTTP {search_resp.status_code}\n\nResponse: {search_resp.text[:500]}"
@@ -184,7 +207,7 @@ def query_financial_data(company_name, query_type):
         
         # 根据查询类型获取数据
         if internal_query_type == "最新财务数据":
-            data_resp = requests.post(
+            data_resp = session.post(
                 f"{MCP_URL}/message",
                 json={
                     "method": "tools/call",
@@ -230,7 +253,7 @@ def query_financial_data(company_name, query_type):
             result += f"- **Source Form**: {create_source_link(source_form, source_url)}\n"
         
         elif internal_query_type == "3年趋势":
-            metrics_resp = requests.post(
+            metrics_resp = session.post(
                 f"{MCP_URL}/message",
                 json={
                     "method": "tools/call",
@@ -312,7 +335,7 @@ def query_financial_data(company_name, query_type):
                 result += f"| {display_period} | {format_value(rev)} | {format_value(inc)} | {format_value(eps_val, 'eps')} | {format_value(opex)} | {format_value(ocf)} | {source_link} |\n"
         
         elif internal_query_type == "5年趋势":
-            metrics_resp = requests.post(
+            metrics_resp = session.post(
                 f"{MCP_URL}/message",
                 json={
                     "method": "tools/call",
@@ -394,7 +417,7 @@ def query_financial_data(company_name, query_type):
         
         elif internal_query_type == "公司报表列表":
             # 查询公司所有报表
-            filings_resp = requests.post(
+            filings_resp = session.post(
                 f"{MCP_URL}/message",
                 json={
                     "method": "tools/call",
@@ -518,6 +541,12 @@ with gr.Blocks(title="SEC Financial Data Query Assistant") as demo:
     gr.Markdown("# 🤖 SEC Financial Data Query Assistant")
     gr.Markdown("Query SEC financial data for US listed companies through MCP Server")
     
+    # 添加 MCP Server 状态提示
+    with gr.Row():
+        gr.Markdown(f"🔗 **MCP Server**: [{MCP_URL}]({MCP_URL})")
+    with gr.Row():
+        gr.Markdown("⚠️ **Note**: First query after idle may take 1-2 minutes (server cold start). Please be patient.")
+    
     with gr.Tab("AI Assistant"):
         # 使用 Gradio ChatInterface（兼容 4.44.1）
         chat = gr.ChatInterface(