Upload app.py

app.py

@@ -73,7 +73,8 @@ def query_financial_data(company_name, query_type):
     query_type_mapping = {
         "Latest Financial Data": "最新财务数据",
         "3-Year Trends": "3年趋势",
-        "5-Year Trends": "5年趋势"
+        "5-Year Trends": "5年趋势",
+        "Company Filings": "公司报表列表"
     }
     internal_query_type = query_type_mapping.get(query_type, query_type)
     
@@ -225,6 +226,51 @@ def query_financial_data(company_name, query_type):
                 
                 result += f"| {period_prefix}{period} | {format_value(rev)} | {format_value(inc)} | {format_value(eps_val, False)} | {format_value(opex)} | {format_value(ocf)} | {source_link} |\n"
         
+        elif internal_query_type == "公司报表列表":
+            # 查询公司所有报表
+            filings_resp = requests.post(
+                f"{MCP_URL}/api/get_company_filings",
+                json={"cik": cik, "limit": 50},  # 限制最多50条
+                headers=HEADERS,
+                timeout=60
+            )
+            
+            if filings_resp.status_code != 200:
+                return result + f"❌ Server Error: HTTP {filings_resp.status_code}\n\n{filings_resp.text[:500]}"
+            
+            try:
+                filings_data = filings_resp.json()
+            except (ValueError, KeyError) as e:
+                return result + f"❌ JSON Parse Error: {str(e)}\n\n{filings_resp.text[:500]}"
+            
+            if isinstance(filings_data, dict) and filings_data.get("error"):
+                return result + f"❌ {filings_data['error']}"
+            
+            filings = filings_data.get('filings', []) if isinstance(filings_data, dict) else filings_data
+            
+            result += f"## Company Filings ({len(filings)} records)\n\n"
+            result += "| Form Type | Filing Date | Accession Number | Primary Document |\n"
+            result += "|-----------|-------------|------------------|------------------|\n"
+            
+            for filing in filings:
+                form_type = filing.get('form_type', 'N/A')
+                filing_date = filing.get('filing_date', 'N/A')
+                accession_num = filing.get('accession_number', 'N/A')
+                primary_doc = filing.get('primary_document', 'N/A')
+                
+                # 创建 SEC EDGAR 原始文档链接
+                if accession_num != 'N/A' and primary_doc != 'N/A':
+                    # 移除 accession_number 中的连字符以匹配 SEC URL 格式
+                    acc_no_clean = accession_num.replace('-', '')
+                    doc_url = f"https://www.sec.gov/Archives/edgar/data/{cik}/{acc_no_clean}/{primary_doc}"
+                    form_link = f"[{form_type}]({doc_url})"
+                    primary_doc_link = f"[{primary_doc}]({doc_url})"
+                else:
+                    form_link = form_type
+                    primary_doc_link = primary_doc
+                
+                result += f"| {form_link} | {filing_date} | {accession_num} | {primary_doc_link} |\n"
+        
         return result
         
     except requests.exceptions.RequestException as e:
@@ -317,7 +363,7 @@ with gr.Blocks(title="SEC Financial Data Query Assistant", css="""
                 scale=2
             )
             query_type = gr.Radio(
-                ["Latest Financial Data", "3-Year Trends", "5-Year Trends"],
+                ["Latest Financial Data", "3-Year Trends", "5-Year Trends", "Company Filings"],
                 label="Query Type",
                 value="Latest Financial Data",
                 scale=1
@@ -332,7 +378,7 @@ with gr.Blocks(title="SEC Financial Data Query Assistant", css="""
                 ["NVIDIA", "Latest Financial Data"],
                 ["Apple", "3-Year Trends"],
                 ["Microsoft", "5-Year Trends"],
-                ["Alibaba", "Latest Financial Data"],
+                ["Alibaba", "Company Filings"],
                 ["Tesla", "3-Year Trends"]
             ],
             inputs=[company_input, query_type],