working locally

app.py

@@ -115,29 +115,22 @@ else:
     st.sidebar.markdown("---")
     st.sidebar.caption("Tip: Choose a wide date range for smoother moving averages and richer AI insights.")
 
-    def load_stock_data(symbol: str, start_dt, end_dt):
-        """Download data and flatten any MultiIndex columns from yfinance."""
-        try:
-            data = yf.download(symbol, start=start_dt, end=end_dt, auto_adjust=False, progress=False)
-        except Exception:
-            st.error("Error fetching stock data. Please check the ticker symbol and date range.")
-            st.stop()
-
-        if data.empty:
-            st.error("No data returned. Try widening the date range or verifying the ticker.")
-            st.stop()
-
-        if isinstance(data.columns, pd.MultiIndex):
-            price_like = {"Open", "High", "Low", "Close", "Adj Close", "Volume"}
-            level0_has_price = any(col in price_like for col in data.columns.get_level_values(0))
-            columns = data.columns.get_level_values(0 if level0_has_price else 1)
-            data = data.copy()
-            data.columns = columns
-
-        data.reset_index(inplace=True)  # Reset index to ensure 'Date' becomes a column
-        if "Date" in data.columns:
-            data["Date"] = pd.to_datetime(data["Date"]).dt.tz_localize(None)
-        return data
+    def load_stock_data(symbol: str, start_dt, end_dt):
+        """Download data with a flat schema to avoid Plotly/MultiIndex issues on some runtimes."""
+        try:
+            ticker = yf.Ticker(symbol)
+            data = ticker.history(start=start_dt, end=end_dt, actions=False, auto_adjust=False)
+        except Exception:
+            st.error("Error fetching stock data. Please check the ticker symbol and date range.")
+            st.stop()
+
+        if data.empty:
+            st.error("No data returned. Try widening the date range or verifying the ticker.")
+            st.stop()
+
+        data = data.reset_index()  # Ensure Date is a column
+        data["Date"] = pd.to_datetime(data["Date"]).dt.tz_localize(None)
+        return data
 
     # Fetch stock data from Yahoo Finance
     df = load_stock_data(ticker_symbol, start_date, end_date)