Update app.py

app.py

@@ -15,8 +15,8 @@ non_empty_isos = df.slice(1).filter(pl.col("ISO639-1") != "").rows()
 all_langs = {iso[0]: (iso[1], iso[2], iso[3]) for iso in non_empty_isos} # {'Romanian': ('ro', 'rum', 'ron')}
 
 # Language options as list, add favourite languages first
-options = list(favourite_langs.keys())
-options.extend(list(all_langs.keys()))
+langs = list(favourite_langs.keys())
+langs.extend(list(all_langs.keys()))
 
 models = ["Helsinki-NLP",
           "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul", "Helsinki-NLP/opus-mt-tc-bible-big-mul-deu_eng_nld",
@@ -35,6 +35,7 @@ models = ["Helsinki-NLP",
           "Unbabel/Tower-Plus-2B", "Unbabel/TowerInstruct-7B-v0.2", "Unbabel/TowerInstruct-Mistral-7B-v0.2",
           "openGPT-X/Teuken-7B-instruct-commercial-v0.4", "openGPT-X/Teuken-7B-instruct-v0.6"
           ]
+DEFAULTS = [langs[0], langs[1], models[0]]
 
 def model_to_cuda(model):
     # Move the model to GPU if available
@@ -472,7 +473,7 @@ def translate_text(input_text: str, s_language: str, t_language: str, model_name
 def swap_languages(src_lang, tgt_lang):
     return tgt_lang, src_lang 
 
-def get_info(model_name, sl=None, tl=None):
+def get_info(model_name: str, sl: str = None, tl: str = None):
     helsinki = '### [Helsinki-NLP](https://huggingface.co/Helsinki-NLP "Helsinki-NLP")'
     if model_name == "Helsinki-NLP" and sl is not None and tl is not None:
         response = requests.get(f'https://huggingface.co/{model_name}/opus-mt-{sl}-{tl}/raw/main/README.md').text
@@ -493,8 +494,8 @@ def create_interface():
         input_text = gr.Textbox(label="Enter text to translate:", placeholder="Type your text here, maximum 512 tokens")
         
         with gr.Row(variant="compact"):
-            s_language = gr.Dropdown(choices=options, value = options[0], label="Source language", interactive=True, scale=3)
-            t_language = gr.Dropdown(choices=options, value = options[1], label="Target language", interactive=True, scale=3)
+            s_language = gr.Dropdown(choices=langs, value = DEFAULTS[0], label="Source language", interactive=True, scale=3)
+            t_language = gr.Dropdown(choices=langs, value = DEFAULTS[1], label="Target language", interactive=True, scale=3)
             swap_btn = gr.Button("Swap Languages", size="md", scale=1)
             swap_btn.click(fn=swap_languages, inputs=[s_language, t_language], outputs=[s_language, t_language], api_name=False, show_api=False)
 
@@ -506,7 +507,7 @@ def create_interface():
         message_text = gr.Textbox(label="Messages:", placeholder="Display field for status and error messages", interactive=False,
                                   value=f'Default translation settings: from {s_language.value} to {t_language.value} with {model_name.value}.')
         allmodels = gr.HTML(label="Model links:", value=', '.join([f'<a href="https://huggingface.co/{model}">{model}</a>' for model in models]))
-        model_info = gr.Markdown(label="Model info:", value=get_info("Helsinki-NLP"), show_copy_button=True)
+        model_info = gr.Markdown(label="Model info:", value=get_info(DEFAULTS[2]), show_copy_button=True)
         model_name.change(fn=get_info, inputs=[model_name, s_language, t_language], outputs=model_info, api_name=False, show_api=False)
 
         translate_btn.click(