Update app.py

app.py

@@ -12,7 +12,7 @@ input_text = st.text_input("Enter text to translate:")
 # Create a list of options for the select box
 options = ["German", "Romanian", "English", "French", "Spanish", "Italian"]
 langs = {"English":"en", "Romanian":"ro", "German":"de", "French":"fr", "Spanish":"es", "Italian":"it"}
-models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "Unbabel/Tower-Plus-2B", "Unbabel/TowerInstruct-Mistral-7B-v0.2", 'Google']
+models = ["Helsinki-NLP", "t5-base", "t5-small", "t5-large", "Unbabel/Tower-Plus-2B", "Unbabel/TowerInstruct-Mistral-7B-v0.2", 'Google', 'Argos']
 
 # Create two columns
 scol, tcol = st.columns(2)
@@ -81,7 +81,27 @@ if submit_button:
         start_marker = "<end_of_turn>"
         if start_marker in translated_text:
             translated_text = translated_text.split(start_marker)[1].strip()
-        translated_text = translated_text.replace('Answer:', '').strip() if translated_text.startswith('Answer:') else translated_text 
+        translated_text = translated_text.replace('Answer:', '').strip() if translated_text.startswith('Answer:') else translated_text
+    if 'Argos' in model_name:   
+        import argostranslate.package
+        import argostranslate.translate
+        
+        from_code = sl
+        to_code = tl
+        
+        # Download and install Argos Translate package
+        argostranslate.package.update_package_index()
+        available_packages = argostranslate.package.get_available_packages()
+        package_to_install = next(
+            filter(
+                lambda x: x.from_code == from_code and x.to_code == to_code, available_packages
+            )
+        )
+        argostranslate.package.install_from_path(package_to_install.download())
+        
+        # Translate
+        translated_text = argostranslate.translate.translate(input_text, from_code, to_code)
+        print(translated_text)
         
     # Display the translated text
     print(translated_text)