Rename data_loader.py to polygon_loader.py

data_loader.py

@@ -1,9 +0,0 @@
-import pandas as pd
-
-def fetch_ohlcv(symbol="AAPL", interval="1d", period="7d"):
-    # Static fallback data for testing
-    data = {
-        'Close': [195.3, 197.1, 198.8, 199.5, 200.4, 201.1, 199.9, 198.5, 197.2, 196.8],
-    }
-    df = pd.DataFrame(data)
-    return df

polygon_loader.py

@@ -0,0 +1,37 @@
+import requests
+import pandas as pd
+from datetime import datetime, timedelta
+
+API_KEY = "HviulInrP_JHBbZQdm7EZdppph2iTwBJ"
+
+def fetch_ohlcv(symbol="AAPL", interval="minute", limit=100):
+    end_time = datetime.utcnow()
+    start_time = end_time - timedelta(minutes=limit)
+
+    url = f"https://api.polygon.io/v2/aggs/ticker/{symbol}/range/1/minute/{start_time.date()}/{end_time.date()}"
+
+    params = {
+        "adjusted": "true",
+        "sort": "asc",
+        "limit": limit,
+        "apiKey": API_KEY,
+    }
+
+    try:
+        res = requests.get(url, params=params)
+        res.raise_for_status()
+        data = res.json()
+
+        if "results" not in data:
+            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
+    except Exception as e:
+        print("Polygon error:", e)
+        return None