import gradio as gr
from transformers import pipeline

pipe = pipeline(task="translation", model="facebook/nllb-200-distilled-600M")


def greet(text,src_lang,tgt_lang):
    # 输入文本
    result=pipe(text,src_lang=src_lang ,tgt_lang=tgt_lang)
    return result[0]["translation_text"]

text = gr.Textbox(lines=5,label="输入文本",placeholder="请输入文本")
src_lang = gr.Dropdown(label="源语言",choices=["afr_Latn",
"amh_Ethi",
"arb_Arab",
"asm_Beng",
"ast_Latn",
"azj_Latn",
"bel_Cyrl",
"ben_Beng",
"bos_Latn",
"bul_Cyrl",
"cat_Latn",
"ceb_Latn",
"ces_Latn",
"ckb_Arab",
"cym_Latn",
"dan_Latn",
"deu_Latn",
"ell_Grek",
"eng_Latn",
"est_Latn",
"fin_Latn",
"fra_Latn",
"fuv_Latn",
"gle_Latn",
"glg_Latn",
"guj_Gujr",
"hau_Latn",
"heb_Hebr",
"hin_Deva",
"hrv_Latn",
"hun_Latn",
"hye_Armn",
"ibo_Latn",
"ind_Latn",
"isl_Latn",
"ita_Latn",
"jav_Latn",
"jpn_Jpan",
"kam_Latn",
"kan_Knda",
"kat_Geor",
"kaz_Cyrl",
"khm_Khmr",
"kir_Cyrl",
"kor_Hang",
"lao_Laoo",
"lij_Latn",
"lim_Latn",
"lin_Latn",
"lit_Latn",
"ltz_Latn",
"lug_Latn",
"luo_Latn",
"lvs_Latn",
"mal_Mlym",
"mar_Deva",
"mkd_Cyrl",
"mlt_Latn",
"khk_Cyrl",
"mri_Latn",
"mya_Mymr",
"nld_Latn",
"nob_Latn",
"npi_Deva",
"nso_Latn",
"nya_Latn",
"oci_Latn",
"gaz_Latn",
"ory_Orya",
"pan_Guru",
"pes_Arab",
"pol_Latn",
"por_Latn",
"pbt_Arab",
"ron_Latn",
"rus_Cyrl",
"slk_Latn",
"sna_Latn",
"snd_Arab",
"som_Latn",
"spa_Latn",
"srp_Cyrl",
"swe_Latn",
"swh_Latn",
"tam_Taml",
"tel_Telu",
"tgk_Cyrl",
"tgl_Latn",
"tha_Thai",
"tur_Latn",
"ukr_Cyrl",
"umb_Latn",
"urd_Arab",
"uzn_Latn",
"vie_Latn",
"wol_Latn",
"xho_Latn",
"yor_Latn",
"zho_Hans",
"zho_Hant",
"zsm_Latn",
"zul_Latn"])
tgt_lang = gr.Dropdown(label="目标语言",choices=["afr_Latn",
"amh_Ethi",
"arb_Arab",
"asm_Beng",
"ast_Latn",
"azj_Latn",
"bel_Cyrl",
"ben_Beng",
"bos_Latn",
"bul_Cyrl",
"cat_Latn",
"ceb_Latn",
"ces_Latn",
"ckb_Arab",
"cym_Latn",
"dan_Latn",
"deu_Latn",
"ell_Grek",
"eng_Latn",
"est_Latn",
"fin_Latn",
"fra_Latn",
"fuv_Latn",
"gle_Latn",
"glg_Latn",
"guj_Gujr",
"hau_Latn",
"heb_Hebr",
"hin_Deva",
"hrv_Latn",
"hun_Latn",
"hye_Armn",
"ibo_Latn",
"ind_Latn",
"isl_Latn",
"ita_Latn",
"jav_Latn",
"jpn_Jpan",
"kam_Latn",
"kan_Knda",
"kat_Geor",
"kaz_Cyrl",
"khm_Khmr",
"kir_Cyrl",
"kor_Hang",
"lao_Laoo",
"lij_Latn",
"lim_Latn",
"lin_Latn",
"lit_Latn",
"ltz_Latn",
"lug_Latn",
"luo_Latn",
"lvs_Latn",
"mal_Mlym",
"mar_Deva",
"mkd_Cyrl",
"mlt_Latn",
"khk_Cyrl",
"mri_Latn",
"mya_Mymr",
"nld_Latn",
"nob_Latn",
"npi_Deva",
"nso_Latn",
"nya_Latn",
"oci_Latn",
"gaz_Latn",
"ory_Orya",
"pan_Guru",
"pes_Arab",
"pol_Latn",
"por_Latn",
"pbt_Arab",
"ron_Latn",
"rus_Cyrl",
"slk_Latn",
"sna_Latn",
"snd_Arab",
"som_Latn",
"spa_Latn",
"srp_Cyrl",
"swe_Latn",
"swh_Latn",
"tam_Taml",
"tel_Telu",
"tgk_Cyrl",
"tgl_Latn",
"tha_Thai",
"tur_Latn",
"ukr_Cyrl",
"umb_Latn",
"urd_Arab",
"uzn_Latn",
"vie_Latn",
"wol_Latn",
"xho_Latn",
"yor_Latn",
"zho_Hans",
"zho_Hant",
"zsm_Latn",
"zul_Latn"])
output_text = gr.Textbox(lines=5,label="翻译文本",placeholder="翻译文本")
demo=gr.Interface(fn=greet, inputs=[text,src_lang,tgt_lang], outputs=output_text)
demo.launch(share=True)