| """ |
| 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 |
|
|
