Spaces:
Sleeping
Sleeping
File size: 3,498 Bytes
96ea36d 03adfb9 96ea36d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 |
import os
import json, json5
from pathlib import Path
import utils
from APIs import VP
def save_voice_presets_metadata(voice_presets_path, metadata):
with open(voice_presets_path / 'metadata.json', 'w') as f:
json.dump(metadata, f, indent=4)
def load_voice_presets_metadata(voice_presets_path, safe_if_metadata_not_exist=False):
metadata_full_path = Path(voice_presets_path) / 'metadata.json'
if safe_if_metadata_not_exist:
if not os.path.exists(metadata_full_path):
return {}
with open(metadata_full_path, 'r') as f:
presets = json5.load(f)
return presets
# return system voice presets and session voice presets individually, each in a list
def get_voice_presets(session_id):
system_presets, session_presets = [], []
# Load system presets
system_presets = load_voice_presets_metadata(utils.get_system_voice_preset_path())
# Load session presets
session_presets = load_voice_presets_metadata(
utils.get_session_voice_preset_path(session_id),
safe_if_metadata_not_exist=True
)
return system_presets, session_presets
# return merged voice presets in a {voice_preset_name: voice_preset} dict
def get_merged_voice_presets(session_id):
system_presets, session_presets = get_voice_presets(session_id)
res = {}
for preset in list(system_presets.values()) + list(session_presets.values()):
res[preset['id']] = preset # session presets with the same id will cover that of system presets
return res
def add_voice_preset(voice_presets_path, presets, id, desc, wav_file_path):
if id in presets:
raise KeyError(f'{id} already in voice preset, path={voice_presets_path}!')
# Convert wav to npz
npz_path = voice_presets_path / 'npz'
VP(wav_file_path, npz_path)
npz_file_path = npz_path / f'{Path(wav_file_path).stem}.npz'
presets[id] = {
'id': id,
'desc': desc,
'npz_path': str(npz_file_path)
}
save_voice_presets_metadata(voice_presets_path, presets)
return presets[id]
def add_session_voice_preset(id, desc, wav_file_path, session_id):
voice_presets_path = utils.get_session_voice_preset_path(session_id)
os.makedirs(voice_presets_path / 'npz', exist_ok=True)
presets = load_voice_presets_metadata(voice_presets_path, safe_if_metadata_not_exist=True)
if len(presets) >= 3:
raise ValueError(f'session voice presets size exceed 3')
if id in presets:
raise KeyError(f'{id} already in voice preset, path={voice_presets_path}!')
return add_voice_preset(voice_presets_path, presets, id, desc, wav_file_path)
def add_system_voice_preset(id, desc, wav_file_path):
voice_presets_path = utils.get_system_voice_preset_path()
presets = load_voice_presets_metadata(voice_presets_path)
return add_voice_preset(voice_presets_path, presets, id, desc, wav_file_path)
# if session_id set to '', we are removing system voice presets
def remove_session_voice_preset(id, session_id):
voice_presets_path = utils.get_session_voice_preset_path(session_id)
presets = load_voice_presets_metadata(
voice_presets_path,
safe_if_metadata_not_exist=True
)
preset = presets.pop(id)
npz_path = preset['npz_path']
try:
os.remove(npz_path)
except FileNotFoundError:
print(f"INFO: trying to delete {npz_path} which does not exist, path={voice_presets_path}.")
save_voice_presets_metadata(voice_presets_path, presets) |