Spaces:
Sleeping
Sleeping
File size: 1,915 Bytes
312e168 | 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 | import os
from dotenv import load_dotenv
from pathlib import Path
import logging
def load_config():
"""
Load configuration based on APP_ENV environment variable.
Follows the user rules for environment-specific config files.
"""
# Get environment
app_env = os.getenv('APP_ENV', 'local')
# Construct config file path following the user rules
config_dir = Path(__file__).parent.parent.parent / 'configs' / 'envs'
config_file = config_dir / f'.env.{app_env}'
# Load environment variables
if config_file.exists():
load_dotenv(config_file)
print(f"✅ Loaded config from {config_file}")
else:
print(f"⚠️ Config file {config_file} not found, using defaults")
# Load default local config
default_config = config_dir / '.env.local'
if default_config.exists():
load_dotenv(default_config)
# Return configuration dictionary
config = {
'APP_ENV': os.getenv('APP_ENV', 'local'),
'DEBUG': os.getenv('DEBUG', 'True').lower() == 'true',
'LOG_LEVEL': os.getenv('LOG_LEVEL', 'INFO'),
'MODEL_CACHE_DIR': os.getenv('MODEL_CACHE_DIR', './models'),
'GRADIO_SERVER_NAME': os.getenv('GRADIO_SERVER_NAME', '127.0.0.1'),
'GRADIO_SERVER_PORT': int(os.getenv('GRADIO_SERVER_PORT', 7860)),
'GRADIO_SHARE': os.getenv('GRADIO_SHARE', 'False').lower() == 'true',
'DEFAULT_MODEL': os.getenv('DEFAULT_MODEL', 'distil-whisper'),
'DEFAULT_LANGUAGE': os.getenv('DEFAULT_LANGUAGE', 'hindi'),
'MAX_AUDIO_LENGTH': int(os.getenv('MAX_AUDIO_LENGTH', 300)),
'ENABLE_GPU': os.getenv('ENABLE_GPU', 'True').lower() == 'true'
}
# Configure logging
logging.basicConfig(
level=getattr(logging, config['LOG_LEVEL']),
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
)
return config
|