Update polygon_loader.py

polygon_loader.py

@@ -2,18 +2,19 @@ import requests
 import pandas as pd
 from datetime import datetime, timedelta
 
-API_KEY = "HviulInrP_JHBbZQdm7EZdppph2iTwBJ"
+API_KEY = "HviulInrP_JHBbZQdm7EZdppph2iTwBJ"  # ← Replace this
 
-def fetch_ohlcv(symbol="AAPL", interval="minute", limit=100):
-    end_time = datetime.utcnow()
-    start_time = end_time - timedelta(minutes=limit)
+def fetch_ohlcv(symbol="AAPL"):
+    # Use a fixed recent date that is guaranteed to have data
+    start = (datetime.utcnow() - timedelta(days=2)).strftime('%Y-%m-%d')
+    end = (datetime.utcnow() - timedelta(days=1)).strftime('%Y-%m-%d')
 
-    url = f"https://api.polygon.io/v2/aggs/ticker/{symbol}/range/1/minute/{start_time.date()}/{end_time.date()}"
+    url = f"https://api.polygon.io/v2/aggs/ticker/{symbol}/range/1/minute/{start}/{end}"
 
     params = {
         "adjusted": "true",
         "sort": "asc",
-        "limit": limit,
+        "limit": 500,
         "apiKey": API_KEY,
     }
 
@@ -22,16 +23,14 @@ def fetch_ohlcv(symbol="AAPL", interval="minute", limit=100):
         res.raise_for_status()
         data = res.json()
 
-        if "results" not in data:
+        if "results" not in data or not data["results"]:
             return None
 
         df = pd.DataFrame(data["results"])
         df["t"] = pd.to_datetime(df["t"], unit="ms")
         df.set_index("t", inplace=True)
         df.rename(columns={"o": "Open", "h": "High", "l": "Low", "c": "Close", "v": "Volume"}, inplace=True)
-        df = df[["Open", "High", "Low", "Close", "Volume"]]
-
-        return df
+        return df[["Open", "High", "Low", "Close", "Volume"]]
     except Exception as e:
-        print("Polygon error:", e)
+        print("Polygon API error:", e)
         return None