36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
import os
|
|
from typing import List
|
|
from pydantic import BaseModel
|
|
from auracast import auracast_config
|
|
|
|
VENV_DIR = os.path.join(os.path.dirname(__file__), './../../venv')
|
|
|
|
class TranslatorLangConfig(auracast_config.AuracastBigConfig):
|
|
translator_llm: str = 'llama3.2:3b-instruct-q4_0'
|
|
llm_client: str = 'ollama'
|
|
llm_host_url: str | None = 'http://localhost:11434'
|
|
llm_host_token: str | None = None
|
|
tts_system: str = 'piper'
|
|
tts_model: str ='de_DE-kerstin-low'
|
|
|
|
|
|
class TranslatorConfigDe(TranslatorLangConfig, auracast_config.AuracastBigConfigDe):
|
|
tts_model: str ='de_DE-thorsten-high'
|
|
|
|
class TranslatorConfigEn(TranslatorLangConfig, auracast_config.AuracastBigConfigEn):
|
|
tts_model: str = 'en_GB-alba-medium'
|
|
|
|
class TranslatorConfigFr(TranslatorLangConfig, auracast_config.AuracastBigConfigFr):
|
|
tts_model: str = 'fr_FR-siwis-medium'
|
|
|
|
class TranslatorConfigEs(TranslatorLangConfig, auracast_config.AuracastBigConfigEs):
|
|
tts_model: str = 'es_ES-sharvard-medium'
|
|
|
|
class TranslatorConfigIt(TranslatorLangConfig, auracast_config.AuracastBigConfigIt):
|
|
tts_model: str = 'it_IT-paola-medium'
|
|
|
|
class TranslatorConfigGroup(auracast_config.AuracastGlobalConfig):
|
|
bigs: List[TranslatorLangConfig] = [
|
|
TranslatorConfigDe(),
|
|
]
|