#!/usr/bin/env python3
import argparse
import argostranslate.package

parser = argparse.ArgumentParser()
parser.add_argument("--source", default="zh")
parser.add_argument("--target", default="en")
args = parser.parse_args()

argostranslate.package.update_package_index()
package = next(
    item for item in argostranslate.package.get_available_packages()
    if item.from_code == args.source and item.to_code == args.target
)
argostranslate.package.install_from_path(package.download())
print(f"Installed Argos model {args.source} to {args.target}")