import os
from pydantic import BaseModel
from auracast import auracast_config

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

class TranslatorBaseconfig(BaseModel):
    big: auracast_config.AuracastBigConfig = auracast_config.AuracastBigConfigDe()
    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(TranslatorBaseconfig):
    big: auracast_config.AuracastBigConfig = auracast_config.AuracastBigConfigDe()
    tts_model: str ='de_DE-thorsten-high'

class TranslatorConfigEn(TranslatorBaseconfig):
    big: auracast_config.AuracastBigConfig = auracast_config.AuracastBigConfigEn()
    tts_model: str = 'en_GB-alba-medium'

class TranslatorConfigFr(TranslatorBaseconfig):
    big: auracast_config.AuracastBigConfig = auracast_config.AuracastBigConfigFr()
    tts_model: str = 'fr_FR-siwis-medium'

class TranslatorConfigEs(TranslatorBaseconfig):
    big: auracast_config.AuracastBigConfig = auracast_config.AuracastBigConfigEs()
    tts_model: str = 'es_ES-sharvard-medium'

class TranslatorConfigIt(TranslatorBaseconfig):
    big: auracast_config.AuracastBigConfig = auracast_config.AuracastBigConfigIt()
    tts_model: str = 'it_IT-paola-medium'