fine tuning + language quality changes

backend/requirements-convert.txt

@@ -2,3 +2,6 @@
 # Runtime inference does not need torch. Install with the CPU wheel index:
 #   pip install -r requirements-convert.txt --extra-index-url https://download.pytorch.org/whl/cpu
 torch==2.5.1
+transformers
+sentencepiece
+protobuf

requirements.txt

@@ -2,3 +2,4 @@ Django>=5.2
 djangorestframework>=3.15
 python-dotenv>=1.0
 requests>=2.31
+langdetect>=1.0.9

translator_app/translator/services/translation_service.py

@@ -8,8 +8,10 @@ from django.conf import settings
 from django.utils import timezone
 
 from ..models import TranslationCache
+from langdetect import detect, DetectorFactory, LangDetectException
 
 logger = logging.getLogger(__name__)
+DetectorFactory.seed = 0
 
 
 class TranslationService:
@@ -17,10 +19,22 @@ class TranslationService:
 
     # NLLB requires its FLORES language tags. MADLAD accepts ISO language codes.
     NLLB_LANGUAGE_CODES = {
-        'en': 'eng_Latn', 'es': 'spa_Latn', 'fr': 'fra_Latn', 'de': 'deu_Latn',
-        'it': 'ita_Latn', 'pt': 'por_Latn', 'ru': 'rus_Cyrl', 'zh': 'zho_Hans',
-        'ja': 'jpn_Jpan', 'ko': 'kor_Hang', 'ar': 'arb_Arab', 'hi': 'hin_Deva',
-        'sw': 'swh_Latn', 'yo': 'yor_Latn', 'ha': 'hau_Latn', 'ig': 'ibo_Latn',
+        "en": "eng_Latn",
+        "es": "spa_Latn",
+        "fr": "fra_Latn",
+        "de": "deu_Latn",
+        "it": "ita_Latn",
+        "pt": "por_Latn",
+        "ru": "rus_Cyrl",
+        "zh": "zho_Hans",
+        "ja": "jpn_Jpan",
+        "ko": "kor_Hang",
+        "ar": "arb_Arab",
+        "hi": "hin_Deva",
+        "sw": "swh_Latn",
+        "yo": "yor_Latn",
+        "ha": "hau_Latn",
+        "ig": "ibo_Latn",
     }
 
     def __init__(self):
@@ -28,106 +42,178 @@ class TranslationService:
 
     def _get_supported_languages(self):
         return {
-            'en': 'English', 'es': 'Spanish', 'fr': 'French', 'de': 'German',
-            'it': 'Italian', 'pt': 'Portuguese', 'ru': 'Russian', 'zh': 'Chinese',
-            'ja': 'Japanese', 'ko': 'Korean', 'ar': 'Arabic', 'hi': 'Hindi',
-            'sw': 'Swahili', 'yo': 'Yoruba', 'ha': 'Hausa', 'ig': 'Igbo',
+            "en": "English",
+            "es": "Spanish",
+            "fr": "French",
+            "de": "German",
+            "it": "Italian",
+            "pt": "Portuguese",
+            "ru": "Russian",
+            "zh": "Chinese",
+            "ja": "Japanese",
+            "ko": "Korean",
+            "ar": "Arabic",
+            "hi": "Hindi",
+            "sw": "Swahili",
+            "yo": "Yoruba",
+            "ha": "Hausa",
+            "ig": "Igbo",
         }
 
     def get_supported_languages(self):
-        return [{'code': code, 'name': name} for code, name in self.supported_languages.items()]
+        return [
+            {"code": code, "name": name}
+            for code, name in self.supported_languages.items()
+        ]
 
     def detect_language(self, text):
-        # Deliberately local and deterministic until the ML team supplies a detector.
-        code = 'en' if all(ord(char) < 128 for char in text) else 'en'
-        return {'code': code, 'name': self.supported_languages[code], 'confidence': 0.6}
+        try:
+            code = detect(text)
+        except LangDetectException:
+            code = "en"
+        if code not in self.supported_languages:
+            code = "en"  # fall back if detected language isn't one you support
+        return {
+            "code": code,
+            "name": self.supported_languages.get(code, code.upper()),
+            "confidence": 0.8 if code in self.supported_languages else 0.3,
+        }
 
     def translate(self, text, target_language, source_language=None, model=None):
         if not text or not text.strip():
-            return {'error': 'Text cannot be empty'}
+            return {"error": "Text cannot be empty"}
 
         model = model or settings.TRANSLATION_DEFAULT_MODEL
-        source_language = source_language or self.detect_language(text)['code']
-        cache_key = self._generate_cache_key(text, source_language, target_language, model)
+        if model == "madlad":
+            source_language = None  # MADLAD auto-detects; any provided value is ignored
+        source_language = source_language or self.detect_language(text)["code"]
+        cache_key = self._generate_cache_key(
+            text, source_language, target_language, model
+        )
         cached = self._get_from_cache(cache_key)
         if cached:
-            cached['model'] = model
+            cached["model"] = model
             return cached
 
-        if model == 'deepl_fallback':
+        if model == "deepl_fallback":
             outcome = self._translate_deepl(text, source_language, target_language)
         elif model in settings.TRANSLATION_MODEL_ENGINES:
-            outcome = self._translate_model_api(text, source_language, target_language, model)
+            outcome = self._translate_model_api(
+                text, source_language, target_language, model
+            )
         else:
-            return {'error': 'The selected translation model is unavailable.'}
+            return {"error": "The selected translation model is unavailable."}
 
-        if outcome.get('error'):
+        if outcome.get("error"):
             return outcome
 
-        translated_text = outcome['translated_text']
-        self._save_to_cache(cache_key, text, translated_text, source_language, target_language)
+        translated_text = outcome["translated_text"]
+        self._save_to_cache(
+            cache_key, text, translated_text, source_language, target_language
+        )
         return {
-            'source_text': text,
-            'translated_text': translated_text,
-            'source_language': source_language,
-            'target_language': target_language,
-            'model': model,
+            "source_text": text,
+            "translated_text": translated_text,
+            "source_language": source_language,
+            "target_language": target_language,
+            "model": model,
         }
 
     def _translate_model_api(self, text, source_language, target_language, model):
-        """Call the FastAPI model server deployed as a Hugging Face Space."""
         if not settings.MODEL_API_URL:
-            return {'error': 'ML translation is not configured. Add MODEL_API_URL to your environment.'}
+            return {
+                "error": "ML translation is not configured. Add MODEL_API_URL to your environment."
+            }
+
+        if isinstance(settings.TRANSLATION_MODEL_ENGINES, dict):
+            engine_name = settings.TRANSLATION_MODEL_ENGINES.get(model, model)
+        else:
+            engine_name = model
+
+        # Every local engine (NLLB, MADLAD) speaks FLORES-200 codes on the wire.
         try:
             source = self.NLLB_LANGUAGE_CODES[source_language]
             target = self.NLLB_LANGUAGE_CODES[target_language]
         except KeyError:
-            return {'error': 'The model service does not support one of these languages.'}
+            return {
+                "error": "The model service does not support one of these languages."
+            }
+
+        payload = {
+            "text": text,
+            "source": source,
+            "target": target,
+            "engine": engine_name,
+        }
+
+        # DEBUGGING: Print exactly what is being sent to the console
+        print(
+            f"\n--- API PAYLOAD DEBUG ---\nURL: {settings.MODEL_API_URL}\nPayload: {payload}\n-------------------------\n"
+        )
+        logger.info(f"Sending translation payload: {payload}")
 
         try:
             response = requests.post(
                 settings.MODEL_API_URL,
-                json={'text': text, 'source': source, 'target': target,
-                      'engine': settings.TRANSLATION_MODEL_ENGINES[model]},
+                json=payload,
                 timeout=60,
             )
+
             if response.status_code != 200:
-                logger.warning('Model API translation failed: %s %s', response.status_code, response.text)
-                return {'error': 'The ML translation service could not complete this request.'}
-            translated_text = response.json().get('translation')
+                logger.warning(
+                    "Model API translation failed: %s %s",
+                    response.status_code,
+                    response.text,
+                )
+                return {
+                    "error": f"The ML service failed with status {response.status_code}."
+                }
+
+            translated_text = response.json().get("translation")
             if not translated_text:
-                return {'error': 'The ML model returned no translation.'}
-            return {'translated_text': translated_text}
+                return {"error": "The ML model returned no translation."}
+            return {"translated_text": translated_text}
+
         except requests.exceptions.Timeout:
-            return {'error': 'The ML translation request timed out.'}
+            return {"error": "The ML translation request timed out."}
         except requests.exceptions.ConnectionError:
-            return {'error': 'Could not connect to the ML translation service.'}
+            return {"error": "Could not connect to the ML translation service."}
         except (ValueError, requests.RequestException) as error:
-            logger.exception('Model API translation failed: %s', error)
-            return {'error': 'The ML translation service returned an invalid response.'}
+            logger.exception("Model API translation failed: %s", error)
+            return {"error": "The ML translation service returned an invalid response."}
 
     def _translate_deepl(self, text, source_language, target_language):
         if not settings.TRANSLATION_API_KEY:
-            return {'error': 'Traditional fallback is not configured.'}
+            return {"error": "Traditional fallback is not configured."}
         try:
             response = requests.post(
                 settings.TRANSLATION_API_URL,
-                headers={'Authorization': f'DeepL-Auth-Key {settings.TRANSLATION_API_KEY}'},
-                json={'text': [text], 'source_lang': source_language.upper(), 'target_lang': target_language.upper()},
+                headers={
+                    "Authorization": f"DeepL-Auth-Key {settings.TRANSLATION_API_KEY}"
+                },
+                json={
+                    "text": [text],
+                    "source_lang": source_language.upper(),
+                    "target_lang": target_language.upper(),
+                },
                 timeout=30,
             )
             if response.status_code != 200:
-                return {'error': 'The fallback translation service could not complete this request.'}
-            translated_text = response.json()['translations'][0]['text']
-            return {'translated_text': translated_text}
+                return {
+                    "error": "The fallback translation service could not complete this request."
+                }
+            translated_text = response.json()["translations"][0]["text"]
+            return {"translated_text": translated_text}
         except (KeyError, IndexError, ValueError, requests.RequestException) as error:
-            logger.exception('Fallback translation failed: %s', error)
-            return {'error': 'The fallback translation service returned an invalid response.'}
+            logger.exception("Fallback translation failed: %s", error)
+            return {
+                "error": "The fallback translation service returned an invalid response."
+            }
 
     @staticmethod
     def _generate_cache_key(text, source_language, target_language, model):
-        content = f'{text}|{source_language}|{target_language}|{model}'
-        return hashlib.sha256(content.encode('utf-8')).hexdigest()
+        content = f"{text}|{source_language}|{target_language}|{model}"
+        return hashlib.sha256(content.encode("utf-8")).hexdigest()
 
     @staticmethod
     def _get_from_cache(cache_key):
@@ -137,21 +223,31 @@ class TranslationService:
                 return None
             entry.access_count += 1
             entry.last_accessed = timezone.now()
-            entry.save(update_fields=['access_count', 'last_accessed'])
-            return {'source_text': entry.source_text, 'translated_text': entry.translated_text,
-                    'source_language': entry.source_language_code, 'target_language': entry.target_language_code,
-                    'from_cache': True}
+            entry.save(update_fields=["access_count", "last_accessed"])
+            return {
+                "source_text": entry.source_text,
+                "translated_text": entry.translated_text,
+                "source_language": entry.source_language_code,
+                "target_language": entry.target_language_code,
+                "from_cache": True,
+            }
         except Exception:
-            logger.exception('Translation cache lookup failed')
+            logger.exception("Translation cache lookup failed")
             return None
 
     @staticmethod
-    def _save_to_cache(cache_key, source_text, translated_text, source_language, target_language):
+    def _save_to_cache(
+        cache_key, source_text, translated_text, source_language, target_language
+    ):
         try:
             TranslationCache.objects.update_or_create(
                 source_text_hash=cache_key,
-                defaults={'source_text': source_text, 'translated_text': translated_text,
-                          'source_language_code': source_language, 'target_language_code': target_language},
+                defaults={
+                    "source_text": source_text,
+                    "translated_text": translated_text,
+                    "source_language_code": source_language,
+                    "target_language_code": target_language,
+                },
             )
         except Exception:
-            logger.exception('Translation cache save failed')
+            logger.exception("Translation cache save failed")

translator_app/translator/templates/translator/translator.html

@@ -4,44 +4,77 @@
   <form id="translation-form">
     {% csrf_token %}
     <div class="flex flex-wrap justify-center items-center gap-3 mb-5">
-      <select id="source-language" aria-label="Source language" class="rounded-full border-stone-200 bg-white px-5 py-2.5 text-sm font-semibold text-primary shadow-sm focus:border-primary focus:ring-primary">
-        <option value="">Detect language</option>{% for language in languages %}<option value="{{ language.code }}"{% if language.code == 'en' %} selected{% endif %}>{{ language.name }}</option>{% endfor %}
+      <select id="source-language" aria-label="Source language"
+        class="rounded-full border-stone-200 bg-white px-5 py-2.5 text-sm font-semibold text-primary shadow-sm focus:border-primary focus:ring-primary">
+        <option value="">Detect language</option>{% for language in languages %}<option value="{{ language.code }}" {%
+          if language.code=='en' %} selected{% endif %}>{{ language.name }}</option>{% endfor %}
       </select>
-      <button id="swap-languages" type="button" title="Swap languages" class="rounded-full bg-sand px-3 py-2.5 text-primary transition hover:rotate-180">⇄</button>
-      <select id="target-language" aria-label="Target language" class="rounded-full border-stone-200 bg-white px-5 py-2.5 text-sm font-semibold text-primary shadow-sm focus:border-primary focus:ring-primary">
-        {% for language in languages %}<option value="{{ language.code }}"{% if language.code == 'sw' %} selected{% endif %}>{{ language.name }}</option>{% endfor %}
+      <button id="swap-languages" type="button" title="Swap languages"
+        class="rounded-full bg-sand px-3 py-2.5 text-primary transition hover:rotate-180">⇄</button>
+      <select id="target-language" aria-label="Target language"
+        class="rounded-full border-stone-200 bg-white px-5 py-2.5 text-sm font-semibold text-primary shadow-sm focus:border-primary focus:ring-primary">
+        {% for language in languages %}<option value="{{ language.code }}" {% if language.code=='sw' %} selected{% endif
+          %}>{{ language.name }}</option>{% endfor %}
       </select>
-      <label class="flex items-center gap-2 rounded-full border border-stone-200 bg-white px-4 py-2 text-sm font-semibold text-stone-600 shadow-sm">Model
-        <select id="translation-model" aria-label="Translation model" class="border-0 bg-transparent py-0 pl-1 pr-7 text-primary focus:ring-0">
-          {% for value, label in models %}<option value="{{ value }}"{% if value == default_model %} selected{% endif %}>{{ label }}</option>{% endfor %}
+      <label
+        class="flex items-center gap-2 rounded-full border border-stone-200 bg-white px-4 py-2 text-sm font-semibold text-stone-600 shadow-sm">Model
+        <select id="translation-model" aria-label="Translation model"
+          class="border-0 bg-transparent py-0 pl-1 pr-7 text-primary focus:ring-0">
+          {% for value, label in models %}<option value="{{ value }}" {% if value==default_model %} selected{% endif %}>
+            {{ label }}</option>{% endfor %}
         </select>
       </label>
     </div>
     <div class="rounded-2xl border border-stone-200 bg-white p-5 md:p-7 shadow-xl shadow-stone-200/40 md:flex md:gap-7">
       <section class="flex min-h-72 flex-1 flex-col">
         <label for="source-text" class="mb-3 text-xs font-bold uppercase tracking-widest text-stone-400">Source</label>
-        <textarea id="source-text" maxlength="5000" class="min-h-52 w-full flex-1 resize-none border-0 p-0 font-display text-2xl leading-relaxed text-primary placeholder:text-stone-300 focus:ring-0" placeholder="Enter text to translate..."></textarea>
-        <div class="mt-3 flex justify-between border-t border-stone-100 pt-3 text-xs text-stone-400"><button id="clear-source" type="button" class="font-semibold text-primary hover:underline">Clear</button><span id="character-count">0 / 5000 characters</span></div>
+        <textarea id="source-text" maxlength="5000"
+          class="min-h-52 w-full flex-1 resize-none border-0 p-0 font-display text-2xl leading-relaxed text-primary placeholder:text-stone-300 focus:ring-0"
+          placeholder="Enter text to translate..."></textarea>
+        <div class="mt-3 flex justify-between border-t border-stone-100 pt-3 text-xs text-stone-400"><button
+            id="clear-source" type="button" class="font-semibold text-primary hover:underline">Clear</button><span
+            id="character-count">0 / 5000 characters</span></div>
       </section>
       <div class="my-5 hidden w-px bg-stone-100 md:block"></div>
       <section class="relative flex min-h-72 flex-1 flex-col">
-        <div class="mb-3 flex items-center justify-between"><span class="text-xs font-bold uppercase tracking-widest text-stone-400">Translation</span><button id="copy-result" type="button" class="hidden rounded-full px-3 py-1 text-xs font-bold text-primary hover:bg-mist">Copy</button></div>
-        <p id="translated-text" aria-live="polite" class="relative flex-1 whitespace-pre-wrap font-display text-2xl leading-relaxed text-primary"><span class="text-stone-300">Your translation will appear here.</span></p>
+        <div class="mb-3 flex items-center justify-between"><span
+            class="text-xs font-bold uppercase tracking-widest text-stone-400">Translation</span><button
+            id="copy-result" type="button"
+            class="hidden rounded-full px-3 py-1 text-xs font-bold text-primary hover:bg-mist">Copy</button></div>
+        <p id="translated-text" aria-live="polite"
+          class="relative flex-1 whitespace-pre-wrap font-display text-2xl leading-relaxed text-primary"><span
+            class="text-stone-300">Your translation will appear here.</span></p>
         <p id="result-meta" class="mt-3 text-xs text-stone-400"></p>
       </section>
     </div>
-    <div class="mt-6 flex flex-col items-center gap-3"><button id="translate-button" class="rounded-full bg-primary px-10 py-4 font-display text-lg font-bold uppercase tracking-wider text-white shadow-lg shadow-primary/30 transition hover:scale-105 disabled:cursor-wait disabled:opacity-70">Translate</button><p id="form-message" role="alert" class="min-h-5 text-sm text-red-700"></p></div>
+    <div class="mt-6 flex flex-col items-center gap-3"><button id="translate-button"
+        class="rounded-full bg-primary px-10 py-4 font-display text-lg font-bold uppercase tracking-wider text-white shadow-lg shadow-primary/30 transition hover:scale-105 disabled:cursor-wait disabled:opacity-70">Translate</button>
+      <p id="form-message" role="alert" class="min-h-5 text-sm text-red-700"></p>
+    </div>
   </form>
-  <div class="mt-12 text-center"><a href="{% url 'translator:history' %}" class="text-sm font-bold text-primary underline underline-offset-4 hover:text-stone-700">View history</a></div>
+  <div class="mt-12 text-center"><a href="{% url 'translator:history' %}"
+      class="text-sm font-bold text-primary underline underline-offset-4 hover:text-stone-700">View history</a></div>
 </main>
 <script>
-const form=document.querySelector('#translation-form'), source=document.querySelector('#source-text'), result=document.querySelector('#translated-text'), message=document.querySelector('#form-message'), button=document.querySelector('#translate-button'), count=document.querySelector('#character-count');
-const csrf=document.querySelector('[name=csrfmiddlewaretoken]').value;
-source.addEventListener('input',()=>count.textContent=`${source.value.length} / 5000 characters`);
-document.querySelector('#clear-source').onclick=()=>{source.value='';source.dispatchEvent(new Event('input'));source.focus()};
-document.querySelector('#swap-languages').onclick=()=>{const a=document.querySelector('#source-language'),b=document.querySelector('#target-language');if(a.value){[a.value,b.value]=[b.value,a.value]}};
-document.querySelector('#copy-result').onclick=async()=>{await navigator.clipboard.writeText(result.textContent);document.querySelector('#copy-result').textContent='Copied';setTimeout(()=>document.querySelector('#copy-result').textContent='Copy',1200)};
-form.addEventListener('submit',async e=>{e.preventDefault();message.textContent='';button.disabled=true;button.textContent='Translating…';result.classList.add('quiet-loading');
-  try {const response=await fetch("{% url 'translator:translate_api' %}",{method:'POST',headers:{'Content-Type':'application/json','X-CSRFToken':csrf},body:JSON.stringify({text:source.value,source_language:document.querySelector('#source-language').value,target_language:document.querySelector('#target-language').value,model:document.querySelector('#translation-model').value})});const data=await response.json();if(!response.ok)throw Error(data.error||'Translation could not be completed.');result.textContent=data.translated_text;document.querySelector('#result-meta').textContent=`${data.model_label||'Translation'} · saved to history`;document.querySelector('#copy-result').classList.remove('hidden')}catch(error){message.textContent=error.message;result.innerHTML='<span class="text-stone-300">Your translation will appear here.</span>'}finally{button.disabled=false;button.textContent='Translate';result.classList.remove('quiet-loading')}});
+  const form = document.querySelector('#translation-form'), source = document.querySelector('#source-text'), result = document.querySelector('#translated-text'), message = document.querySelector('#form-message'), button = document.querySelector('#translate-button'), count = document.querySelector('#character-count');
+  const csrf = document.querySelector('[name=csrfmiddlewaretoken]').value;
+  const modelSelect = document.querySelector('#translation-model');
+  const sourceSelect = document.querySelector('#source-language');
+
+  function syncSourceForModel() {
+    const isMadlad = modelSelect.value === 'madlad';
+    sourceSelect.disabled = isMadlad;
+    if (isMadlad) sourceSelect.value = '';
+  }
+  modelSelect.addEventListener('change', syncSourceForModel);
+  syncSourceForModel(); // run once on load in case MADLAD is the default
+  source.addEventListener('input', () => count.textContent = `${source.value.length} / 5000 characters`);
+  document.querySelector('#clear-source').onclick = () => { source.value = ''; source.dispatchEvent(new Event('input')); source.focus() };
+  document.querySelector('#swap-languages').onclick = () => { const a = document.querySelector('#source-language'), b = document.querySelector('#target-language'); if (a.value) { [a.value, b.value] = [b.value, a.value] } };
+  document.querySelector('#copy-result').onclick = async () => { await navigator.clipboard.writeText(result.textContent); document.querySelector('#copy-result').textContent = 'Copied'; setTimeout(() => document.querySelector('#copy-result').textContent = 'Copy', 1200) };
+  form.addEventListener('submit', async e => {
+    e.preventDefault(); message.textContent = ''; button.disabled = true; button.textContent = 'Translating…'; result.classList.add('quiet-loading');
+    try { const response = await fetch("{% url 'translator:translate_api' %}", { method: 'POST', headers: { 'Content-Type': 'application/json', 'X-CSRFToken': csrf }, body: JSON.stringify({ text: source.value, source_language: document.querySelector('#source-language').value, target_language: document.querySelector('#target-language').value, model: document.querySelector('#translation-model').value }) }); const data = await response.json(); if (!response.ok) throw Error(data.error || 'Translation could not be completed.'); result.textContent = data.translated_text; document.querySelector('#result-meta').textContent = `${data.model_label || 'Translation'} · saved to history`; document.querySelector('#copy-result').classList.remove('hidden') } catch (error) { message.textContent = error.message; result.innerHTML = '<span class="text-stone-300">Your translation will appear here.</span>' } finally { button.disabled = false; button.textContent = 'Translate'; result.classList.remove('quiet-loading') }
+  });
 </script>
-{% endblock %}
+{% endblock %}

translator_app/translator_project/settings.py

@@ -12,7 +12,10 @@ SECRET_KEY = os.getenv("DJANGO_SECRET_KEY", "your-secret-key-here")
 
 DEBUG = True
 
-ALLOWED_HOSTS = ["localhost", "127.0.0.1"]
+ALLOWED_HOSTS = [
+    "localhost",
+    "127.0.0.1",
+]
 
 # Application definition
 INSTALLED_APPS = [
@@ -101,16 +104,16 @@ DEFAULT_AUTO_FIELD = "django.db.models.BigAutoField"
 # ML translation service. Point this at the deployed Hugging Face Space, e.g.
 # https://<account>-<space>.hf.space/api/translate
 MODEL_API_URL = os.getenv("MODEL_API_URL")
-TRANSLATION_DEFAULT_MODEL = os.getenv("TRANSLATION_DEFAULT_MODEL", "nllb_600m")
+TRANSLATION_DEFAULT_MODEL = os.getenv("TRANSLATION_DEFAULT_MODEL", "nllb")
 TRANSLATION_MODELS = (
-    ("nllb_600m", "NLLB-200 (600M · Fast)"),
-    ("nllb_1.3b", "NLLB-200 (1.3B · Accurate)"),
-    ("madlad_3b", "MADLAD-400 (3B · Large)"),
+    ("nllb", "NLLB-200 (600M · Fast)"),
+    ("nllb_1_3b", "NLLB-200 (1.3B · Accurate)"),
+    ("madlad", "MADLAD-400 (3B · Large)"),
 )
 TRANSLATION_MODEL_ENGINES = {
-    "nllb_600m": "nllb",
-    "nllb_1.3b": "nllb_1_3b",
-    "madlad_3b": "madlad",
+    "nllb": "nllb",
+    "nllb_1_3b": "nllb_1_3b",
+    "madlad": "madlad",
 }
 
 # Optional traditional-provider fallback. Add this option to TRANSLATION_MODELS
@@ -120,6 +123,6 @@ TRANSLATION_API_URL = os.getenv(
     "TRANSLATION_API_URL", "https://api-free.deepl.com/v2/translate"
 )
 if TRANSLATION_API_KEY:
-    TRANSLATION_MODELS += (("deepl_fallback", "DeepL · Traditional fallback"),)
+    TRANSLATION_MODELS += (("deepl_fallback", "DeepL"),)
 # Or for Google Translate:
 # TRANSLATION_API_URL = 'https://translation.googleapis.com/language/translate/v2'