Spaces:
Running
Running
| import logging | |
| from typing import Optional, List | |
| # Configure logging | |
| logger = logging.getLogger(__name__) | |
| # Import the base class | |
| from utils.tts_base import TTSEngineBase, DummyTTSEngine | |
| class TTSFactory: | |
| """Factory class for creating TTS engines | |
| This class is responsible for creating the appropriate TTS engine based on | |
| availability and configuration. | |
| """ | |
| def create_engine(engine_type: Optional[str] = None, lang_code: str = 'z') -> TTSEngineBase: | |
| """Create a TTS engine instance | |
| Args: | |
| engine_type (str, optional): Type of engine to create ('kokoro', 'kokoro_space', 'dia', 'dummy') | |
| If None, the best available engine will be used | |
| lang_code (str): Language code for the engine | |
| Returns: | |
| TTSEngineBase: An instance of a TTS engine | |
| """ | |
| from utils.tts_engines import get_available_engines, create_engine | |
| # Get available engines | |
| available_engines = get_available_engines() | |
| logger.info(f"Available TTS engines: {available_engines}") | |
| # If engine_type is specified, try to create that specific engine | |
| if engine_type is not None: | |
| if engine_type in available_engines: | |
| logger.info(f"Creating requested engine: {engine_type}") | |
| return create_engine(engine_type, lang_code) | |
| else: | |
| logger.warning(f"Requested engine '{engine_type}' is not available") | |
| # Try to create the best available engine | |
| # Priority: kokoro > kokoro_space > dia > dummy | |
| for engine in ['kokoro', 'kokoro_space', 'dia']: | |
| if engine in available_engines: | |
| logger.info(f"Creating best available engine: {engine}") | |
| return create_engine(engine, lang_code) | |
| # Fall back to dummy engine | |
| logger.warning("No TTS engines available, falling back to dummy engine") | |
| return DummyTTSEngine(lang_code) |