Update app.py

app.py

@@ -1,10 +1,11 @@
 import os
-# 禁用 yfinance 的磁盘缓存，防止在只读环境中报错
+# 禁用 yfinance 的磁盘缓存
 os.environ['YF_NOCACHE'] = 'True'
 
 from fastapi import FastAPI
 import yfinance as yf
 import math
+import pandas as pd
 import uvicorn
 
 app = FastAPI()
@@ -26,23 +27,44 @@ def home():
 def get_stock(ticker: str):
     try:
         stock = yf.Ticker(ticker.upper())
-        info = stock.info
         
-        # 获取分析师盈利估计详情
-        # earnings_estimate 通常是一个以时间为行，指标(avg, low, high等)为列的 DataFrame
-        est = stock.earnings_estimate
+        # 1. 尝试从 stock.info 获取数据 (这是最快且最稳定的方法)
+        # yfinance 的 info 中通常直接包含 forwardEps (下一财年预测)
+        info = stock.info
         
-        # 默认初始化为 None
+        # 2. 尝试获取详细的分析师预测表
+        try:
+            est = stock.earnings_estimate
+        except Exception:
+            est = None
+
         eps_est_current = None
         eps_est_next = None
 
+        # --- 修复逻辑开始 ---
         if est is not None and not est.empty:
-            # yfinance 的索引通常是 '0y' 代表 Current Year, '1y' 代表 Next Year
-            # 安全地获取 avg 列中的对应值
-            if '0y' in est.index:
-                eps_est_current = est.loc['0y', 'avg']
-            if '1y' in est.index:
-                eps_est_next = est.loc['1y', 'avg']
+            try:
+                # 打印一下索引看看结构 (调试用，生产环境可注释)
+                # print(f"Index type: {est.index}")
+                
+                # 方法 A: 使用位置索引 iloc (更通用)
+                # 假设数据按时间升序排列，第0行是当前年，第1行是明年
+                # 注意：有时 yfinance 返回的顺序可能是反的，或者包含季度数据，
+                # 但 earnings_estimate 通常是年度数据且按时间排序。
+                
+                if len(est) > 0:
+                    eps_est_current = est.iloc[0]['avg']  # 当前财年
+                if len(est) > 1:
+                    eps_est_next = est.iloc[1]['avg']     # 下一财年
+                    
+            except Exception as e:
+                print(f"Error parsing dataframe: {e}")
+        # --- 修复逻辑结束 ---
+
+        # 3. 如果从表格获取失败，使用 info 中的备用数据
+        # forwardEps 通常代表下一财年的预测 EPS
+        if eps_est_next is None and info:
+            eps_est_next = info.get('forwardEps')
 
         if not info or len(info) < 5:
             return {"error": "No data found", "ticker": ticker}
@@ -51,16 +73,17 @@ def get_stock(ticker: str):
             "ticker": ticker.upper(),
             "pe_static": clean_num(info.get('trailingPE')),
             "pe_dynamic": clean_num(info.get('forwardPE')),
-            # info 里的 trailingEps 是过去一年的实际值
-            #"eps_trailing": clean_num(info.get('trailingEps')), 
-            # 从 earnings_estimate 中获取的分析师平均预测值
+            
+            # 优先显示从表格获取的数据，如果没有则显示 info 中的数据
             "eps_est_avg_current_year": clean_num(eps_est_current),
             "eps_est_avg_next_year": clean_num(eps_est_next),
+            
+            # 调试信息：告诉你数据来源是 info 还是 table
+            "debug_note": "Data from table" if eps_est_current is not None else "Data from info/fallback",
             "source": "hf_space_v1"
         }
     except Exception as e:
         return {"error": str(e)}
 
 if __name__ == "__main__":
-    # Hugging Face 会自动分配端口，通常是 7860
     uvicorn.run(app, host="0.0.0.0", port=7860)