import gradio as gr
from transformers import pipeline

titulo = "Traductor EN-ES / ES-EN"
def modelo(text, model):
    if model == "Ingles a Español":
        model = "Helsinki-NLP/opus-mt-en-es"
    else:
        model = "Helsinki-NLP/opus-mt-es-en"

    pipe = pipeline("translation", model=model)
    response = pipe(text)
    return response[0]['translation_text']

descripcion = """<div style="display: flex; justify-content: space-between; align-items: center;">
    <div style="width: 100%; padding-right: 10px;">
        El objetivo de esta pagina es traducir un texto tanto de Ingles a Español como de Español a Ingles
    </div>
</div>
"""


# Creamos el interface pasandole la funcion los inputs y los outputs ademas del titulo, el tema y el article
demo = gr.Interface(
    fn=modelo,
    inputs=[gr.Text(), gr.Dropdown(["Español a Ingles", "Ingles a Español"], label="Selecciona a que idioma quieres traducir")],
    outputs='text',
    title=titulo,
    theme="gstaff/xkcd",
    description=descripcion
)

# Con autentificacion

# demo.launch(auth=("iabd", "ia)

# Sin autentificacion

demo.launch()