Update utils/config.py

utils/config.py

@@ -1,233 +1,152 @@
-"""
-Configuration module for the MONA application.
-
-This module centralizes all configuration settings and constants used throughout
-the application, making it easier to maintain and modify settings in one place.
-"""
-
 import os
-from pathlib import Path
+from typing import Dict, Any, Optional
+from dataclasses import dataclass, asdict
+import streamlit as st
 
-# Base paths
-BASE_DIR = Path(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
-DATA_DIR = BASE_DIR / "data"
-EXPORT_DIR = DATA_DIR / "exports"
-BACKUP_DIR = DATA_DIR / "backups"
+from utils.logging import get_logger
+from utils.error_handling import ConfigError
 
-# Ensure directories exist
-os.makedirs(DATA_DIR, exist_ok=True)
-os.makedirs(EXPORT_DIR, exist_ok=True)
-os.makedirs(BACKUP_DIR, exist_ok=True)
+logger = get_logger(__name__)
 
-# File paths
-FILE_PATHS = {
-    "tasks": DATA_DIR / "tasks.json",
-    "notes": DATA_DIR / "notes.json",
-    "goals": DATA_DIR / "goals.json",
-    "settings": DATA_DIR / "settings.json",
-    "usage": DATA_DIR / "usage.json",
-    "activity": DATA_DIR / "activity.json",
-    "focus": DATA_DIR / "focus.json",
-    "mood": DATA_DIR / "mood.json",
-    "wellness": DATA_DIR / "wellness.json",
-    "integrations": DATA_DIR / "integrations.json",
-}
+@dataclass
+class AppConfig:
+    """Application configuration data class"""
+    app_name: str = "MONA Dashboard"
+    version: str = "1.0.0"
+    debug_mode: bool = False
+    max_data_points: int = 1000
+    refresh_interval: int = 30
+    default_chart_type: str = "line"
+    theme: str = "light"
+    
+    # UI Configuration
+    sidebar_width: int = 300
+    chart_height: int = 400
+    table_page_size: int = 50
+    
+    # Data Configuration
+    data_cache_timeout: int = 300
+    max_file_size_mb: int = 10
+    supported_file_types: list = None
+    
+    def __post_init__(self):
+        if self.supported_file_types is None:
+            self.supported_file_types = ['.csv', '.xlsx', '.json', '.parquet']
 
-# AI Models configuration
-AI_MODELS = {
-    "text_generation": {
-        "name": "microsoft/DialoGPT-medium",
-        "max_length": 100,
-        "temperature": 0.7,
-    },
-    "question_answering": {
-        "name": "distilbert-base-uncased-distilled-squad",
-    },
-    "image_captioning": {
-        "name": "Salesforce/blip-image-captioning-base",
-        "max_length": 50,
-    },
-    "speech_to_text": {
-        "name": "openai/whisper-small",
-    },
-    "translation": {
-        "name": "Helsinki-NLP/opus-mt-en-de",
-    },
-    "sentiment": {
-        "name": "cardiffnlp/twitter-roberta-base-sentiment-latest",
-    },
-    "summarization": {
-        "name": "facebook/bart-large-cnn",
-        "max_length": 150,
-        "min_length": 30,
-    },
-    "code_generation": {
-        "name": "microsoft/CodeBERT-base",
-    },
-}
+# Global configuration instance
+_config = None
 
-# Default application settings
-DEFAULT_SETTINGS = {
-    "appearance": {
-        "theme": "light",
-        "primary_color": "#7B68EE",
-        "secondary_color": "#FF6B6B",
-        "font_size": "medium",
-        "sidebar_position": "left",
-        "default_page": "dashboard",
-        "compact_mode": False,
-        "show_welcome": True,
-    },
-    "notifications": {
-        "enable_notifications": True,
-        "notification_sound": True,
-        "task_reminders": True,
-        "goal_reminders": True,
-        "focus_reminders": True,
-        "reminder_time": "09:00",
-    },
-    "ai_preferences": {
-        "enable_suggestions": True,
-        "preferred_model": "balanced",
-        "usage_limit": 100,
-        "current_usage": 0,
-    },
-    "data_management": {
-        "auto_backup": True,
-        "backup_frequency": "weekly",
-        "data_retention": "1 year",
-        "auto_archive": True,
-    },
-    "user_profile": {
-        "name": "User",
-        "email": "",
-        "timezone": "UTC",
-        "language": "English",
-        "date_format": "MM/DD/YYYY",
-        "time_format": "12h",
-    },
-    "api_keys": {
-        "OpenWeatherMap": "",
-        "GitHub": "",
-        "Google Calendar": "",
-        "Telegram": "",
-        "News API": "",
-        "Crypto API": "",
-    },
-}
+def load_config() -> AppConfig:
+    """Load application configuration"""
+    global _config
+    
+    if _config is None:
+        try:
+            # Create default config
+            _config = AppConfig()
+            
+            # Override with environment variables if available
+            _config.debug_mode = os.getenv('DEBUG', 'False').lower() == 'true'
+            _config.max_data_points = int(os.getenv('MAX_DATA_POINTS', '1000'))
+            _config.refresh_interval = int(os.getenv('REFRESH_INTERVAL', '30'))
+            _config.theme = os.getenv('THEME', 'light')
+            
+            # Override with Streamlit session state if available
+            if 'app_config' in st.session_state:
+                session_config = st.session_state.app_config
+                for key, value in session_config.items():
+                    if hasattr(_config, key):
+                        setattr(_config, key, value)
+            
+            logger.info("Configuration loaded successfully")
+            
+        except Exception as e:
+            logger.error(f"Failed to load configuration: {str(e)}")
+            raise ConfigError(f"Configuration loading failed: {str(e)}")
+    
+    return _config
 
-# UI Constants
-UI_CONSTANTS = {
-    "sidebar_width": 300,
-    "header_height": 60,
-    "card_border_radius": "10px",
-    "animation_duration": "0.3s",
-    "max_items_per_page": 20,
-    "chart_colors": ["#7B68EE", "#FF6B6B", "#64DFDF", "#FFAB4C", "#9D65C9"],
-}
+def save_config(config: AppConfig) -> bool:
+    """Save configuration to session state"""
+    try:
+        st.session_state.app_config = asdict(config)
+        global _config
+        _config = config
+        logger.info("Configuration saved successfully")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to save configuration: {str(e)}")
+        return False
 
-# UI Colors for components
-UI_COLORS = {
-    "blue": "#3498db",
-    "green": "#2ecc71",
-    "red": "#e74c3c",
-    "yellow": "#f1c40f",
-    "purple": "#9b59b6",
-    "orange": "#e67e22",
-    "teal": "#1abc9c",
-    "gray": "#95a5a6",
-    "dark": "#34495e",
-    "primary": "#7B68EE",
-    "secondary": "#FF6B6B",
-    "success": "#2ecc71",
-    "warning": "#f1c40f",
-    "danger": "#e74c3c",
-    "info": "#3498db",
-}
+def get_config_value(key: str, default_value: Any = None) -> Any:
+    """Get a specific configuration value"""
+    try:
+        config = load_config()
+        return getattr(config, key, default_value)
+    except Exception as e:
+        logger.error(f"Failed to get config value for key {key}: {str(e)}")
+        return default_value
 
-# UI Sizes for components
-UI_SIZES = {
-    "xs": "0.75rem",
-    "sm": "0.875rem",
-    "md": "1rem",
-    "lg": "1.25rem",
-    "xl": "1.5rem",
-    "2xl": "2rem",
-    "3xl": "3rem",
-    "icon_sm": "16px",
-    "icon_md": "24px",
-    "icon_lg": "32px",
-    "spacing_xs": "0.25rem",
-    "spacing_sm": "0.5rem",
-    "spacing_md": "1rem",
-    "spacing_lg": "1.5rem",
-    "spacing_xl": "2rem",
-}
+def update_config_value(key: str, value: Any) -> bool:
+    """Update a specific configuration value"""
+    try:
+        config = load_config()
+        if hasattr(config, key):
+            setattr(config, key, value)
+            return save_config(config)
+        else:
+            logger.warning(f"Configuration key '{key}' does not exist")
+            return False
+    except Exception as e:
+        logger.error(f"Failed to update config value for key {key}: {str(e)}")
+        return False
 
-# Feature flags for enabling/disabling features
-FEATURE_FLAGS = {
-    "enable_voice_input": True,
-    "enable_image_analysis": True,
-    "enable_code_generation": True,
-    "enable_integrations": True,
-    "enable_export": True,
-    "enable_analytics": True,
-}
+def reset_config() -> AppConfig:
+    """Reset configuration to defaults"""
+    global _config
+    _config = AppConfig()
+    save_config(_config)
+    logger.info("Configuration reset to defaults")
+    return _config
 
-# Integration settings
-INTEGRATION_SETTINGS = {
-    "google_calendar": {
-        "enabled": False,
-        "auth_url": "https://accounts.google.com/o/oauth2/auth",
-        "token_url": "https://oauth2.googleapis.com/token",
-        "scope": "https://www.googleapis.com/auth/calendar.readonly",
-    },
-    "microsoft_todo": {
-        "enabled": False,
-        "auth_url": "https://login.microsoftonline.com/common/oauth2/v2.0/authorize",
-        "token_url": "https://login.microsoftonline.com/common/oauth2/v2.0/token",
-        "scope": "Tasks.ReadWrite",
-    },
-    "notion": {
-        "enabled": False,
-        "auth_url": "https://api.notion.com/v1/oauth/authorize",
-        "token_url": "https://api.notion.com/v1/oauth/token",
-    },
-    "weather_api": {
-        "enabled": False,
-        "base_url": "https://api.openweathermap.org/data/2.5/weather",
-        "units": "metric",
-    },
-}
+def get_config_dict() -> Dict[str, Any]:
+    """Get configuration as dictionary"""
+    try:
+        config = load_config()
+        return asdict(config)
+    except Exception as e:
+        logger.error(f"Failed to get configuration dictionary: {str(e)}")
+        return {}
 
-# Logging configuration
-LOGGING_CONFIG = {
-    "version": 1,
-    "disable_existing_loggers": False,
-    "formatters": {
-        "standard": {
-            "format": "%(asctime)s [%(levelname)s] %(name)s: %(message)s"
-        },
-    },
-    "handlers": {
-        "console": {
-            "class": "logging.StreamHandler",
-            "level": "INFO",
-            "formatter": "standard",
-        },
-        "file": {
-            "class": "logging.FileHandler",
-            "level": "DEBUG",
-            "formatter": "standard",
-            "filename": DATA_DIR / "mona.log",
-            "mode": "a",
-        },
-    },
-    "loggers": {
-        "": {  # root logger
-            "handlers": ["console", "file"],
-            "level": "DEBUG",
-            "propagate": True,
-        },
-    },
-}
+def validate_config(config: AppConfig) -> bool:
+    """Validate configuration values"""
+    try:
+        # Validate numeric values
+        if config.max_data_points <= 0:
+            raise ConfigError("max_data_points must be positive")
+        
+        if config.refresh_interval <= 0:
+            raise ConfigError("refresh_interval must be positive")
+        
+        if config.sidebar_width <= 0:
+            raise ConfigError("sidebar_width must be positive")
+        
+        if config.chart_height <= 0:
+            raise ConfigError("chart_height must be positive")
+        
+        # Validate string values
+        if config.theme not in ['light', 'dark']:
+            raise ConfigError("theme must be 'light' or 'dark'")
+        
+        if config.default_chart_type not in ['line', 'bar', 'scatter', 'area']:
+            raise ConfigError("invalid default_chart_type")
+        
+        logger.info("Configuration validation successful")
+        return True
+        
+    except ConfigError as e:
+        logger.error(f"Configuration validation failed: {str(e)}")
+        raise e
+    except Exception as e:
+        logger.error(f"Unexpected error during configuration validation: {str(e)}")
+        raise ConfigError(f"Configuration validation error: {str(e)}")