Update core/data.py

core/data.py

@@ -41,8 +41,15 @@ def load_data(data_src="yahoo", ticker="AAPL", start="2020-01-01", end="2023-01-
             df = df.rename(columns={'Close': 'value'})
         else:
             df = yf.download(ticker, start=start, end=end, interval=interval, progress=False, auto_adjust=False)
-            df = df.rename(columns={'Close': 'value'})
-            df['Date'] = df.index
+            # Normalize potential MultiIndex columns from yfinance
+            if isinstance(df.columns, pd.MultiIndex):
+                df.columns = [c[0] if isinstance(c, tuple) else c for c in df.columns]
+            # Ensure expected columns
+            rename_map = {'Adj Close': 'Adj Close', 'Close': 'value'}
+            if 'Close' in df.columns:
+                df = df.rename(columns={'Close': 'value'})
+            if 'Date' not in df.columns:
+                df['Date'] = df.index
             df = df.reset_index(drop=True)
 
         if interval in ['1m', '5m', '15m', '30m', '60m'] and alpha_api_key: