config.py

from dotenv import load_dotenv
import os

load_dotenv()

# ==================== SERVICE URLS ====================
ORCHESTRATOR_URL = "https://mcp-1st-birthday-rewardpilot-orchestrator.hf.space"
SMART_WALLET_URL = "https://mcp-1st-birthday-rewardpilot-smart-wallet.hf.space"
REWARDS_RAG_URL = "https://mcp-1st-birthday-rewardpilot-rewards-rag.hf.space"
SPEND_FORECAST_URL = "https://mcp-1st-birthday-rewardpilot-spend-forecast.hf.space"

# ==================== GEMINI CONFIGURATION ====================
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "")
GEMINI_MODEL = "gemini-2.5-flash" 
USE_GEMINI = os.getenv("USE_GEMINI", "true").lower() == "true"


# ==================== OPENAI CONFIGURATION ====================
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
USE_OPENAI = bool(OPENAI_API_KEY)

# ==================== LLAMA CONFIGURATION ====================
HF_TOKEN = os.getenv("HF_TOKEN", "")
LLM_MODEL = os.getenv("LLM_MODEL", "meta-llama/Llama-3.2-3B-Instruct")
LLM_ENABLED = os.getenv("LLM_ENABLED", "true").lower() == "true"

# ==================== UI CONFIGURATION ====================
APP_TITLE = "RewardPilot - AI-Powered Credit Card Optimizer"
APP_DESCRIPTION = "Maximize your credit card rewards with intelligent recommendations"
THEME = "soft"
CACHE_TTL = 300

# ==================== MCC CATEGORIES ====================
MCC_CATEGORIES = {
    "Groceries": "5411",
    "Restaurants": "5812",
    "Wholesale Club": "5300",
    "Fast Food": "5814",
    "Bars/Taverns": "5813",
    "Gas Stations": "5541",
    "Airlines": "3000",
    "Hotels": "7011",
    "Movie Theaters": "7832",
    "Entertainment": "7841",
    "Drugstores": "5912",
    "General Retail": "5999"
}

# ==================== MERCHANTS BY CATEGORY ====================
MERCHANTS_BY_CATEGORY = {
    "Groceries": ["Whole Foods", "Trader Joe's", "Safeway", "Kroger", "Costco"],
    "Restaurants": ["Olive Garden", "Chipotle", "The Cheesecake Factory", "P.F. Chang's", "Red Lobster"],
    "Fast Food": ["McDonald's", "Starbucks", "Subway", "Taco Bell", "Dunkin'"],
    "Bars/Taverns": ["Local Bar", "Sports Bar", "Pub & Grill", "Wine Bar", "Brewery"],
    "Gas Stations": ["Shell", "Chevron", "BP", "Exxon", "Mobil"],
    "Airlines": ["United Airlines", "Delta", "American Airlines", "Southwest", "JetBlue"],
    "Hotels": ["Marriott", "Hilton", "Hyatt", "Holiday Inn", "Best Western"],
    "Movie Theaters": ["AMC Theatres", "Regal Cinemas", "Cinemark", "Alamo Drafthouse"],
    "Entertainment": ["Concert Venue", "Theme Park", "Museum", "Sports Arena", "Comedy Club"],
    "Drugstores": ["CVS", "Walgreens", "Rite Aid", "Duane Reade"],
    "General Retail": ["Target", "Walmart", "Amazon", "Best Buy", "Home Depot"]
}

# ==================== SAMPLE USERS ====================
SAMPLE_USERS = ["u_alice", "u_bob", "u_charlie"]

# ==================== FEATURE FLAGS ====================
ENABLE_ANALYTICS = True
ENABLE_COMPARISON = True
ENABLE_HISTORY = True