Create app.py

app.py

@@ -0,0 +1,147 @@
+import gradio as gr
+import requests
+
+MCP_URL = "https://jc321-easyreportdatemcp.hf.space"
+
+def query_financial_data(company_name, query_type):
+    """查询财务数据的主函数"""
+    
+    if not company_name:
+        return "请输入公司名称或股票代码"
+    
+    try:
+        # 搜索公司
+        search_resp = requests.post(
+            f"{MCP_URL}/api/advanced_search",
+            json={"company_input": company_name},
+            timeout=30
+        )
+        company = search_resp.json()
+        
+        if company.get("error"):
+            return f"❌ 错误: {company['error']}"
+        
+        result = f"# {company['name']}\n\n"
+        result += f"**股票代码**: {company.get('tickers', ['N/A'])[0] if company.get('tickers') else 'N/A'}\n"
+        result += f"**行业**: {company.get('sic_description', 'N/A')}\n\n---\n\n"
+        
+        cik = company['cik']
+        
+        # 根据查询类型获取数据
+        if query_type == "最新财务数据":
+            data_resp = requests.post(
+                f"{MCP_URL}/api/get_latest_financial_data",
+                json={"cik": cik},
+                timeout=30
+            )
+            data = data_resp.json()
+            
+            if data.get("error"):
+                return result + f"❌ {data['error']}"
+            
+            result += f"## 财年 {data['period']}\n\n"
+            result += f"- **总收入**: ${data['total_revenue']:,.0f} (${data['total_revenue']/1e9:.2f}B)\n"
+            result += f"- **净利润**: ${data['net_income']:,.0f} (${data['net_income']/1e9:.2f}B)\n"
+            result += f"- **每股收益**: ${data['earnings_per_share']:.2f}\n"
+            result += f"- **来源**: {data['source_form']}\n"
+        
+        elif query_type == "3年趋势":
+            metrics_resp = requests.post(
+                f"{MCP_URL}/api/extract_financial_metrics",
+                json={"cik": cik, "years": 3},
+                timeout=60
+            )
+            metrics = metrics_resp.json()
+            
+            if metrics.get("error"):
+                return result + f"❌ {metrics['error']}"
+            
+            result += f"## 财务趋势 (共 {metrics['count']} 个期间)\n\n"
+            
+            # 只显示年度数据
+            annual_data = [m for m in metrics['metrics'] if 'Q' not in m['period']][:3]
+            
+            result += "| 期间 | 收入(十亿) | 净利润(十亿) | EPS |\n"
+            result += "|------|-----------|------------|-----|\n"
+            
+            for m in annual_data:
+                period = m['period']
+                rev = m.get('total_revenue', 0) / 1e9
+                inc = m.get('net_income', 0) / 1e9
+                eps = m.get('earnings_per_share', 0)
+                result += f"| FY{period} | ${rev:.2f}B | ${inc:.2f}B | ${eps:.2f} |\n"
+        
+        elif query_type == "5年趋势":
+            metrics_resp = requests.post(
+                f"{MCP_URL}/api/extract_financial_metrics",
+                json={"cik": cik, "years": 5},
+                timeout=60
+            )
+            metrics = metrics_resp.json()
+            
+            if metrics.get("error"):
+                return result + f"❌ {metrics['error']}"
+            
+            annual_data = [m for m in metrics['metrics'] if 'Q' not in m['period']][:5]
+            
+            result += f"## 5年财务趋势\n\n"
+            result += "| 期间 | 收入 | 净利润 | EPS |\n"
+            result += "|------|------|--------|-----|\n"
+            
+            for m in annual_data:
+                rev = m.get('total_revenue', 0) / 1e9
+                inc = m.get('net_income', 0) / 1e9
+                eps = m.get('earnings_per_share', 0)
+                result += f"| FY{m['period']} | ${rev:.2f}B | ${inc:.2f}B | ${eps:.2f} |\n"
+        
+        return result
+        
+    except Exception as e:
+        return f"❌ 发生错误: {str(e)}"
+
+# 创建 Gradio 界面
+with gr.Blocks(title="SEC 财务数据查询助手", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🤖 SEC 财务数据查询助手")
+    gr.Markdown("通过调用 MCP Server 查询美国上市公司的财务数据")
+    
+    with gr.Row():
+        company_input = gr.Textbox(
+            label="公司名称或股票代码",
+            placeholder="例如: NVIDIA, Apple, TSMC, AAPL",
+            scale=2
+        )
+        query_type = gr.Radio(
+            ["最新财务数据", "3年趋势", "5年趋势"],
+            label="查询类型",
+            value="最新财务数据",
+            scale=1
+        )
+    
+    submit_btn = gr.Button("🔍 查询", variant="primary", size="lg")
+    output = gr.Markdown(label="查询结果")
+    
+    # 示例
+    gr.Examples(
+        examples=[
+            ["NVIDIA", "最新财务数据"],
+            ["Apple", "3年趋势"],
+            ["Microsoft", "5年趋势"],
+            ["TSMC", "最新财务数据"],
+            ["Tesla", "3年趋势"]
+        ],
+        inputs=[company_input, query_type],
+        outputs=output,
+        fn=query_financial_data,
+        cache_examples=False
+    )
+    
+    submit_btn.click(
+        fn=query_financial_data,
+        inputs=[company_input, query_type],
+        outputs=output
+    )
+    
+    gr.Markdown("---")
+    gr.Markdown(f"**数据来源**: SEC EDGAR | **MCP Server**: `{MCP_URL}`")
+
+demo.launch()