Upload 2 files

app.py

@@ -18,14 +18,11 @@ def query_financial_data(company_name, query_type):
         return "请输入公司名称或股票代码"
     
     try:
-        # 直接使用 HTTP POST 调用 Gradio API
+        # 直接调用 FastAPI 的 REST API 端点
         # 先搜索公司
         search_resp = requests.post(
-            f"{MCP_URL}/api/predict",
-            json={
-                "data": [company_name],
-                "fn_index": 0  # advanced_search 函数索引
-            },
+            f"{MCP_URL}/api/advanced_search",
+            json={"company_input": company_name},
             headers=HEADERS,
             timeout=30
         )
@@ -34,12 +31,7 @@ def query_financial_data(company_name, query_type):
             return f"❌ Server Error: HTTP {search_resp.status_code}\n\nResponse: {search_resp.text[:500]}"
         
         try:
-            result_data = search_resp.json()
-            # Gradio API 返回格式: {"data": [result]}
-            if "data" in result_data and result_data["data"]:
-                company = result_data["data"][0]
-            else:
-                return f"❌ 无法解析响应: {result_data}"
+            company = search_resp.json()
         except (ValueError, KeyError) as e:
             return f"❌ JSON Parse Error: {str(e)}\n\nResponse: {search_resp.text[:500]}"
         
@@ -55,11 +47,8 @@ def query_financial_data(company_name, query_type):
         # 根据查询类型获取数据
         if query_type == "最新财务数据":
             data_resp = requests.post(
-                f"{MCP_URL}/api/predict",
-                json={
-                    "data": [cik],
-                    "fn_index": 1  # get_latest_financial_data 函数索引
-                },
+                f"{MCP_URL}/api/get_latest_financial_data",
+                json={"cik": cik},
                 headers=HEADERS,
                 timeout=30
             )
@@ -68,8 +57,7 @@ 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_result = data_resp.json()
-                data = data_result["data"][0] if "data" in data_result else data_result
+                data = data_resp.json()
             except (ValueError, KeyError) as e:
                 return result + f"❌ JSON Parse Error: {str(e)}\n\n{data_resp.text[:500]}"
             
@@ -84,11 +72,8 @@ def query_financial_data(company_name, query_type):
         
         elif query_type == "3年趋势":
             metrics_resp = requests.post(
-                f"{MCP_URL}/api/predict",
-                json={
-                    "data": [cik, 3],
-                    "fn_index": 2  # extract_financial_metrics 函数索引
-                },
+                f"{MCP_URL}/api/extract_financial_metrics",
+                json={"cik": cik, "years": 3},
                 headers=HEADERS,
                 timeout=60
             )
@@ -97,8 +82,7 @@ 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_result = metrics_resp.json()
-                metrics = metrics_result["data"][0] if "data" in metrics_result else metrics_result
+                metrics = metrics_resp.json()
             except (ValueError, KeyError) as e:
                 return result + f"❌ JSON Parse Error: {str(e)}\n\n{metrics_resp.text[:500]}"
             
@@ -122,11 +106,8 @@ def query_financial_data(company_name, query_type):
         
         elif query_type == "5年趋势":
             metrics_resp = requests.post(
-                f"{MCP_URL}/api/predict",
-                json={
-                    "data": [cik, 5],
-                    "fn_index": 2  # extract_financial_metrics 函数索引
-                },
+                f"{MCP_URL}/api/extract_financial_metrics",
+                json={"cik": cik, "years": 5},
                 headers=HEADERS,
                 timeout=60
             )
@@ -135,8 +116,7 @@ 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_result = metrics_resp.json()
-                metrics = metrics_result["data"][0] if "data" in metrics_result else metrics_result
+                metrics = metrics_resp.json()
             except (ValueError, KeyError) as e:
                 return result + f"❌ JSON Parse Error: {str(e)}\n\n{metrics_resp.text[:500]}"