Update app.py

app.py

@@ -11,7 +11,24 @@ options = ["German", "Romanian", "English", "French", "Spanish", "Italian"]
 langs = {"English": "en", "Romanian": "ro", "German": "de", "French": "fr", "Spanish": "es", "Italian": "it"}
 models = ["Helsinki-NLP", "Argos", "t5-base", "t5-small", "t5-large", "Unbabel/Tower-Plus-2B", 
           "Unbabel/TowerInstruct-Mistral-7B-v0.2", "winninghealth/WiNGPT-Babel-2", "Google"]
-
+allmodels = ["Helsinki-NLP",
+          "Helsinki-NLP/opus-mt-tc-bible-big-mul-mul", "Helsinki-NLP/opus-mt-tc-bible-big-mul-deu_eng_nld",
+          "Helsinki-NLP/opus-mt-tc-bible-big-mul-deu_eng_fra_por_spa", "Helsinki-NLP/opus-mt-tc-bible-big-deu_eng_fra_por_spa-mul",
+          "Helsinki-NLP/opus-mt-tc-bible-big-roa-deu_eng_fra_por_spa", "Helsinki-NLP/opus-mt-tc-bible-big-deu_eng_fra_por_spa-roa",
+          "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",
+          "facebook/m2m100_418M", "facebook/m2m100_1.2B", "Lego-MT/Lego-MT",
+          "bigscience/mt0-small", "bigscience/mt0-base", "bigscience/mt0-large", "bigscience/mt0-xl",
+          "bigscience/bloomz-560m", "bigscience/bloomz-1b1", "bigscience/bloomz-1b7", "bigscience/bloomz-3b",
+          "t5-small", "t5-base", "t5-large",
+          "google/flan-t5-small", "google/flan-t5-base", "google/flan-t5-large", "google/flan-t5-xl",
+          "google/madlad400-3b-mt", "jbochi/madlad400-3b-mt",
+          "Argos", "Google",
+          "HuggingFaceTB/SmolLM3-3B", "winninghealth/WiNGPT-Babel-2",
+          "utter-project/EuroLLM-1.7B", "utter-project/EuroLLM-1.7B-Instruct",
+          "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"
+          ]
 class Translators:
     def __init__(self, model_name: str, sl: str, tl: str, input_text: str):
         self.model_name = model_name
@@ -383,7 +400,7 @@ def wingpt(model_name, sl, tl, input_text):
     result = rawresult.split('\n')[-1].strip() if '\n' in rawresult else rawresult.strip()
     return result
 
-def translate_text(input_text: str, s_language: str, t_language: str, model_name: str) -> tuple[str, str]:
+def translate_text(model_name: str, s_language: str, t_language: str, input_text: str) -> tuple[str, str]:
     """
     Translates the input text from the source language to the target language  using a specified model.
 
@@ -542,7 +559,8 @@ if model_name.startswith('t5'):
 
 st.write("Selected language combination:", sselected_language, " - ", tselected_language, "Selected model:", model_name)
 submit_button = st.button("Translate")
-translated_textarea = st.text("")
+translated_textarea = st.text(label="Translation", placeholder="Translation area", value='')
+message_textarea = st.text(label="Messages", placeholder="Messages area", value='')
 
 # Handle the submit button click
 if submit_button:
@@ -598,5 +616,7 @@ if submit_button:
         
     # Display the translated text
     print(translated_text)
-    st.write(f"Translated text from {sselected_language} to {tselected_language} using {model_name}:")
-    translated_textarea = st.text(translated_text)
+    message = f"Translated from {sselected_language} to {tselected_language} using {model_name}."
+    translated_textarea = st.text(translated_text)
+    message_textarea = st.text(value=message)
+    message = st.write(f"Translated text from {sselected_language} to {tselected_language} using {model_name}.")