Upload 3 files

app.py

@@ -91,22 +91,25 @@ def query_financial_data(company_name, query_type):
             if isinstance(metrics, dict) and metrics.get("error"):
                 return result + f"❌ {metrics['error']}"
             
-            result += f"## Financial Trends ({metrics.get('count', 0)} periods)\n\n"
+            result += f"## 3-Year Financial Trends ({metrics.get('count', 0)} periods)\n\n"
             
-            # 只显示年度数据
-            annual_data = [m for m in metrics.get('metrics', []) if 'Q' not in m.get('period', '')][:3]
+            # 显示所有数据（包括年度和季度）
+            all_data = metrics.get('metrics', [])
             
-            result += "| Period | Revenue (B) | Net Income (B) | EPS | Operating Expenses (B) | Operating Cash Flow (B) |\n"
-            result += "|--------|-------------|----------------|-----|------------------------|-------------------------|\n"
+            result += "| Period | Revenue (B) | Net Income (B) | EPS | Operating Expenses (B) | Operating Cash Flow (B) | Source Form |\n"
+            result += "|--------|-------------|----------------|-----|------------------------|-------------------------|-------------|\n"
             
-            for m in annual_data:
+            for m in all_data:
                 period = m.get('period', 'N/A')
                 rev = m.get('total_revenue', 0) / 1e9
                 inc = m.get('net_income', 0) / 1e9
                 eps = m.get('earnings_per_share', 0)
                 opex = m.get('operating_expenses', 0) / 1e9
                 ocf = m.get('operating_cash_flow', 0) / 1e9
-                result += f"| FY{period} | ${rev:.2f}B | ${inc:.2f}B | ${eps:.2f} | ${opex:.2f}B | ${ocf:.2f}B |\n"
+                source = m.get('source_form', 'N/A')
+                # 区分年度和季度
+                period_prefix = "FY" if 'Q' not in period else ""
+                result += f"| {period_prefix}{period} | ${rev:.2f}B | ${inc:.2f}B | ${eps:.2f} | ${opex:.2f}B | ${ocf:.2f}B | {source} |\n"
         
         elif query_type == "5年趋势":
             metrics_resp = requests.post(
@@ -127,20 +130,24 @@ def query_financial_data(company_name, query_type):
             if isinstance(metrics, dict) and metrics.get("error"):
                 return result + f"❌ {metrics['error']}"
             
-            annual_data = [m for m in metrics.get('metrics', []) if 'Q' not in m.get('period', '')][:5]
+            # 显示所有数据（包括年度和季度）
+            all_data = metrics.get('metrics', [])
             
-            result += f"## 5-Year Financial Trends\n\n"
-            result += "| Period | Revenue (B) | Net Income (B) | EPS | Operating Expenses (B) | Operating Cash Flow (B) |\n"
-            result += "|--------|-------------|----------------|-----|------------------------|-------------------------|\n"
+            result += f"## 5-Year Financial Trends ({metrics.get('count', 0)} periods)\n\n"
+            result += "| Period | Revenue (B) | Net Income (B) | EPS | Operating Expenses (B) | Operating Cash Flow (B) | Source Form |\n"
+            result += "|--------|-------------|----------------|-----|------------------------|-------------------------|-------------|\n"
             
-            for m in annual_data:
+            for m in all_data:
                 period = m.get('period', 'N/A')
                 rev = m.get('total_revenue', 0) / 1e9
                 inc = m.get('net_income', 0) / 1e9
                 eps = m.get('earnings_per_share', 0)
                 opex = m.get('operating_expenses', 0) / 1e9
                 ocf = m.get('operating_cash_flow', 0) / 1e9
-                result += f"| FY{period} | ${rev:.2f}B | ${inc:.2f}B | ${eps:.2f} | ${opex:.2f}B | ${ocf:.2f}B |\n"
+                source = m.get('source_form', 'N/A')
+                # 区分年度和季度
+                period_prefix = "FY" if 'Q' not in period else ""
+                result += f"| {period_prefix}{period} | ${rev:.2f}B | ${inc:.2f}B | ${eps:.2f} | ${opex:.2f}B | ${ocf:.2f}B | {source} |\n"
         
         return result