Update config.py

config.py

@@ -1,17 +1,31 @@
-"""Configuration for RewardPilot Web UI"""
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
 
-# Service URLs
+# ==================== 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"
 
-# UI Configuration
-APP_TITLE = "💳 RewardPilot"
-APP_DESCRIPTION = "AI-Powered Credit Card Recommendation Engine"
-THEME = "soft"  # Gradio theme
+# ==================== 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"
+
+# ==================== 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"
 
-# MCC Categories
+# ==================== 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",
@@ -26,7 +40,7 @@ MCC_CATEGORIES = {
     "General Retail": "5999"
 }
 
-# Merchant options by category
+# ==================== 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"],
@@ -41,37 +55,10 @@ MERCHANTS_BY_CATEGORY = {
     "General Retail": ["Target", "Walmart", "Amazon", "Best Buy", "Home Depot"]
 }
 
-# Sample Users
+# ==================== SAMPLE USERS ====================
 SAMPLE_USERS = ["u_alice", "u_bob", "u_charlie"]
 
-# Feature Flags
+# ==================== FEATURE FLAGS ====================
 ENABLE_ANALYTICS = True
 ENABLE_COMPARISON = True
-ENABLE_HISTORY = True
-
-from dotenv import load_dotenv
-import os
-
-load_dotenv()
-
-
-# LLM Configuration
-
-# 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"
-
-HF_TOKEN = os.getenv("HF_TOKEN", "")  # Hugging Face API 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  # 5 minutes
+ENABLE_HISTORY = True