Upload data_fetcher.py

src/data_fetcher.py

@@ -25,25 +25,40 @@ class DataFetcher:
     def fetch_market_data(self, days=50):
         """
         Fetches raw OHLCV and VIX data from Yahoo Finance.
+        Falls back to local CSV in the data/ folder if Yahoo blocks the server IP.
         """
-        print(f"📡 Fetching last {days} days of {self.ticker} and {self.vix_ticker}...")
+        print(f"📡 Attempting to fetch last {days} days of {self.ticker} and {self.vix_ticker}...")
         
-        # Download Index Data
-        df = yf.download(self.ticker, period=f"{days}d", interval="1d", progress=False)
-        # Download VIX
-        df_vix = yf.download(self.vix_ticker, period=f"{days}d", interval="1d", progress=False)
+        try:
+            # 1. TRY TO FETCH LIVE DATA
+            #df = yf.download(self.ticker, period=f"{days}d", interval="1d", progress=False)
+            #df_vix = yf.download(self.vix_ticker, period=f"{days}d", interval="1d", progress=False)
 
-        # Handle yfinance MultiIndex columns if they exist
-        if isinstance(df.columns, pd.MultiIndex):
-            df.columns = df.columns.get_level_values(0)
-        if isinstance(df_vix.columns, pd.MultiIndex):
-            df_vix.columns = df_vix.columns.get_level_values(0)
+            # Handle yfinance MultiIndex columns if they exist
+            if isinstance(df.columns, pd.MultiIndex):
+                df.columns = df.columns.get_level_values(0)
+            if isinstance(df_vix.columns, pd.MultiIndex):
+                df_vix.columns = df_vix.columns.get_level_values(0)
 
-        # Merge VIX and ensure data is clean
-        df['VIX'] = df_vix['Close']
-        df = df.ffill() 
-        
-        return df
+            df['VIX'] = df_vix['Close']
+            df = df.ffill() 
+            
+            # If the dataframe is empty (Yahoo stealth-blocked us), force an error
+            if df.empty:
+                raise ValueError("Yahoo Finance returned empty data.")
+                
+            return df
+            
+        except Exception as e:
+            # 2. FALLBACK TO LOCAL CSV IF BLOCKED
+            print(f"⚠️ Live fetch failed ({e}). Loading backup data from data/ folder...")
+            
+            # Load the CSV from your new data folder
+            backup_path = "data/market_data_backup.csv"
+            df_backup = pd.read_csv(backup_path, index_col=0, parse_dates=True)
+            
+            # Return only the requested number of days
+            return df_backup.tail(days)
 
     # 🛡️ STREAMLIT CACHE: Ignores '_self' so it doesn't try to hash the Finnhub client.
     # ttl=3600 caches the news for 1 hour so repeated button clicks load instantly.