Spaces:
Sleeping
Sleeping
File size: 1,771 Bytes
ad0da04 |
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 |
#!/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) |