import os
from typing import Optional

class Config:
    """Application configuration"""

    # Server settings
    HOST: str = os.getenv("HOST", "0.0.0.0")
    PORT: int = int(os.getenv("PORT", "8000"))

    # Model settings
    MODEL_NAME: str = os.getenv("MODEL_NAME", "facebook/mbart-large-50-many-to-many-mmt")
    MAX_LENGTH: int = int(os.getenv("MAX_LENGTH", "512"))
    DEVICE: str = os.getenv("DEVICE", "cpu")

    # Supported languages for mBART-50
    SUPPORTED_LANGUAGES = {
        "ko": "ko_KR",  # Korean
        "en": "en_XX",  # English
        "ja": "ja_XX",  # Japanese
        "zh": "zh_CN",  # Chinese (Simplified)
        "es": "es_XX",  # Spanish
        "fr": "fr_XX",  # French
        "de": "de_DE",  # German
        "ru": "ru_RU",  # Russian
        "ar": "ar_AR",  # Arabic
        "hi": "hi_IN",  # Hindi
        "vi": "vi_VN",  # Vietnamese
        "th": "th_TH",  # Thai
        "id": "id_ID",  # Indonesian
        "tr": "tr_TR",  # Turkish
        "pt": "pt_XX",  # Portuguese
        "it": "it_IT",  # Italian
        "nl": "nl_XX",  # Dutch
        "pl": "pl_PL",  # Polish
    }

config = Config()