import sqlite3
import pandas as pd
import os

universe = "AAPL,MSFT,NVDA,AMD,META,GOOG,TSLA,AMZN,JPM,GS,V,JNJ,PFE,UNH,XOM,CVX,WMT,KO,CAT,GE,DIS,NFLX,SPY,QQQ,IWM,INTC,BAC,SOFI,PLTR,UBER,SHOP,PYPL,MA,CSCO,PEP,MRK,BRK.B,LLY,AVGO,HD,PG,COST,ABBV,CRM,MCD,LIN,IONQ,NBIS,VRT".split(",")

conn = sqlite3.connect('data/market_data.db')
selected = []
for sym in universe:
    try:
        df = pd.read_sql_query(f"SELECT close FROM bars_1d WHERE symbol='{sym}' ORDER BY timestamp DESC LIMIT 1", conn)
        if not df.empty:
            price = df['close'].iloc[0]
            if 50 <= price <= 140:
                selected.append(sym)
    except Exception as e:
        pass

new_univ = ",".join(selected)
print("New Universe:", new_univ)

with open('.env', 'r', encoding='utf-8') as f:
    lines = f.readlines()

with open('.env', 'w', encoding='utf-8') as f:
    for line in lines:
        if line.startswith('UNIVERSE="'):
            f.write(f'UNIVERSE="{new_univ}"\n')
        else:
            f.write(line)