"""Translation-internal language metadata."""

from __future__ import annotations

from typing import Dict, Tuple


LANGUAGE_LABELS: Dict[str, str] = {
    "zh": "Chinese",
    "en": "English",
    "ru": "Russian",
    "ar": "Arabic",
    "ja": "Japanese",
    "es": "Spanish",
    "de": "German",
    "fr": "French",
    "it": "Italian",
    "pt": "Portuguese",
}


QWEN_LANGUAGE_CODES: Dict[str, str] = {
    "zh": "Chinese",
    "en": "English",
    "ru": "Russian",
    "ar": "Arabic",
    "ja": "Japanese",
    "es": "Spanish",
    "de": "German",
    "fr": "French",
    "it": "Italian",
    "pt": "Portuguese",
}


DEEPL_LANGUAGE_CODES: Dict[str, str] = {
    "zh": "ZH",
    "en": "EN",
    "ru": "RU",
    "ar": "AR",
    "ja": "JA",
    "es": "ES",
    "de": "DE",
    "fr": "FR",
    "it": "IT",
    "pt": "PT",
}


NLLB_LANGUAGE_CODES: Dict[str, str] = {
    "en": "eng_Latn",
    "zh": "zho_Hans",
    "ru": "rus_Cyrl",
    "ar": "arb_Arab",
    "ja": "jpn_Jpan",
    "es": "spa_Latn",
    "de": "deu_Latn",
    "fr": "fra_Latn",
    "it": "ita_Latn",
    "pt": "por_Latn",
}


MARIAN_LANGUAGE_DIRECTIONS: Dict[str, Tuple[str, str]] = {
    "opus-mt-zh-en": ("zh", "en"),
    "opus-mt-en-zh": ("en", "zh"),
}