38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import sys
|
|
import threading
|
|
from io import BufferedWriter
|
|
|
|
from fastapi import FastAPI, Query
|
|
from pydantic import BaseModel, Field
|
|
from starlette.websockets import WebSocket
|
|
|
|
import dashscope
|
|
from dashscope.audio.tts import SpeechSynthesizer
|
|
|
|
from app.config.env import env
|
|
|
|
dashscope.api_key = env.llm_key_bailian
|
|
|
|
|
|
def add_websocket_voice_generate_route(app: FastAPI):
|
|
@app.post("/voice_generate")
|
|
async def voice_generate(body: VoiceGenerateParamSchema):
|
|
|
|
result = SpeechSynthesizer.call(**body.model_dump())
|
|
if result.get_audio_data() is not None:
|
|
with open('output.mp3', 'wb') as f:
|
|
f.write(result.get_audio_data())
|
|
print('SUCCESS: get audio data: %dbytes in output.mp3' %
|
|
(sys.getsizeof(result.get_audio_data())))
|
|
else:
|
|
print('ERROR: response is %s' % (result.get_response()))
|
|
|
|
|
|
class VoiceGenerateParamSchema(BaseModel):
|
|
text: str = Field(..., description="要生成的文本")
|
|
model: str = Field(default="sambert-zhichu-v1", description="模型名称")
|
|
sample_rate: int = Field(default=16000, description="采样率")
|
|
format: str = Field(default="mp3", description="音频格式")
|
|
volume: int = Field(default=100, description="音量大小")
|
|
rate: int = Field(default=1, description="语速")
|