Upload 2 files

app.py

@@ -3,6 +3,12 @@ import requests
 
 MCP_URL = "https://jc321-easyreportdatemcp.hf.space"
 
+# 设置请求头
+HEADERS = {
+    "Content-Type": "application/json",
+    "User-Agent": "SEC-Query-Assistant/1.0 (jtyxabc@gmail.com)"
+}
+
 def query_financial_data(company_name, query_type):
     """查询财务数据的主函数"""
     
@@ -10,10 +16,12 @@ def query_financial_data(company_name, query_type):
         return "请输入公司名称或股票代码"
     
     try:
-        # 搜索公司
+        # Gradio Space API 调用方式
+        # 先搜索公司
         search_resp = requests.post(
-            f"{MCP_URL}/api/advanced_search",
-            json={"company_input": company_name},
+            f"{MCP_URL}/call/advanced_search",
+            json={"data": [company_name]},
+            headers=HEADERS,
             timeout=30
         )
         
@@ -22,7 +30,12 @@ def query_financial_data(company_name, query_type):
             return f"❌ Server Error: HTTP {search_resp.status_code}\n\nResponse: {search_resp.text[:500]}"
         
         try:
-            company = search_resp.json()
+            resp_data = search_resp.json()
+            # Gradio API 返回格式: {"data": [result]}
+            if isinstance(resp_data, dict) and "data" in resp_data:
+                company = resp_data["data"][0] if resp_data["data"] else {}
+            else:
+                company = resp_data
         except ValueError as e:
             return f"❌ JSON Parse Error: {str(e)}\n\nResponse Text: {search_resp.text[:500]}"
         
@@ -38,8 +51,9 @@ def query_financial_data(company_name, query_type):
         # 根据查询类型获取数据
         if query_type == "最新财务数据":
             data_resp = requests.post(
-                f"{MCP_URL}/api/get_latest_financial_data",
-                json={"cik": cik},
+                f"{MCP_URL}/call/get_latest_financial_data",
+                json={"data": [cik]},
+                headers=HEADERS,
                 timeout=30
             )
             
@@ -47,7 +61,8 @@ def query_financial_data(company_name, query_type):
                 return result + f"❌ Server Error: HTTP {data_resp.status_code}\n\n{data_resp.text[:500]}"
             
             try:
-                data = data_resp.json()
+                resp_data = data_resp.json()
+                data = resp_data["data"][0] if isinstance(resp_data, dict) and "data" in resp_data else resp_data
             except ValueError as e:
                 return result + f"❌ JSON Parse Error: {str(e)}\n\n{data_resp.text[:500]}"
             
@@ -62,8 +77,9 @@ def query_financial_data(company_name, query_type):
         
         elif query_type == "3年趋势":
             metrics_resp = requests.post(
-                f"{MCP_URL}/api/extract_financial_metrics",
-                json={"cik": cik, "years": 3},
+                f"{MCP_URL}/call/extract_financial_metrics",
+                json={"data": [cik, 3]},
+                headers=HEADERS,
                 timeout=60
             )
             
@@ -71,7 +87,8 @@ def query_financial_data(company_name, query_type):
                 return result + f"❌ Server Error: HTTP {metrics_resp.status_code}\n\n{metrics_resp.text[:500]}"
             
             try:
-                metrics = metrics_resp.json()
+                resp_data = metrics_resp.json()
+                metrics = resp_data["data"][0] if isinstance(resp_data, dict) and "data" in resp_data else resp_data
             except ValueError as e:
                 return result + f"❌ JSON Parse Error: {str(e)}\n\n{metrics_resp.text[:500]}"
             
@@ -95,8 +112,9 @@ def query_financial_data(company_name, query_type):
         
         elif query_type == "5年趋势":
             metrics_resp = requests.post(
-                f"{MCP_URL}/api/extract_financial_metrics",
-                json={"cik": cik, "years": 5},
+                f"{MCP_URL}/call/extract_financial_metrics",
+                json={"data": [cik, 5]},
+                headers=HEADERS,
                 timeout=60
             )
             
@@ -104,7 +122,8 @@ def query_financial_data(company_name, query_type):
                 return result + f"❌ Server Error: HTTP {metrics_resp.status_code}\n\n{metrics_resp.text[:500]}"
             
             try:
-                metrics = metrics_resp.json()
+                resp_data = metrics_resp.json()
+                metrics = resp_data["data"][0] if isinstance(resp_data, dict) and "data" in resp_data else resp_data
             except ValueError as e:
                 return result + f"❌ JSON Parse Error: {str(e)}\n\n{metrics_resp.text[:500]}"
             
@@ -125,8 +144,11 @@ def query_financial_data(company_name, query_type):
         
         return result
         
+    except requests.exceptions.RequestException as e:
+        return f"❌ Network Error: {str(e)}\n\nMCP Server: {MCP_URL}"
     except Exception as e:
-        return f"❌ 发生错误: {str(e)}"
+        import traceback
+        return f"❌ Unexpected Error: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
 
 # 创建 Gradio 界面
 with gr.Blocks(title="SEC Financial Data Query Assistant") as demo: