import logging

from pydantic import BaseModel
from pydantic_settings import BaseSettings
from typing import Optional, Literal

logger = logging.getLogger(__name__)


class ModelSettings(BaseSettings):
    asr_model: str = "distil-whisper/distil-large-v3"

class InferenceConfig(BaseModel):
    task: Literal["transcribe", "translate"] = "transcribe"
    batch_size: int = 24
    chunk_length_s: int = 30
    sampling_rate: int = 16000
    language: Optional[str] = None


model_settings = ModelSettings()

logger.info(f"asr model: {model_settings.asr_model}")