Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -47,19 +47,18 @@ def download_argos_model(from_code, to_code):
|
|
| 47 |
)
|
| 48 |
)
|
| 49 |
argostranslate.package.install_from_path(package_to_install.download())
|
|
|
|
| 50 |
|
| 51 |
def argos(sl, tl, input_text):
|
| 52 |
import argostranslate.translate, argostranslate.package
|
| 53 |
# Translate
|
| 54 |
try:
|
| 55 |
-
download_argos_model(sl, tl)
|
| 56 |
translated_text = argostranslate.translate.translate(input_text, sl, tl)
|
| 57 |
except StopIteration:
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
print(type(available_packages), available_packages)
|
| 62 |
-
translated_text = f"No Argos model for {sl} to {tl}. Try other model or languages combination from the available Argos models: {available_packages}."
|
| 63 |
except Exception as error:
|
| 64 |
translated_text = error
|
| 65 |
print(error)
|
|
|
|
| 47 |
)
|
| 48 |
)
|
| 49 |
argostranslate.package.install_from_path(package_to_install.download())
|
| 50 |
+
return available_packages
|
| 51 |
|
| 52 |
def argos(sl, tl, input_text):
|
| 53 |
import argostranslate.translate, argostranslate.package
|
| 54 |
# Translate
|
| 55 |
try:
|
| 56 |
+
available_packages = download_argos_model(sl, tl)
|
| 57 |
translated_text = argostranslate.translate.translate(input_text, sl, tl)
|
| 58 |
except StopIteration:
|
| 59 |
+
# packages_info = ', '.join(f"{pkg.get_description()}->{str(pkg.links)} {str(pkg.source_languages)}" for pkg in argostranslate.package.get_available_packages())
|
| 60 |
+
packages_info = ', '.join(f"{pkg.from_name} ({pkg.from_code}) -> {pkg.to_name} ({pkg.to_code})" for pkg in available_packages())
|
| 61 |
+
translated_text = f"No Argos model for {sl} to {tl}. Try other model or languages combination from the available Argos models: {packages_info}."
|
|
|
|
|
|
|
| 62 |
except Exception as error:
|
| 63 |
translated_text = error
|
| 64 |
print(error)
|