Update app.py

app.py

@@ -25,10 +25,25 @@ def home():
 @app.get("/stock/{ticker}")
 def get_stock(ticker: str):
     try:
-        # yfinance 在 HF 的环境下表现非常稳定
         stock = yf.Ticker(ticker.upper())
         info = stock.info
         
+        # 获取分析师盈利估计详情
+        # earnings_estimate 通常是一个以时间为行，指标(avg, low, high等)为列的 DataFrame
+        est = stock.earnings_estimate
+        
+        # 默认初始化为 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']
+
         if not info or len(info) < 5:
             return {"error": "No data found", "ticker": ticker}
 
@@ -36,12 +51,13 @@ def get_stock(ticker: str):
             "ticker": ticker.upper(),
             "pe_static": clean_num(info.get('trailingPE')),
             "pe_dynamic": clean_num(info.get('forwardPE')),
-            "eps_current_year": clean_num(info.get('trailingEps')),
-            "eps_next_year": clean_num(info.get('forwardEps')),
+            # info 里的 trailingEps 是过去一年的实际值
+            "eps_trailing": clean_num(info.get('trailingEps')), 
+            # 从 earnings_estimate 中获取的分析师平均预测值
+            "eps_est_avg_current_year": clean_num(eps_est_current),
+            "eps_est_avg_next_year": clean_num(eps_est_next),
             "source": "hf_space_v1"
         }
-    except Exception as e:
-        return {"error": str(e)}
 
 if __name__ == "__main__":
     # Hugging Face 会自动分配端口，通常是 7860