Refactor project structure and add new features

2025-10-23 00:42:09 +03:00
parent 144414fcf8
commit 852f7975b9
63 changed files with 8101 additions and 1341 deletions
@@ -0,0 +1,109 @@
+"""
+Модуль utils.py содержит утилитарные функции для работы с аудио.
+"""
+
+import os
+import subprocess
+import wave
+import numpy as np
+import logging
+from typing import Tuple
+
+logger = logging.getLogger('app.audio_utils')
+
+
+class AudioUtils:
+    """Утилитарный класс для работы с аудио."""
+    
+    @staticmethod
+    def load_audio(file_path: str, sr: int = 16000) -> Tuple[np.ndarray, int]:
+        """
+        Загрузка аудиофайла с использованием встроенной библиотеки wave.
+
+        Args:
+            file_path: Путь к аудиофайлу.
+            sr: Целевая частота дискретизации.
+
+        Returns:
+            Кортеж (массив numpy, частота дискретизации).
+            
+        Raises:
+            Exception: Если не удалось загрузить аудиофайл.
+        """
+        try:
+            # Открываем WAV файл
+            with wave.open(file_path, 'rb') as wav_file:
+                # Проверяем, что это моно-аудио
+                if wav_file.getnchannels() != 1:
+                    logger.warning(f"Файл {file_path} не моно-аудио, конвертируем в моно")
+                
+                # Читаем аудиоданные
+                frames = wav_file.readframes(-1)
+                # Конвертируем 16-битные целые числа в float32 в диапазоне [-1.0, 1.0]
+                # 32768.0 - это 2^15, максимальное значение для 16-битного знакового целого
+                audio_array = np.frombuffer(frames, dtype=np.int16).astype(np.float32) / 32768.0
+                
+                # Получаем частоту дискретизации
+                sampling_rate = wav_file.getframerate()
+                
+                # Если частота дискретизации не совпадает с целевой, выполняем ресемплинг
+                if sampling_rate != sr:
+                    from scipy.signal import resample
+                    num_samples = int(len(audio_array) * sr / sampling_rate)
+                    audio_array = resample(audio_array, num_samples)
+                    sampling_rate = sr
+                
+                return audio_array, sampling_rate
+                
+        except Exception as e:
+            logger.error(f"Ошибка при загрузке аудио {file_path}: {e}")
+            raise
+    
+    @staticmethod
+    def get_audio_duration(file_path: str) -> float:
+        """
+        Определяет длительность аудиофайла с использованием ffprobe.
+        
+        Args:
+            file_path: Путь к аудиофайлу.
+            
+        Returns:
+            Длительность в секундах.
+        """
+        try:
+            # Проверяем, что файл существует
+            if not os.path.exists(file_path):
+                logger.error(f"Файл не существует: {file_path}")
+                raise Exception(f"Файл не существует: {file_path}")
+                
+            cmd = [
+                "ffprobe",
+                "-v", "error",
+                "-show_entries", "format=duration",
+                "-of", "default=noprint_wrappers=1:nokey=1",
+                file_path
+            ]
+            
+            result = subprocess.run(
+                cmd,
+                capture_output=True,
+                text=True,
+                check=True,
+                timeout=10  # Ограничение по времени выполнения
+            )
+            
+            duration = float(result.stdout.strip())
+            return duration
+            
+        except subprocess.TimeoutExpired:
+            logger.error(f"Таймаут при определении длительности файла {file_path}")
+            raise Exception(f"Таймаут при определении длительности файла {file_path}")
+        except subprocess.CalledProcessError as e:
+            logger.error(f"Ошибка при выполнении ffprobe для файла {file_path}: {e.stderr}")
+            raise Exception(f"Ошибка при выполнении ffprobe для файла {file_path}: {e.stderr}")
+        except (ValueError, TypeError) as e:
+            logger.error(f"Ошибка при преобразовании длительности для файла {file_path}: {e}")
+            raise Exception(f"Ошибка при преобразовании длительности для файла {file_path}: {e}")
+        except Exception as e:
+            logger.error(f"Неожиданная ошибка при определении длительности файла {file_path}: {e}")
+            raise Exception(f"Неожиданная ошибка при определении длительности файла {file_path}: {e}")