mirror of
https://github.com/denizsafak/abogen.git
synced 2026-09-20 11:40:57 +02:00
Artem Akymenko
d3ded8af0e
- voice_resolver: log spec resolution and cache hits - conversion_executor: log chapter start/end, voice resolution, timing - conversion_planner: log plan summary - conversion_service: log entry point params and outcome - conversion_runner: log job params and lifecycle - form.py: log PendingJob creation - synthesize.py: log pipeline creation and device selection
82 lines
2.9 KiB
Python
82 lines
2.9 KiB
Python
"""AppVoiceResolver — voice resolution inside the application layer.
|
|
|
|
Resolves voice specs into loaded voices using profiles, pipeline pool,
|
|
and voice cache. Replaces UI-specific resolvers (WebUIVoiceResolver,
|
|
PyQtVoiceResolver) with a single app-layer implementation.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Any, Dict, Optional
|
|
|
|
from abogen.application.conversion_ports import ResolvedVoice, VoiceResolver
|
|
from abogen.application.conversion_request import ConversionRequest
|
|
from abogen.domain.pipeline_factory import PipelinePool
|
|
from abogen.domain.voice_loader import VoiceCache, resolve_voice
|
|
from abogen.domain.voice_utils import resolve_voice_target
|
|
|
|
|
|
class AppVoiceResolver:
|
|
"""App-layer implementation of VoiceResolver protocol.
|
|
|
|
Uses ConversionRequest instead of Job. Loads profiles, creates
|
|
resolver internally — UIs don't need to manage this.
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
request: ConversionRequest,
|
|
normalized_profiles: Dict[str, Dict[str, Any]],
|
|
pool: PipelinePool,
|
|
cache: VoiceCache,
|
|
):
|
|
self._request = request
|
|
self._profiles = normalized_profiles
|
|
self._cache = cache
|
|
self._pool = pool
|
|
|
|
def resolve(self, voice_spec: str) -> ResolvedVoice:
|
|
"""Resolve a voice spec into a loaded voice."""
|
|
provider, resolved, speed, steps = resolve_voice_target(
|
|
voice_spec,
|
|
self._profiles,
|
|
job_voice=self._request.voice,
|
|
job_tts_provider=self._request.tts_provider,
|
|
job_supertonic_total_steps=self._request.supertonic_total_steps,
|
|
job_speed=self._request.speed,
|
|
)
|
|
|
|
cache_key = f"{provider}:{resolved}" if resolved else provider
|
|
cached = self._cache.get(cache_key)
|
|
if cached is not None:
|
|
logging.info("[resolver] Cache hit: spec=%s -> provider=%s resolved=%s", voice_spec, provider, resolved)
|
|
return ResolvedVoice(
|
|
provider=provider,
|
|
resolved_spec=resolved,
|
|
voice=cached,
|
|
speed=speed,
|
|
supertonic_steps=steps or 0,
|
|
)
|
|
|
|
if provider == "kokoro":
|
|
kokoro_backend = self._pool.get(
|
|
"kokoro", self._request.language, self._request.use_gpu,
|
|
)
|
|
loaded = resolve_voice(
|
|
resolved, kokoro_backend, self._request.use_gpu, cache=self._cache,
|
|
)
|
|
else:
|
|
loaded = resolved
|
|
|
|
self._cache.set(cache_key, loaded)
|
|
logging.info("[resolver] Resolved: spec=%s -> provider=%s resolved=%s speed=%.2f steps=%s",
|
|
voice_spec, provider, resolved, speed, steps)
|
|
return ResolvedVoice(
|
|
provider=provider,
|
|
resolved_spec=resolved,
|
|
voice=loaded,
|
|
speed=speed,
|
|
supertonic_steps=steps or 0,
|
|
)
|