Upload app.py

app.py

@@ -171,13 +171,34 @@ def query_financial_data(company_name, query_type):
             # 显示所有数据（包括年度和季度）
             all_data = metrics.get('metrics', [])
             
+            # 去重：根据period和source_form去重
+            seen = set()
+            unique_data = []
+            for m in all_data:
+                key = (m.get('period', 'N/A'), m.get('source_form', 'N/A'))
+                if key not in seen:
+                    seen.add(key)
+                    unique_data.append(m)
+            
             # 按期间降序排序，确保显示最近的3年数据
-            all_data = sorted(all_data, key=lambda x: x.get('period', '0000'), reverse=True)
+            # 使用更智能的排序：先按年份，再按是否是季度
+            def sort_key(x):
+                period = x.get('period', '0000')
+                # 提取年份（前4位）
+                year = period[:4] if len(period) >= 4 else '0000'
+                # 如果有Q，提取季度号，否则设为5（让FY排在Q后面）
+                if 'Q' in period:
+                    quarter = period[period.index('Q')+1] if period.index('Q')+1 < len(period) else '0'
+                    return (year, 0, 5 - int(quarter))  # Q4, Q3, Q2, Q1
+                else:
+                    return (year, 1, 0)  # FY 排在同年的所有Q之后
+            
+            unique_data = sorted(unique_data, key=sort_key, reverse=True)
             
             result += "| Period | Revenue (B) | Net Income (B) | EPS | Operating Expenses (B) | Operating Cash Flow (B) | Source Form |\n"
             result += "|--------|-------------|----------------|-----|------------------------|-------------------------|-------------|\n"
             
-            for m in all_data:
+            for m in unique_data:
                 period = m.get('period', 'N/A')
                 rev = (m.get('total_revenue') or 0) / 1e9
                 inc = (m.get('net_income') or 0) / 1e9
@@ -187,11 +208,17 @@ def query_financial_data(company_name, query_type):
                 source_form = m.get('source_form', 'N/A')
                 source_url = m.get('source_url', None)  # 从后端获取URL
                 
-                # 区分年度和季度
-                period_prefix = "FY" if 'Q' not in period else ""
+                # 区分年度和季度，修复双重FY前缀问题
+                if 'Q' in period:
+                    # 季度数据，不添加前缀
+                    display_period = period
+                else:
+                    # 年度数据，只在没有FY的情况下添加
+                    display_period = period if period.startswith('FY') else f"FY{period}"
+                
                 source_link = create_source_link(source_form, source_url)
                 
-                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"
+                result += f"| {display_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 == "5年趋势":
             metrics_resp = requests.post(
@@ -215,14 +242,35 @@ def query_financial_data(company_name, query_type):
             # 显示所有数据（包括年度和季度）
             all_data = metrics.get('metrics', [])
             
+            # 去重：根据period和source_form去重
+            seen = set()
+            unique_data = []
+            for m in all_data:
+                key = (m.get('period', 'N/A'), m.get('source_form', 'N/A'))
+                if key not in seen:
+                    seen.add(key)
+                    unique_data.append(m)
+            
             # 按期间降序排序，确保显示最近的5年数据
-            all_data = sorted(all_data, key=lambda x: x.get('period', '0000'), reverse=True)
+            # 使用更智能的排序：先按年份，再按是否是季度
+            def sort_key(x):
+                period = x.get('period', '0000')
+                # 提取年份（前4位）
+                year = period[:4] if len(period) >= 4 else '0000'
+                # 如果有Q，提取季度号，否则设为5（让FY排在Q后面）
+                if 'Q' in period:
+                    quarter = period[period.index('Q')+1] if period.index('Q')+1 < len(period) else '0'
+                    return (year, 0, 5 - int(quarter))  # Q4, Q3, Q2, Q1
+                else:
+                    return (year, 1, 0)  # FY 排在同年的所有Q之后
+            
+            unique_data = sorted(unique_data, key=sort_key, reverse=True)
             
             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 all_data:
+            for m in unique_data:
                 period = m.get('period', 'N/A')
                 rev = (m.get('total_revenue') or 0) / 1e9
                 inc = (m.get('net_income') or 0) / 1e9
@@ -232,11 +280,17 @@ def query_financial_data(company_name, query_type):
                 source_form = m.get('source_form', 'N/A')
                 source_url = m.get('source_url', None)  # 从后端获取URL
                 
-                # 区分年度和季度
-                period_prefix = "FY" if 'Q' not in period else ""
+                # 区分年度和季度，修复双重FY前缀问题
+                if 'Q' in period:
+                    # 季度数据，不添加前缀
+                    display_period = period
+                else:
+                    # 年度数据，只在没有FY的情况下添加
+                    display_period = period if period.startswith('FY') else f"FY{period}"
+                
                 source_link = create_source_link(source_form, source_url)
                 
-                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"
+                result += f"| {display_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 == "公司报表列表":
             # 查询公司所有报表