"""
Audio processing utilities
"""

import logging
from pathlib import Path
import librosa
import numpy as np

logger = logging.getLogger(__name__)

TARGET_SAMPLE_RATE = 16000


def load_audio(audio_path: Path, sr: int = TARGET_SAMPLE_RATE) -> np.ndarray:
    """
    Загрузить аудио файл.
    
    Args:
        audio_path: Путь к аудио файлу
        sr: Целевая частота дискретизации
        
    Returns:
        Аудио данные как numpy array
    """
    try:
        logger.debug(f"Loading audio: {audio_path}")
        audio, original_sr = librosa.load(str(audio_path), sr=None)
        
        # Ресемплинг если необходимо
        if original_sr != sr:
            logger.debug(f"Resampling from {original_sr}Hz to {sr}Hz")
            audio = librosa.resample(audio, orig_sr=original_sr, target_sr=sr)
        
        logger.info(f"Audio loaded: {len(audio)} samples, {sr}Hz")
        return audio
        
    except Exception as e:
        logger.error(f"Error loading audio: {e}")
        raise


def resample_audio(audio: np.ndarray, orig_sr: int, target_sr: int) -> np.ndarray:
    """
    Изменить частоту дискретизации аудио.
    
    Args:
        audio: Аудио данные
        orig_sr: Исходная частота
        target_sr: Целевая частота
        
    Returns:
        Обработанные аудио данные
    """
    if orig_sr == target_sr:
        return audio
    
    logger.debug(f"Resampling: {orig_sr}Hz -> {target_sr}Hz")
    return librosa.resample(audio, orig_sr=orig_sr, target_sr=target_sr)


def get_audio_duration(audio: np.ndarray, sr: int = TARGET_SAMPLE_RATE) -> float:
    """
    Получить длительность аудио в секундах.
    
    Args:
        audio: Аудио данные
        sr: Частота дискретизации
        
    Returns:
        Длительность в секундах
    """
    return len(audio) / sr


def validate_audio(audio_path: Path) -> bool:
    """
    Проверить валидность аудио файла.
    
    Args:
        audio_path: Путь к файлу
        
    Returns:
        True если файл валиден
    """
    try:
        audio = load_audio(audio_path)
        return len(audio) > 0
    except Exception as e:
        logger.error(f"Audio validation failed: {e}")
        return False