Update app.py

app.py

@@ -6,6 +6,18 @@ import httpx
 
 logging.set_verbosity_error()
 
+def download_argos_model(from_code, to_code):
+    import argostranslate.package
+    # 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())
+
 # Create the app layout
 st.header("Text Machine Translation")
 input_text = st.text_input("Enter text to translate:")
@@ -83,26 +95,14 @@ if submit_button:
             translated_text = translated_text.split(start_marker)[1].strip()
         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())
-        
+        import argostranslate.translate       
         # Translate
-        translated_text = argostranslate.translate.translate(input_text, from_code, to_code)
+        try:
+            translated_text = argostranslate.translate.translate(input_text, sl, tl)
+            # download_argos_model(sl, tl)
+        except StopIteration:
+            translated_text = f"No Argos model for {sselected_language} to {tselected_language}. Try other model or languages combination!"
         print(translated_text)
-        
     # Display the translated text
     print(translated_text)
     st.write(f"Translated text from {sselected_language} to {tselected_language} using {model_name}:")