from easynmt import EasyNMT
import os
#zh_en_naive_model = EasyNMT("m2m_100_418M")
#p00 = os.path.join(model_path, "zh_en_m2m")
p00 = "zh_en_m2m"
assert os.path.exists(p00)
zh_en_naive_model = EasyNMT(p00)
zh_en_naive_model.translate(["宁波在哪?"], source_lang="zh", target_lang = "en")

'''
en_question = zh_en_naive_model.translate([zh_question], source_lang="zh", target_lang = "en")[0]
'''

import gradio as gr

example_sample = [
    "宁波在哪个省份？",
    "美国的通货是什么？",
]

def demo_func(zh_question):
    assert type(zh_question) == type("")
    en_question = zh_en_naive_model.translate([zh_question], source_lang="zh", target_lang = "en")[0]
    return {
        "English Question": en_question
    }


demo = gr.Interface(
        fn=demo_func,
        inputs=gr.Text(label = "Chinese Question"),
        outputs="json",
        title=f"Translate Chinese into English 🐱 demonstration",
        examples=example_sample if example_sample else None,
        cache_examples = False
    )

demo.launch(server_name=None, server_port=None)