# config.py
import os
from dotenv import load_dotenv

load_dotenv()

# API Keys
SWARMS_API_KEY = os.getenv("SWARMS_API_KEY")
FRED_API_KEY = os.getenv("FRED_API_KEY")

# API URLs
SWARMS_BASE_URL = "https://api.swarms.world"
FRED_BASE_URL = "https://api.stlouisfed.org/fred/series/observations"

# Default Values
DEFAULT_REPO_RATE = 6.5
DEFAULT_INFLATION_RATE = 5.7

# Nifty 50 Companies (could be moved to a separate file if too large)
NIFTY50_COMPANIES = {
    'ADANIENT.NS': 'Adani Enterprises', 'ADANIPORTS.NS': 'Adani Ports',
    'APOLLOHOSP.NS': 'Apollo Hospitals', 'ASIANPAINT.NS': 'Asian Paints',
    'AXISBANK.NS': 'Axis Bank', 'BAJAJ-AUTO.NS': 'Bajaj Auto',
    'BAJFINANCE.NS': 'Bajaj Finance', 'BAJAJFINSV.NS': 'Bajaj Finserv',
    'BEL.NS': 'Bharat Electronics', 'BHARTIARTL.NS': 'Bharti Airtel',
    'CIPLA.NS': 'Cipla', 'COALINDIA.NS': 'Coal India', 'DRREDDY.NS': 'Dr Reddy Labs',
    'EICHERMOT.NS': 'Eicher Motors', 'GRASIM.NS': 'Grasim Industries',
    'HCLTECH.NS': 'HCL Technologies', 'HDFCBANK.NS': 'HDFC Bank',
    'HDFCLIFE.NS': 'HDFC Life', 'HEROMOTOCO.NS': 'Hero MotoCorp',
    'HINDALCO.NS': 'Hindalco', 'HINDUNILVR.NS': 'Hindustan Unilever',
    'ICICIBANK.NS': 'ICICI Bank', 'INDUSINDBK.NS': 'IndusInd Bank',
    'INFY.NS': 'Infosys', 'ITC.NS': 'ITC', 'JIOFIN.NS': 'Jio Financial',
    'JSWSTEEL.NS': 'JSW Steel', 'KOTAKBANK.NS': 'Kotak Mahindra Bank',
    'LT.NS': 'Larsen & Toubro', 'M&M.NS': 'Mahindra & Mahindra',
    'MARUTI.NS': 'Maruti Suzuki', 'NESTLEIND.NS': 'Nestle India',
    'NTPC.NS': 'NTPC', 'ONGC.NS': 'ONGC', 'POWERGRID.NS': 'Power Grid Corp',
    'RELIANCE.NS': 'Reliance Industries', 'SBILIFE.NS': 'SBI Life',
    'SHRIRAMFIN.NS': 'Shriram Finance', 'SBIN.NS': 'State Bank of India',
    'SUNPHARMA.NS': 'Sun Pharma', 'TATACONSUM.NS': 'Tata Consumer',
    'TCS.NS': 'Tata Consultancy Services', 'TATAMOTORS.NS': 'Tata Motors',
    'TATASTEEL.NS': 'Tata Steel', 'TECHM.NS': 'Tech Mahindra',
    'TITAN.NS': 'Titan Company', 'TRENT.NS': 'Trent',
    'ULTRACEMCO.NS': 'UltraTech Cement', 'WIPRO.NS': 'Wipro',
    # Newly added stocks
    'ADANIGREEN.NS': 'Adani Green Energy',
    'AMBUJACEM.NS': 'Ambuja Cements',
    'AUBANK.NS': 'AU Small Finance Bank',
    'BANKINDIA.NS': 'Bank of India',
    'BSE.NS': 'Bombay Stock Exchange',
    'CENTRALBK.NS': 'Central Bank of India',
    'DMART.NS': 'Avenue Supermarts (DMart)',
    'FEDERALBNK.NS': 'Federal Bank',
    'HINDPETRO.NS': 'Hindustan Petroleum',
    'INDIGO.NS': 'InterGlobe Aviation (IndiGo)',
    'IOC.NS': 'Indian Oil Corporation',
    'IRCTC.NS': 'Indian Railway Catering and Tourism Corporation',
    'LALPATHLAB.NS': 'Lal PathLabs',
    'MAXFINA.NS': 'Max Financial Services',
    'NAUKRI.NS': 'Info Edge (Naukri.com)',
    'NMDC.NS': 'NMDC',
    'PETRONET.NS': 'Petronet LNG',
    'PIIND.NS': 'PI Industries',
    'PVRINOX.NS': 'PVR INOX',
    'SAIL.NS': 'Steel Authority of India Limited (SAIL)',
    'SIEMENS.NS': 'Siemens',
    'SYNGENE.NS': 'Syngene International',
    'UPL.NS': 'UPL Limited',
    'VEDANTA.NS': 'Vedanta',
    'ETERNAL.NS': 'Zomato'
}

# Sector ETFs
SECTOR_ETFS = {
    'BANKBEES.NS': 'Banking',
    'ITBEES.NS': 'IT',
    'PHARMABEES.NS': 'Pharma',
    'AUTOBEES.NS': 'Auto',
    'NIFTYBEES.NS': 'Metal' # Note: This is Nifty 50 ETF, not Metal. You might want a proper Metal ETF.
}

# Market Indices
MARKET_INDICES = {
    '^NSEI': 'Nifty 50',
    '^BSESN': 'BSE Sensex',
    'NIFTY_MIDCAP_100.NS': 'Nifty Midcap 100'
}