import os
from pathlib import Path

BASE_DIR = Path(__file__).parent
CACHE_DIR = BASE_DIR / "cache"
CACHE_DIR.mkdir(exist_ok=True)

LLM_API_KEY = os.getenv("DEEPSEEK_API_KEY", "sk-5e81f3314d34412d9e25af1d4d2b2f77")
LLM_BASE_URL = os.getenv("DEEPSEEK_BASE_URL", "https://api.deepseek.com")
LLM_MODEL = os.getenv("DEEPSEEK_MODEL", "deepseek-chat")

DATA_CACHE_HOURS = 4
TRAINING_DATA_DIR = BASE_DIR / "training_data"
TRAINING_DATA_DIR.mkdir(exist_ok=True)

STOCK_FILTER = {
    "exclude_st": True,
    "exclude_new_days": 60,
    "min_turnover": 50_000_000,
    "exclude_suspended": True,
}

SCREENER_TOP_N = 20
FEATURE_WINDOWS = [5, 10, 20]