import streamlit as st
from transformers import pipeline

# Título de la aplicación
st.title("Traducción Instantánea: De Inglés a Español")

# Cargar el modelo de traducción
Nombre_Modelo = 'Helsinki-NLP/opus-mt-en-es'
TraductorEN_ES = pipeline('translation', model=Nombre_Modelo, device='cuda')

# Entrada de texto del usuario
texto_a_traducir = st.text_area("Introduce el texto en inglés que deseas traducir:", "")

if st.button("Traducir"):
    if texto_a_traducir:
        # Realizar la traducción
        resultado = TraductorEN_ES([texto_a_traducir])
        # Mostrar el resultado
        st.write("Traducción al español:")
        st.write(resultado[0]['translation_text'])
    else:
        st.warning("Por favor, introduce un texto para traducir.")