from transformers import pipeline
import gradio as gr
modelo_en_es="Helsinki-NLP/opus-mt-en-es"
option="Choose a language"
opcion="Escoge una opcion"
text1="Text to translate"
text2="Translation"
modelo_es_en="Helsinki-NLP/opus-mt-es-en"
texto1="Texto a traducir"
texto2="Texto traducido"
enviar="Enviar"

def traductor(prompt,modelo):
    pipe = pipeline("translation", model= modelo_en_es if modelo=="En-Es" else modelo_es_en)
    frase_traducida=pipe(prompt)
    return frase_traducida[0]['translation_text']

# def actualizar(modelo):
#     if modelo== "En-es":
#         return text1, text2
#     else:
#         return texto1, texto2

def actualizar(modelo):
    if modelo == "En-Es":
        return gr.update(label=text1), gr.update(label=text2), gr.update(label=option)
    else:
        return gr.update(label=texto1), gr.update(label=texto2), gr.update(label=opcion)


with gr.Blocks() as demo:
    with gr.Row():
        modelo=gr.Dropdown(["En-Es","Es-En"],label="Choose a language", value="En-Es")
    with gr.Row():
        with gr.Column():
            # txtbox1= gr.Textbox(label= text1 if modelo == "En-Es" else texto1)
            txtbox1= gr.Textbox(label=text1)

        with gr.Column():
            # txtbox2= gr.Textbox(label= text2 if modelo == "En-Es" else texto2)
            txtbox2= gr.Textbox(label= text2)

    submit=gr.Button("Submit")
    modelo.change(actualizar,inputs=modelo, outputs=[txtbox1, txtbox2,modelo])
    submit.click(traductor,inputs=[txtbox1,modelo],outputs=txtbox2)

demo.launch()