config.py

# Indonesian Stock Exchange (IDX) major stocks
IDX_STOCKS = {
    "BBCA.JK": "Bank Central Asia",
    "BMRI.JK": "Bank Mandiri",
    "BBNI.JK": "Bank BNI",
    "BBRI.JK": "Bank BRI",
    "TLKM.JK": "Telkom Indonesia",
    "UNVR.JK": "Unilever Indonesia",
    "INDF.JK": "Indofood Sukses Makmur",
    "KLBF.JK": "Kalbe Farma",
    "ASII.JK": "Astra International",
    "ICBP.JK": "Indofood CBP Sukses Makmur",
    "SMGR.JK": "Semen Indonesia",
    "INTP.JK": "Indocement Tunggal Prakasa",
    "ANTM.JK": "Aneka Tambang",
    "TINS.JK": "Timah",
    "PTBA.JK": "Bukit Asam",
    "PGAS.JK": "Perusahaan Gas Negara",
    "EXCL.JK": "XL Axiata",
    "ISAT.JK": "Indosat Ooredoo",
    "FREN.JK": "Smartfren Telecom",
    "BKLA.JK": "Bank Bukopin",
    "BANK.JK": "Bank Danamon",
    "BSIM.JK": "Bank Syariah Indonesia",
    "MDKA.JK": "Merdeka Copper Gold",
    "SMCB.JK": "Semen Baturaja",
    "WIKA.JK": "Wijaya Karya",
    "ADHI.JK": "Adhi Karya",
    "PTPP.JK": "PP (Persero)",
    "JSMR.JK": "Jasa Marga",
    "TPIA.JK": "Chandra Asri Petrochemical",
    "SRIL.JK": "Sri Rejeki Isman",
    "GGRM.JK": "Gudang Garam",
    "HMSP.JK": "HM Sampoerna",
    "TCID.JK": "MNC Investama",
    "MNCN.JK": "Media Nusantara Citra",
    "BHITT.JK": "Bumi Resources Minerals",
    "DOID.JK": "Delta Dunia Makmur",
    "MEDC.JK": "Medco Energi International"
}

# Default settings
DEFAULT_PERIOD = 90  # days
DEFAULT_FORECAST_HORIZON = 7  # days
MAX_FORECAST_HORIZON = 30  # days

# Model settings
MODEL_NAME = "amazon/chronos-bolt-base"
MODEL_CACHE_DIR = "./model_cache"

# Data settings
MIN_DATA_POINTS = 30  # minimum historical data points required
MAX_DATA_POINTS = 365  # maximum historical data points to use

# Visualization settings
PLOT_HEIGHT = 600
PLOT_WIDTH = 1000
CONFIDENCE_LEVEL = 0.95  # for confidence intervals

# API settings
YFINANCE_TIMEOUT = 30  # seconds
MAX_RETRIES = 3

# Colors for plotting
COLORS = {
    'historical': '#1f77b4',  # blue
    'forecast': '#ff7f0e',    # orange
    'confidence': 'rgba(255, 127, 14, 0.2)',  # orange with transparency
    'volume': '#87CEEB'       # sky blue
}