Create universe.py

universe.py

@@ -0,0 +1,30 @@
+import pandas as pd
+import os
+
+class UniverseManager:
+    def __init__(self, csv_file="EQUITY_L.csv"):
+        self.file = csv_file
+        self.universe = self.load_universe()
+
+    def load_universe(self):
+        # Fallback if CSV is missing
+        if not os.path.exists(self.file):
+            print(f"⚠️ {self.file} not found. Using fallback tickers.")
+            return ["RELIANCE.NS", "TCS.NS", "INFY.NS", "HDFCBANK.NS"]
+        
+        try:
+            df = pd.read_csv(self.file)
+            # Clean column names (strip spaces)
+            df.columns = [c.strip() for c in df.columns]
+            
+            # Filter for Equity Series (EQ)
+            if 'SERIES' in df.columns:
+                df = df[df['SERIES'] == 'EQ']
+                
+            # Create Yahoo Finance compatible symbols
+            tickers = [f"{x}.NS" for x in df['SYMBOL'].tolist()]
+            print(f"✅ Universe loaded: {len(tickers)} stocks.")
+            return tickers
+        except Exception as e:
+            print(f"❌ Error loading universe CSV: {e}")
+            return ["RELIANCE.NS", "TCS.NS"]