mirror of
https://github.com/RVC-Boss/GPT-SoVITS.git
synced 2026-05-19 17:08:11 +08:00
800f01790e
Rename several methods in EngineApiFacade to follow a consistent private naming convention, enhancing code clarity. Update EngineApiDelegates to remove redundant method definitions, streamlining the interface. Introduce EnginePublicInterface to encapsulate public API methods, improving organization and maintainability of the TTS system. Additionally, update the EngineCompositionBuilder to use the new scheduler worker state retrieval method.
54 lines
1.9 KiB
Python
54 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
from GPT_SoVITS.TTS_infer_pack.unified_engine_components import DirectTTSExecution, SchedulerDebugExecution, SchedulerSubmitExecution
|
|
|
|
|
|
class EnginePublicInterface:
|
|
PUBLIC_API_METHODS = (
|
|
"run_direct_tts_async",
|
|
"run_scheduler_submit",
|
|
"run_scheduler_debug",
|
|
"get_runtime_state",
|
|
"set_refer_audio",
|
|
"set_gpt_weights",
|
|
"set_sovits_weights",
|
|
"handle_control",
|
|
)
|
|
|
|
async def run_direct_tts_async(self, req: dict) -> DirectTTSExecution:
|
|
return await self.api_facade.run_direct_tts_async(req)
|
|
|
|
async def run_scheduler_debug(self, request_items: list[dict], max_steps: int, seed: int) -> SchedulerDebugExecution:
|
|
return await self.api_facade.run_scheduler_debug(request_items, max_steps, seed)
|
|
|
|
async def run_scheduler_submit(self, payload: dict) -> SchedulerSubmitExecution:
|
|
return await self.api_facade.run_scheduler_submit(payload)
|
|
|
|
def get_runtime_state(self) -> dict:
|
|
return self.runtime_facade.get_runtime_state()
|
|
|
|
def set_refer_audio(self, refer_audio_path: str | None) -> dict:
|
|
return self.runtime_facade.set_refer_audio(refer_audio_path)
|
|
|
|
def set_gpt_weights(self, weights_path: str) -> dict:
|
|
return self.runtime_facade.set_gpt_weights(weights_path)
|
|
|
|
def set_sovits_weights(self, weights_path: str) -> dict:
|
|
return self.runtime_facade.set_sovits_weights(weights_path)
|
|
|
|
def handle_control(self, command: str) -> None:
|
|
self.runtime_facade.handle_control(command)
|
|
|
|
|
|
class EngineCompatInterface:
|
|
COMPAT_API_METHODS = (
|
|
"run_direct_tts",
|
|
"get_scheduler_state",
|
|
)
|
|
|
|
def run_direct_tts(self, req: dict) -> DirectTTSExecution:
|
|
return self.api_facade.run_direct_tts(req)
|
|
|
|
def get_scheduler_state(self) -> dict:
|
|
return self.runtime_facade.get_scheduler_state()
|