Update app.py

app.py

@@ -46,21 +46,24 @@ if symbols_input:
         stock_data = yf.download(symbol, 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": etf, "Kết luận": "Thiếu dữ liệu"})
+        if stock_data is None or stock_data.empty or etf_data is None or etf_data.empty:
+            results.append({"Symbol": symbol, "Sector ETF": etf, "Kết luận": "Không có dữ liệu"})
             continue
 
-        pct_stock = (stock_data[-1] - stock_data[0]) / stock_data[0] * 100
-        pct_sector = (etf_data[-1] - etf_data[0]) / etf_data[0] * 100
-        verdict = "Outperform" if pct_stock > pct_sector else "Underperform"
+        if len(stock_data) >= 2 and len(etf_data) >= 2:
+            pct_stock = (stock_data.iloc[-1] - stock_data.iloc[0]) / stock_data.iloc[0] * 100
+            pct_sector = (etf_data.iloc[-1] - etf_data.iloc[0]) / etf_data.iloc[0] * 100
+            verdict = "Outperform" if pct_stock > pct_sector else "Underperform"
 
-        results.append({
-            "Symbol": symbol,
-            "Sector ETF": etf,
-            "% Stock": f"{pct_stock:.2f}%",
-            "% Sector": f"{pct_sector:.2f}%",
-            "Kết luận": verdict
-        })
+            results.append({
+                "Symbol": symbol,
+                "Sector ETF": etf,
+                "% Stock": f"{pct_stock:.2f}%",
+                "% Sector": f"{pct_sector:.2f}%",
+                "Kết luận": verdict
+            })
+        else:
+            results.append({"Symbol": symbol, "Sector ETF": etf, "Kết luận": "Dữ liệu không đủ"})
 
     df = pd.DataFrame(results)
     st.dataframe(df)