| import base64 | |
| import io | |
| import json | |
| import os | |
| import uuid | |
| from typing import Optional | |
| import pandas as pd | |
| import requests | |
| from pydub import AudioSegment | |
| from ..tts_utils import mix_background_music | |
| class avaliable_voice_type: | |
| 语言: Optional[str] = "" | |
| 场景: Optional[str] = "" | |
| 音色名称: str | |
| voice_type: str | |
| 时间戳支持: bool = False | |
| 支持情感与风格类型: Optional[str] = "" | |
| 支持语言类型: Optional[str] = "" | |
| def __repr__(self): | |
| data = self.__dict__ | |
| text = "" | |
| text += f"音色: {data['音色名称']}" | |
| if data["语言"]: | |
| text += f"——{data['语言']}" | |
| if data["场景"]: | |
| text += f"——{data['场景']}" | |
| if data["支持情感与风格类型"]: | |
| text += f"——{data['支持情感与风格类型']}" | |
| if data["支持语言类型"]: | |
| text += f"——{data['支持语言类型']}" | |
| return text | |
| def get_data_map(filename="voice_list.xlsx"): | |
| path = os.path.join(os.path.dirname(os.path.abspath(__file__)), filename) | |
| df = pd.read_excel(path) | |
| df.fillna('', inplace=True) | |
| useful_voice = {} | |
| for index, row in df.iterrows(): | |
| data = avaliable_voice_type() | |
| data.语言 = row['语言'] | |
| data.场景 = row['场景'] | |
| data.音色名称 = row['音色名称'] | |
| data.voice_type = row['voice_type'] | |
| data.时间戳支持 = row['时间戳'] | |
| data.支持情感与风格类型 = row['支持情感/风格类型'] | |
| data.支持语言类型 = row['支持语言类型'] | |
| useful_voice[str(data)] = data | |
| return useful_voice | |
| def tts(text, appid, access_token, voice, speed_ratio, | |
| volume_ratio, pitch_ratio, 背景音乐, speaker_up, back_up): | |
| host = "openspeech.bytedance.com" | |
| api_url = f"https://{host}/api/v1/tts" | |
| header = {"Authorization": f"Bearer;{access_token}"} | |
| request_json = { | |
| "app": { | |
| "appid": appid, | |
| "token": "access_token", | |
| "cluster": "volcano_tts" | |
| }, | |
| "user": { | |
| "uid": "388808087185088" | |
| }, | |
| "audio": { | |
| "voice_type": useful_voice[voice].voice_type, | |
| "encoding": "mp3", | |
| "speed_ratio": speed_ratio, | |
| "volume_ratio": volume_ratio, | |
| "pitch_ratio": pitch_ratio, | |
| }, | |
| "request": { | |
| "reqid": str(uuid.uuid4()), | |
| "text": text, | |
| "text_type": "plain", | |
| "operation": "query", | |
| "with_frontend": 1, | |
| "frontend_type": "unitTson" | |
| } | |
| } | |
| try: | |
| resp = requests.post(api_url, data=json.dumps(request_json), headers=header) | |
| if "data" not in resp.json(): | |
| return str(resp.json()), None, None | |
| data = resp.json()["data"] | |
| mp3_file = base64.b64decode(data) | |
| original_audio = AudioSegment.from_mp3(io.BytesIO(mp3_file)) | |
| return None, *mix_background_music(original_audio, 背景音乐, speaker_up, | |
| back_up) | |
| except Exception as e: | |
| return str(e), None, None | |
| useful_voice = get_data_map() | |