Update app.py

app.py

@@ -8,12 +8,15 @@ import polars as pl
 
 logging.set_verbosity_error()
 favourite_langs = {"German": "de", "Romanian": "ro", "English": "en", "-----": "-----"}
-all_langs = languagecodes.iso_languages
 df = pl.read_parquet("isolanguages.parquet")
+non_empty_isos = df.slice(1).filter(pl.col("ISO639-1") != "").rows()
+# all_langs = languagecodes.iso_languages
+all_langs = {iso[0]: (iso[1], iso[2], iso[3]) for iso in non_empty_isos}
 
 # Language options as list, add favourite languages first
 options = list(favourite_langs.keys())
 options.extend(list(all_langs.keys()))
+
 models = ["Helsinki-NLP",
           "facebook/nllb-200-distilled-600M", "facebook/nllb-200-distilled-1.3B", "facebook/nllb-200-1.3B", "facebook/nllb-200-3.3B",
           "facebook/mbart-large-50-many-to-many-mmt", "facebook/mbart-large-50-one-to-many-mmt", "facebook/mbart-large-50-many-to-one-mmt",
@@ -110,9 +113,8 @@ class Translators:
             except EnvironmentError as error:
                 try: # Last resort: multi to multi
                     model_name = "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul"
-                    pipe = pipeline("translation", model=model_name)
-                    non_empty_iso = df.slice(1).filter(pl.col("ISO639-1") != "").rows()
-                    iso1_dict = {iso[1]: (iso[0], iso[2], iso[3]) for iso in non_empty_iso}
+                    pipe = pipeline("translation", model=model_name)                
+                    iso1_dict = {iso[1]: (iso[0], iso[2], iso[3]) for iso in non_empty_isos}
                     iso3tl = iso1_dict.get(self.tl)[2] # 'deu', 'ron', 'eng', 'fra'
                     translation = pipe(f'>>{iso3tl}<< {self.input_text}')
                     return translation[0]['translation_text'], f'Translated from {self.sl} to {self.tl} with {model_name}.'
@@ -353,8 +355,8 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
         ("Hallo Welt", "Translated from English to German with Helsinki-NLP.")
     """
     
-    sl = all_langs[s_language]
-    tl = all_langs[t_language]
+    sl = all_langs[s_language][0]
+    tl = all_langs[t_language][0]
     message_text = f'Translated from {s_language} to {t_language} with {model_name}'
     try:    
         if model_name.startswith("Helsinki-NLP"):