Update app.py

app.py

@@ -383,6 +383,108 @@ def wingpt(model_name, sl, tl, input_text):
     result = rawresult.split('\n')[-1].strip() if '\n' in rawresult else rawresult.strip()
     return result
 
+@spaces.GPU
+def translate_text(input_text: str, s_language: str, t_language: str, model_name: str) -> tuple[str, str]:
+    """
+    Translates the input text from the source language to the target language  using a specified model.
+
+    Parameters:
+        input_text (str): The source text to be translated
+        s_language (str): The source language of the input text
+        t_language (str): The target language in which the input text is translated
+        model_name (str): The selected translation model name
+
+    Returns:
+        tuple: 
+            translated_text(str): The input text translated to the selected target language
+            message_text(str):  A descriptive message summarizing the translation process. Example: "Translated from English to German with Helsinki-NLP."
+    
+    Example:
+        >>> translate_text("Hello world", "English", "German", "Helsinki-NLP")
+        ("Hallo Welt", "Translated from English to German with Helsinki-NLP.")
+    """
+    
+    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}'
+    translated_text = None
+    try:    
+        if "-mul" in model_name.lower() or "mul-" in model_name.lower() or "-roa" in model_name.lower():
+            translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP_mulroa()
+        
+        elif model_name == "Helsinki-NLP":
+            translated_text, message_text = Translators(model_name, sl, tl, input_text).HelsinkiNLP()
+        
+        elif model_name == 'Argos':
+            translated_text = Translators(model_name, sl, tl, input_text).argos()
+    
+        elif model_name == 'Google':
+            translated_text = Translators(model_name, sl, tl, input_text).google()
+    
+        elif "m2m" in model_name.lower():
+            translated_text = Translators(model_name, sl, tl, input_text).mtom()
+
+        elif "lego" in model_name.lower():
+            translated_text = Translators(model_name, sl, tl, input_text).LegoMT()
+        
+        elif model_name.startswith('t5'):
+            translated_text = Translators(model_name, s_language, t_language, input_text).tfive()
+            
+        elif 'flan' in model_name.lower():
+            translated_text = Translators(model_name, s_language, t_language, input_text).flan()
+
+        elif 'madlad' in model_name.lower():
+            translated_text = Translators(model_name, sl, tl, input_text).madlad()
+            
+        elif 'mt0' in model_name.lower():
+            translated_text = Translators(model_name, s_language, t_language, input_text).bigscience()
+    
+        elif 'bloomz' in model_name.lower():
+            translated_text = Translators(model_name, s_language, t_language, input_text).bloomz()
+            
+        elif 'nllb' in model_name.lower():
+            nnlbsl, nnlbtl = languagecodes.nllb_language_codes[s_language], languagecodes.nllb_language_codes[t_language]
+            translated_text = Translators(model_name, nnlbsl, nnlbtl, input_text).nllb()
+        
+        elif model_name == "facebook/mbart-large-50-many-to-many-mmt":
+            translated_text = Translators(model_name, s_language, t_language, input_text).mbart_many_to_many()
+    
+        elif model_name == "facebook/mbart-large-50-one-to-many-mmt":
+            translated_text = Translators(model_name, s_language, t_language, input_text).mbart_one_to_many()
+    
+        elif model_name == "facebook/mbart-large-50-many-to-one-mmt":
+            translated_text = Translators(model_name, s_language, t_language, input_text).mbart_many_to_one()
+        
+        elif 'teuken' in model_name.lower():
+            translated_text = Translators(model_name, s_language, t_language, input_text).teuken()
+    
+        elif model_name == "utter-project/EuroLLM-1.7B-Instruct":
+            translated_text = Translators(model_name, s_language, t_language, input_text).eurollm_instruct()
+        
+        elif model_name == "utter-project/EuroLLM-1.7B":
+            translated_text = Translators(model_name, s_language, t_language, input_text).eurollm()
+                
+        elif 'Unbabel' in model_name:   
+            translated_text = Translators(model_name, s_language, t_language, input_text).unbabel()
+            
+        elif model_name == "HuggingFaceTB/SmolLM3-3B":
+            translated_text = Translators(model_name, s_language, t_language, input_text).smollm()
+
+        elif model_name == "winninghealth/WiNGPT-Babel-2":      
+            translated_text = Translators(model_name, s_language, t_language, input_text).wingpt()
+
+        elif "LLaMAX" in model_name:      
+            translated_text = Translators(model_name, s_language, t_language, input_text).LLaMAX()            
+
+        elif model_name == "Bergamot":
+            translated_text, message_text = Translators(model_name, s_language, t_language, input_text).bergamot()
+            
+    except Exception as error:
+        translated_text = error
+    finally:
+        print(input_text, translated_text, message_text)
+        return translated_text, message_text
+
 # App layout
 st.header("Text Machine Translation")
 input_text = st.text_input("Enter text to translate:")