Spaces:
Running
Running
File size: 7,928 Bytes
329ce4a |
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 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 |
import base64
import json
import os
import shutil
import zipfile
from io import BytesIO
from pathlib import Path
from typing import Dict
from fastapi import HTTPException
from pydantic import ValidationError
from semver.version import Version
from voicevox_engine.model import DownloadableLibrary, InstalledLibrary, VvlibManifest
__all__ = ["LibraryManager"]
INFO_FILE = "metas.json"
class LibraryManager:
def __init__(
self,
library_root_dir: Path,
supported_vvlib_version: str | None,
brand_name: str,
engine_name: str,
engine_uuid: str,
):
self.library_root_dir = library_root_dir
self.library_root_dir.mkdir(exist_ok=True)
if supported_vvlib_version is not None:
self.supported_vvlib_version = Version.parse(supported_vvlib_version)
else:
# supported_vvlib_versionがNoneの時は0.0.0として扱う
self.supported_vvlib_version = Version.parse("0.0.0")
self.engine_brand_name = brand_name
self.engine_name = engine_name
self.engine_uuid = engine_uuid
def downloadable_libraries(self):
# == ダウンロード情報をネットワーク上から取得する場合
# url = "https://example.com/downloadable_libraries.json"
# response = requests.get(url)
# return list(map(DownloadableLibrary.parse_obj, response.json()))
# == ダウンロード情報をjsonファイルから取得する場合
# with open(
# self.root_dir / "engine_manifest_assets" / "downloadable_libraries.json",
# encoding="utf-8",
# ) as f:
# return list(map(DownloadableLibrary.parse_obj, json.load(f)))
# ダミーとして、speaker_infoのアセットを読み込む
with open(
"./engine_manifest_assets/downloadable_libraries.json",
encoding="utf-8",
) as f:
libraries = json.load(f)
speaker_info = libraries[0]["speakers"][0]["speaker_info"]
mock_root_dir = Path("./speaker_info/7ffcb7ce-00ec-4bdc-82cd-45a8889e43ff")
speaker_info["policy"] = (mock_root_dir / "policy.md").read_text()
speaker_info["portrait"] = base64.b64encode(
(mock_root_dir / "portrait.png").read_bytes()
)
for style_info in speaker_info["style_infos"]:
style_id = style_info["id"]
style_info["icon"] = base64.b64encode(
(mock_root_dir / "icons" / f"{style_id}.png").read_bytes()
)
style_info["voice_samples"] = [
base64.b64encode(
(
mock_root_dir / "voice_samples" / f"{style_id}_{i:0>3}.wav"
).read_bytes()
)
for i in range(1, 4)
]
return list(map(DownloadableLibrary.parse_obj, libraries))
def installed_libraries(self) -> Dict[str, InstalledLibrary]:
library = {}
for library_dir in self.library_root_dir.iterdir():
if library_dir.is_dir():
library_uuid = os.path.basename(library_dir)
with open(library_dir / INFO_FILE, encoding="utf-8") as f:
library[library_uuid] = json.load(f)
# アンインストール出来ないライブラリを作る場合、何かしらの条件でFalseを設定する
library[library_uuid]["uninstallable"] = True
return library
def install_library(self, library_id: str, file: BytesIO):
for downloadable_library in self.downloadable_libraries():
if downloadable_library.uuid == library_id:
library_info = downloadable_library.dict()
break
else:
raise HTTPException(
status_code=404, detail=f"指定された音声ライブラリ {library_id} が見つかりません。"
)
library_dir = self.library_root_dir / library_id
library_dir.mkdir(exist_ok=True)
with open(library_dir / INFO_FILE, "w", encoding="utf-8") as f:
json.dump(library_info, f, indent=4, ensure_ascii=False)
if not zipfile.is_zipfile(file):
raise HTTPException(
status_code=422, detail=f"音声ライブラリ {library_id} は不正なファイルです。"
)
with zipfile.ZipFile(file) as zf:
if zf.testzip() is not None:
raise HTTPException(
status_code=422, detail=f"音声ライブラリ {library_id} は不正なファイルです。"
)
# validate manifest version
vvlib_manifest = None
try:
vvlib_manifest = json.loads(
zf.read("vvlib_manifest.json").decode("utf-8")
)
except KeyError:
raise HTTPException(
status_code=422,
detail=f"指定された音声ライブラリ {library_id} にvvlib_manifest.jsonが存在しません。",
)
except Exception:
raise HTTPException(
status_code=422,
detail=f"指定された音声ライブラリ {library_id} のvvlib_manifest.jsonは不正です。",
)
try:
VvlibManifest.validate(vvlib_manifest)
except ValidationError:
raise HTTPException(
status_code=422,
detail=f"指定された音声ライブラリ {library_id} のvvlib_manifest.jsonに不正なデータが含まれています。",
)
if not Version.is_valid(vvlib_manifest["version"]):
raise HTTPException(
status_code=422, detail=f"指定された音声ライブラリ {library_id} のversionが不正です。"
)
try:
vvlib_manifest_version = Version.parse(
vvlib_manifest["manifest_version"]
)
except ValueError:
raise HTTPException(
status_code=422,
detail=f"指定された音声ライブラリ {library_id} のmanifest_versionが不正です。",
)
if vvlib_manifest_version > self.supported_vvlib_version:
raise HTTPException(
status_code=422, detail=f"指定された音声ライブラリ {library_id} は未対応です。"
)
if vvlib_manifest["engine_uuid"] != self.engine_uuid:
raise HTTPException(
status_code=422,
detail=f"指定された音声ライブラリ {library_id} は{self.engine_name}向けではありません。",
)
zf.extractall(library_dir)
return library_dir
def uninstall_library(self, library_id: str):
installed_libraries = self.installed_libraries()
if library_id not in installed_libraries.keys():
raise HTTPException(
status_code=404, detail=f"指定された音声ライブラリ {library_id} はインストールされていません。"
)
if not installed_libraries[library_id]["uninstallable"]:
raise HTTPException(
status_code=403, detail=f"指定された音声ライブラリ {library_id} はアンインストールできません。"
)
try:
shutil.rmtree(self.library_root_dir / library_id)
except Exception:
raise HTTPException(
status_code=500, detail=f"指定された音声ライブラリ {library_id} の削除に失敗しました。"
)
|