lanhao
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
from flask import Blueprint, request, jsonify
|
|
import base64
|
|
import os
|
|
import requests
|
|
from app.services.minimax_service import synthesize_minimax
|
|
from app.services.cosyvoice_service import synthesize_cosyvoice
|
|
|
|
bp = Blueprint('tts', __name__, url_prefix='/api')
|
|
|
|
MINIMAX_API_KEY = os.getenv('MINIMAX_API_KEY', '')
|
|
MINIMAX_API_URL = 'https://api.minimax.chat/v1/t2a_v2'
|
|
|
|
COSYVOICE_API_URL = os.getenv('COSYVOICE_API_URL', 'http://127.0.0.1:5001/v1/audio/synthesis')
|
|
|
|
@bp.route('/synthesize', methods=['POST'])
|
|
def synthesize():
|
|
data = request.get_json()
|
|
|
|
if not data or 'text' not in data:
|
|
return jsonify({'error': '缺少text参数'}), 400
|
|
|
|
text = data.get('text')
|
|
engine = data.get('engine', 'minimax')
|
|
speed = data.get('speed', 1.0)
|
|
pitch = data.get('pitch', 0)
|
|
volume = data.get('volume', 1.0)
|
|
instruction = data.get('instruction') # 用于CosyVoice的情绪指令
|
|
emotion = data.get('emotion', 'neutral') # 用于MiniMax的情绪参数
|
|
|
|
if not text.strip():
|
|
return jsonify({'error': '文本不能为空'}), 400
|
|
|
|
if len(text) > 2000:
|
|
return jsonify({'error': '文本不能超过2000字'}), 400
|
|
|
|
try:
|
|
if engine == 'cosyvoice':
|
|
audio_data = synthesize_cosyvoice(
|
|
text=text,
|
|
speed=speed,
|
|
pitch=pitch,
|
|
volume=volume,
|
|
instruction=instruction
|
|
)
|
|
else:
|
|
audio_data = synthesize_minimax(
|
|
text=text,
|
|
speed=speed,
|
|
pitch=pitch,
|
|
volume=volume,
|
|
emotion=emotion
|
|
)
|
|
|
|
return jsonify({
|
|
'success': True,
|
|
'audio': base64.b64encode(audio_data).decode('utf-8')
|
|
})
|
|
except Exception as e:
|
|
return jsonify({'error': str(e)}), 500
|