from transformers import pipeline
import gradio as gr

# Cargar el pipeline de traducción de español a inglés
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-es-en")

# Función para traducir el texto
def translate_text(text):
    # Generamos la traducción del texto ingresado
    translation = translator(text)
    return translation[0]['translation_text']

# Crear la interfaz
with gr.Blocks() as demo:
    gr.Markdown("""
    # Traducción Español a Inglés
    Esta aplicación traduce textos de español a inglés utilizando un modelo de traducción automática. 
    Solo debes ingresar un texto en español, y el modelo generará la traducción correspondiente al inglés.
    """)

    # Crear la interfaz de usuario
    gr.Interface(
        fn=translate_text,
        inputs=gr.Textbox(label="Introduce el texto en español", placeholder="Escribe un texto en español aquí..."),
        outputs=gr.Textbox(label="Traducción al inglés"),
    )

    # Pie de página
    gr.Markdown("""
    ---
    Demostración de traducción usando el modelo [Helsinki-NLP/opus-mt-es-en](https://huggingface.co/Helsinki-NLP/opus-mt-es-en).
    
    Desarrollado con ❤️ por [@srjosueaaron](https://www.instagram.com/srjosueaaron/).
    """)

if __name__ == "__main__":
    demo.launch()