import gradio as gr
from argostranslate.package import install_from_path
from argostranslate.translate import get_translation_from_codes
from huggingface_hub import hf_hub_download
from json import load as json_load
from os import environ
lang = environ["LANG"]
lang_name = json_load(open("languages.json"))[lang]
install_from_path(hf_hub_download("shethjenil/argostranslate",f"translate-{lang}_en.argosmodel"))
install_from_path(hf_hub_download("shethjenil/argostranslate",f"translate-en_{lang}.argosmodel"))
translation_lang2eng = get_translation_from_codes(lang, "en")
translation_eng2lang = get_translation_from_codes("en", lang)
gr.TabbedInterface([gr.Interface(translation_lang2eng.translate, inputs=gr.Text(), outputs=gr.Text()),gr.Interface(translation_eng2lang.translate, inputs=gr.Text(), outputs=gr.Text())], [f"{lang_name} -> English",f"English -> {lang_name}"],title="ArgosTranslate").launch()