import os
from pydantic import BaseModel

VENV_DIR = os.path.join(os.path.dirname(__file__), './../../venv')

class TranslatorLangConfig(BaseModel):
    translator_llm: str = 'llama3.2:3b-instruct-q4_0' # TODO: this was migrated to translator_models - remove this
    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 TranslatorConfig(BaseModel):
    deu: TranslatorLangConfig = TranslatorLangConfig(tts_model = 'de_DE-thorsten-high')
    eng: TranslatorLangConfig = TranslatorLangConfig(tts_model = 'en_GB-alba-medium')
    fra: TranslatorLangConfig = TranslatorLangConfig(tts_model = 'fr_FR-siwis-medium')
    spa: TranslatorLangConfig = TranslatorLangConfig(tts_model = 'es_ES-sharvard-medium')
    ita: TranslatorLangConfig = TranslatorLangConfig(tts_model = 'it_IT-paola-medium')