Update app.py

app.py

@@ -210,10 +210,25 @@ with st.expander("Train with Indicators (RSI, EMA, Stochastic)"):
                     )
             if df.empty:
                 st.error("No data returned. Try a shorter lookback for intraday (e.g., 30d/60d) or use Interval=1d.")
-                st.stop()
-
-            # Keep only needed cols and drop NaNs
-            df = df[["Close", "High", "Low"]].dropna().copy()
+                st.stop()            # Handle MultiIndex columns (yfinance can return 2-level columns)
+            if isinstance(df.columns, pd.MultiIndex):
+                try:
+                    sym = df.columns.get_level_values(1).unique()[0]
+                    df = df.xs(sym, axis=1, level=1)
+                except Exception:
+                    # Fallback: flatten by taking the top-level name (Close/High/Low)
+                    df.columns = [c[0] for c in df.columns.to_flat_index()]
+
+            # Keep only needed cols
+            df = df[["Close", "High", "Low"]].copy()
+
+            # Ensure each column is 1-D (avoid (N,1) arrays)
+            for _c in ["Close", "High", "Low"]:
+                if isinstance(df[_c], pd.DataFrame):
+                    df[_c] = df[_c].iloc[:, 0]
+                df[_c] = pd.Series(np.asarray(df[_c]).reshape(-1), index=df.index)
+
+            df = df.dropna()
 
             # 2) Indicators (helpers above)
             df["rsi14"] = rsi(df["Close"], 14)