File size: 5,930 Bytes
f1ba2d2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 | """
Configuration settings management.
"""
import os
import yaml
from typing import Dict, Any, Optional
from pathlib import Path
class Settings:
"""Application settings manager."""
def __init__(self, config_file: str = "config/config.yaml"):
"""
Initialize settings from config file and environment variables.
Args:
config_file (str): Path to configuration file
"""
self.config_file = config_file
self.config = self._load_config()
self._override_with_env()
def _load_config(self) -> Dict[str, Any]:
"""Load configuration from YAML file."""
try:
config_path = Path(self.config_file)
if config_path.exists():
with open(config_path, 'r', encoding='utf-8') as f:
return yaml.safe_load(f) or {}
else:
return self._get_default_config()
except Exception as e:
print(f"Error loading config file: {e}")
return self._get_default_config()
def _get_default_config(self) -> Dict[str, Any]:
"""Get default configuration."""
return {
'app': {
'title': 'AI-Powered YouTube Transcript Tutor',
'description': 'Ask questions from YouTube lecture transcripts using AI',
'version': '1.0.0'
},
'ui': {
'theme': 'light',
'sidebar_width': 300,
'max_chat_history_display': 50,
'enable_animations': True
},
'processing': {
'default_chunk_size': 1000,
'chunk_overlap': 200,
'max_transcript_length': 1000000,
'supported_languages': ['en', 'es', 'fr', 'de', 'it', 'pt', 'ru', 'ja', 'ko', 'zh'],
'default_language': 'en'
},
'ai': {
'model_temperature': 0.7,
'max_tokens': 2000,
'retrieval_k': 4,
'chain_type': 'stuff'
},
'export': {
'formats': ['pdf', 'txt', 'json'],
'max_export_entries': 1000,
'pdf_page_size': 'A4'
},
'cache': {
'enable_vectorstore_cache': True,
'cache_directory': 'cache',
'max_cache_size_mb': 500
},
'logging': {
'level': 'INFO',
'file': 'logs/app.log',
'max_file_size_mb': 10,
'backup_count': 5
},
'security': {
'max_url_length': 2048,
'allowed_domains': ['youtube.com', 'youtu.be', 'm.youtube.com'],
'rate_limit_requests': 100,
'rate_limit_window_minutes': 60
}
}
def _override_with_env(self):
"""Override configuration with environment variables."""
# OpenAI API Key
openai_key = os.getenv('OPENAI_API_KEY')
if openai_key:
if 'ai' not in self.config:
self.config['ai'] = {}
self.config['ai']['openai_api_key'] = openai_key
# Log level
log_level = os.getenv('LOG_LEVEL')
if log_level:
self.config['logging']['level'] = log_level.upper()
# Cache directory
cache_dir = os.getenv('CACHE_DIRECTORY')
if cache_dir:
self.config['cache']['cache_directory'] = cache_dir
def get(self, key: str, default: Any = None) -> Any:
"""
Get configuration value using dot notation.
Args:
key (str): Configuration key (e.g., 'app.title')
default (Any): Default value if key not found
Returns:
Any: Configuration value
"""
keys = key.split('.')
value = self.config
try:
for k in keys:
value = value[k]
return value
except (KeyError, TypeError):
return default
def set(self, key: str, value: Any):
"""
Set configuration value using dot notation.
Args:
key (str): Configuration key (e.g., 'app.title')
value (Any): Value to set
"""
keys = key.split('.')
config = self.config
for k in keys[:-1]:
if k not in config:
config[k] = {}
config = config[k]
config[keys[-1]] = value
def get_openai_api_key(self) -> Optional[str]:
"""Get OpenAI API key from config or environment."""
return self.get('ai.openai_api_key') or os.getenv('OPENAI_API_KEY')
def get_app_config(self) -> Dict[str, Any]:
"""Get application configuration."""
return self.get('app', {})
def get_ui_config(self) -> Dict[str, Any]:
"""Get UI configuration."""
return self.get('ui', {})
def get_processing_config(self) -> Dict[str, Any]:
"""Get processing configuration."""
return self.get('processing', {})
def get_ai_config(self) -> Dict[str, Any]:
"""Get AI configuration."""
return self.get('ai', {})
def get_export_config(self) -> Dict[str, Any]:
"""Get export configuration."""
return self.get('export', {})
def get_cache_config(self) -> Dict[str, Any]:
"""Get cache configuration."""
return self.get('cache', {})
def get_logging_config(self) -> Dict[str, Any]:
"""Get logging configuration."""
return self.get('logging', {})
def get_security_config(self) -> Dict[str, Any]:
"""Get security configuration."""
return self.get('security', {})
# Global settings instance
settings = Settings()
|