TiberiuCristianLeon commited on
Commit
a48172f
·
verified ·
1 Parent(s): f3a46cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
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
- available_packages = ', '.join(f"{pkg.type}->{str(pkg.links)} {str(pkg.languages)}" for pkg in argostranslate.package.get_available_packages())
59
- print(available_packages)
60
- available_packages = ', '.join(f"{pkg.from_name} ({pkg.from_code}) -> {pkg.to_name} ({pkg.to_code})" for pkg in argostranslate.package.get_available_packages())
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)