mirror of
https://github.com/remsky/Kokoro-FastAPI.git
synced 2025-08-05 16:48:53 +00:00
67 lines
2.6 KiB
Python
67 lines
2.6 KiB
Python
|
"""Audio conversion service"""
|
||
|
|
from io import BytesIO
|
||
|
|
import numpy as np
|
||
|
|
import scipy.io.wavfile as wavfile
|
||
|
|
import soundfile as sf
|
||
|
|
import logging
|
||
|
|
|
||
|
|
logger = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
class AudioService:
|
||
|
|
"""Service for audio format conversions"""
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def convert_audio(audio_data: np.ndarray, sample_rate: int, output_format: str) -> bytes:
|
||
|
|
"""Convert audio data to specified format
|
||
|
|
|
||
|
|
Args:
|
||
|
|
audio_data: Numpy array of audio samples
|
||
|
|
sample_rate: Sample rate of the audio
|
||
|
|
output_format: Target format (wav, mp3, etc.)
|
||
|
|
|
||
|
|
Returns:
|
||
|
|
Bytes of the converted audio
|
||
|
|
"""
|
||
|
|
buffer = BytesIO()
|
||
|
|
|
||
|
|
try:
|
||
|
|
if output_format == 'wav':
|
||
|
|
logger.info("Writing to WAV format...")
|
||
|
|
wavfile.write(buffer, sample_rate, audio_data)
|
||
|
|
return buffer.getvalue()
|
||
|
|
|
||
|
|
elif output_format == 'mp3':
|
||
|
|
# For MP3, we need to convert to WAV first
|
||
|
|
logger.info("Converting to MP3 format...")
|
||
|
|
wav_buffer = BytesIO()
|
||
|
|
wavfile.write(wav_buffer, sample_rate, audio_data)
|
||
|
|
wav_buffer.seek(0)
|
||
|
|
|
||
|
|
# Convert WAV to MP3 using soundfile
|
||
|
|
buffer = BytesIO()
|
||
|
|
sf.write(buffer, audio_data, sample_rate, format='mp3')
|
||
|
|
return buffer.getvalue()
|
||
|
|
|
||
|
|
elif output_format == 'opus':
|
||
|
|
logger.info("Converting to Opus format...")
|
||
|
|
sf.write(buffer, audio_data, sample_rate, format='ogg', subtype='opus')
|
||
|
|
return buffer.getvalue()
|
||
|
|
|
||
|
|
elif output_format == 'flac':
|
||
|
|
logger.info("Converting to FLAC format...")
|
||
|
|
sf.write(buffer, audio_data, sample_rate, format='flac')
|
||
|
|
return buffer.getvalue()
|
||
|
|
|
||
|
|
elif output_format == 'aac':
|
||
|
|
raise ValueError("AAC format is not currently supported. Please use wav, mp3, opus, or flac.")
|
||
|
|
|
||
|
|
elif output_format == 'pcm':
|
||
|
|
raise ValueError("PCM format is not currently supported. Please use wav, mp3, opus, or flac.")
|
||
|
|
|
||
|
|
else:
|
||
|
|
raise ValueError(f"Format {output_format} not supported. Supported formats are: wav, mp3, opus, flac.")
|
||
|
|
|
||
|
|
except Exception as e:
|
||
|
|
logger.error(f"Error converting audio to {output_format}: {str(e)}")
|
||
|
|
raise ValueError(f"Failed to convert audio to {output_format}: {str(e)}")
|