Update trade_analysis/data.py

trade_analysis/data.py

@@ -54,52 +54,48 @@ class UnifiedDataProvider:
             return data
         return None
 
+    
     async def fetch_multi_timeframe_stock_data(self, symbol: str) -> dict:
         """
-        Fetches OHLCV data from Finnhub for multiple timeframes.
-        This completely replaces the unreliable yfinance implementation.
+        REWRITTEN FOR FREE TIER: Fetches the current day's quote data from Finnhub.
+        Historical candle data is a premium feature.
         """
-        print(f"Fetching Finnhub OHLCV for {symbol}...")
+        print(f"Fetching Finnhub REAL-TIME QUOTE for {symbol}...")
         dfs = {}
-        
-        # Mapping from our names to Finnhub's resolution codes and time periods
-        timeframes = {
-            "daily": ("D", 90),    # Daily data for the last year
-            # "hourly": ("60", 28),   # Hourly data for the last 2 years
-            # "15m": ("15", 7),       # 15-minute data for the last 60 days
-            # "5m": ("5", 2),
-        }
-
-        base_url = "https://finnhub.io/api/v1/stock/candle"
-
-        for name, (resolution, days_back) in timeframes.items():
-            end_ts = int(datetime.now().timestamp())
-            start_ts = int((datetime.now() - timedelta(days=days_back)).timestamp())
-
-            params = {
-                "symbol": symbol,
-                "resolution": resolution,
-                "from": start_ts,
-                "to": end_ts,
-                "token": config.FINNHUB_KEY
-            }
-            
-            try:
-                res = await self.client.get(base_url, params=params)
-                res.raise_for_status()
-                data = res.json()
-                processed_df = _process_finnhub_data(data, symbol)
+        base_url = "https://finnhub.io/api/v1/quote"
+        params = {"symbol": symbol, "token": config.FINNHUB_KEY}
+    
+        try:
+            res = await self.client.get(base_url, params=params)
+            res.raise_for_status()
+            data = res.json()
+    
+            if data and 'c' in data and data['c'] != 0:
+                # Create a single-row DataFrame
+                df = pd.DataFrame([{
+                    'Open': data['o'],
+                    'High': data['h'],
+                    'Low': data['l'],
+                    'Close': data['c'],
+                    # Volume is not in the quote endpoint, so we put a placeholder
+                    'Volume': 0 
+                }])
+                # Use the current time for the index
+                df.index = pd.to_datetime([datetime.now(tz=pytz.UTC)])
                 
-                if not processed_df.empty:
-                    print(f"  - Fetched and processed {len(processed_df)} points for {name} interval from Finnhub.")
-                    dfs[name] = processed_df
-                else:
-                    dfs[name] = pd.DataFrame()
-            except (httpx.RequestError, httpx.HTTPStatusError) as e:
-                print(f"ERROR: Finnhub OHLCV request failed for {symbol} ({name}): {e}")
-                dfs[name] = pd.DataFrame()
+                # Since we only have one row, we can't do multi-timeframe.
+                # We will return this single row for the 'daily' key.
+                dfs["daily"] = df
+                print(f"  - Fetched real-time quote for {symbol} from Finnhub.")
+            else:
+                dfs["daily"] = pd.DataFrame()
+    
+        except (httpx.RequestError, httpx.HTTPStatusError) as e:
+            print(f"ERROR: Finnhub Quote request failed for {symbol}: {e}")
+            dfs["daily"] = pd.DataFrame()
+            
         return dfs
-
+    
     async def fetch_news(self, symbol: str, days: int = 3) -> tuple:
         """Fetches news from Finnhub, with local file fallback."""
         local_data = self._load_local_data(symbol)