57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
from banban.service.message_audio_storage import MessageAudioStorageService
|
|
from utils.logger import session_logger
|
|
|
|
|
|
message_audio_storage_service = MessageAudioStorageService()
|
|
|
|
|
|
# async def save_audio_file(audio_data: bytes, device_id: str) -> str:
|
|
# """Upload device audio to COS and return its object key."""
|
|
# try:
|
|
# stored = await message_audio_storage_service.upload_audio(
|
|
# device_id=device_id,
|
|
# content=audio_data,
|
|
# )
|
|
# session_logger.info(device_id, "audio", f"audio uploaded to COS: {stored.file_key}")
|
|
# return stored.file_key
|
|
# except Exception as e:
|
|
# session_logger.error(device_id, "audio", f"failed to store audio: {e}", exc_info=True)
|
|
# raise
|
|
|
|
|
|
import os
|
|
import uuid
|
|
from config import settings
|
|
from utils.logger import session_logger
|
|
# from utils.audio_denoiser import reduce_background_noise
|
|
|
|
|
|
async def save_audio_file(audio_data: bytes, device_id: str) -> str:
|
|
"""
|
|
保存音频数据到 assets/audio 目录
|
|
|
|
Args:
|
|
audio_data: 音频二进制数据
|
|
device_id: 设备ID
|
|
|
|
Returns:
|
|
音频文件的相对路径
|
|
"""
|
|
try:
|
|
audio_dir = os.path.join(settings.assets_dir, "audio")
|
|
os.makedirs(audio_dir, exist_ok=True)
|
|
|
|
filename = f"{device_id}_{uuid.uuid4().hex[:8]}.mp3"
|
|
filepath = os.path.join(audio_dir, filename)
|
|
|
|
with open(filepath, 'wb') as f:
|
|
f.write(audio_data)
|
|
|
|
# relative_path = f"assets/audio/{filename}"
|
|
session_logger.info(device_id, "audio", f"音频文件已保存: {filepath}")
|
|
|
|
# reduce_background_noise(filepath, relative_path,noise_path='assets/audio/noise_sample.wav',normalize_volume=True)
|
|
return filepath
|
|
except Exception as e:
|
|
session_logger.error(device_id, "audio", f"保存音频文件时出错: {e}", exc_info=True)
|
|
raise |