chorus_detection/config.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""Configuration settings for the chorus detection system."""

import os
from pathlib import Path
from typing import Dict, Any, Union, Optional

# Audio processing settings
SR: int = 12000  # Sample rate
HOP_LENGTH: int = 128  # Hop length for signal processing
MAX_FRAMES: int = 300  # Maximum frames per segment
MAX_METERS: int = 201  # Maximum meters per song
N_FEATURES: int = 15  # Number of features

# Project paths
PROJECT_ROOT: Path = Path(__file__).parent.parent.parent.resolve()
MODEL_DIR: Path = PROJECT_ROOT / "models" / "CRNN"
MODEL_PATH: Path = MODEL_DIR / "best_model_V3.h5"
AUDIO_TEMP_PATH: Path = PROJECT_ROOT / "output" / "temp"
LOG_DIR: Path = PROJECT_ROOT / "logs"

# Alternative Docker paths
DOCKER_MODEL_PATH: str = "/app/models/CRNN/best_model_V3.h5"
DOCKER_TEMP_PATH: str = "/app/output/temp"


def get_env_path(env_var: str, default_path: Path) -> Path:
    """Get a path from environment variable or use the default.
    
    Args:
        env_var: Name of the environment variable
        default_path: Default path to use if environment variable is not set
        
    Returns:
        Path object for the specified location
    """
    env_value = os.environ.get(env_var)
    if env_value:
        return Path(env_value).resolve()
    return default_path


# Override paths with environment variables if provided
MODEL_PATH = get_env_path("CHORUS_MODEL_PATH", MODEL_PATH)
AUDIO_TEMP_PATH = get_env_path("CHORUS_TEMP_PATH", AUDIO_TEMP_PATH)
LOG_DIR = get_env_path("CHORUS_LOG_DIR", LOG_DIR)

# Create necessary directories
os.makedirs(MODEL_DIR, exist_ok=True)
os.makedirs(AUDIO_TEMP_PATH, exist_ok=True)
os.makedirs(LOG_DIR, exist_ok=True)
os.makedirs(PROJECT_ROOT / "output", exist_ok=True)