Update app.py

app.py

@@ -1,39 +1,53 @@
 import yfinance as yf
 import streamlit as st
 import pandas as pd
-
-# Map đơn giản giữa cổ phiếu và ETF ngành (có thể mở rộng)
-sector_etf_map = {
-    'HIMS': 'XLV',    # Healthcare
-    'AAPL': 'XLK',    # Technology
-    'MSFT': 'XLK',
-    'TSLA': 'XLY',    # Consumer Discretionary
-    'JPM':  'XLF',    # Financials
-    'CVX':  'XLE',    # Energy
-    'NKE':  'XLY',
-    'NVDA': 'XLK'
-}
+import requests
 
 st.title("So sánh Outperform / Underperform theo Sector")
 
 symbols_input = st.text_input("Nhập mã cổ phiếu (cách nhau bằng dấu phẩy, ví dụ: HIMS,AAPL,TSLA):")
 period = st.selectbox("Chọn khoảng thời gian so sánh:", ['5d', '1mo', '3mo', '6mo'])
 
+# Tự động dò ETF theo sector dựa trên yfinance
+sector_to_etf = {
+    'Technology': 'XLK',
+    'Health Care': 'XLV',
+    'Financial Services': 'XLF',
+    'Consumer Cyclical': 'XLY',
+    'Energy': 'XLE',
+    'Industrials': 'XLI',
+    'Consumer Defensive': 'XLP',
+    'Utilities': 'XLU',
+    'Materials': 'XLB',
+    'Real Estate': 'XLRE',
+    'Communication Services': 'XLC'
+}
+
+def get_sector_etf(symbol):
+    try:
+        ticker = yf.Ticker(symbol)
+        info = ticker.info
+        sector = info.get("sector", None)
+        etf = sector_to_etf.get(sector, None)
+        return sector, etf
+    except:
+        return None, None
+
 if symbols_input:
     symbols = [s.strip().upper() for s in symbols_input.split(",")]
     results = []
 
     for symbol in symbols:
-        sector_etf = sector_etf_map.get(symbol, None)
-        if not sector_etf:
+        sector, etf = get_sector_etf(symbol)
+        if not etf:
             results.append({"Symbol": symbol, "Sector ETF": "Không rõ", "Kết luận": "Không phân tích được"})
             continue
 
         stock_data = yf.download(symbol, period=period)['Close']
-        etf_data = yf.download(sector_etf, period=period)['Close']
+        etf_data = yf.download(etf, period=period)['Close']
 
         if len(stock_data) < 2 or len(etf_data) < 2:
-            results.append({"Symbol": symbol, "Sector ETF": sector_etf, "Kết luận": "Thiếu dữ liệu"})
+            results.append({"Symbol": symbol, "Sector ETF": etf, "Kết luận": "Thiếu dữ liệu"})
             continue
 
         pct_stock = (stock_data[-1] - stock_data[0]) / stock_data[0] * 100
@@ -42,7 +56,7 @@ if symbols_input:
 
         results.append({
             "Symbol": symbol,
-            "Sector ETF": sector_etf,
+            "Sector ETF": etf,
             "% Stock": f"{pct_stock:.2f}%",
             "% Sector": f"{pct_sector:.2f}%",
             "Kết luận": verdict