import streamlit as st
import time
from transformers import pipeline

st.title("Traductormultilenguaje")

translation_models = {
  "English to German": "Helsinki-NLP/opus-mt-en-de",
  "German to English": "Helsinki-NLP/opus-mt-de-en",
  "English to French": "Helsinki-NLP/opus-mt-en-fr",
  "French to English": "Helsinki-NLP/opus-mt-fr-en",
  "English to Urdu": "Helsinki-NLP/opus-mt-en-ur",
  "Urdu to English": "Helsinki-NLP/opus-mt-ur-en",
  "English to Spanish": "Helsinki-NLP/opus-mt-en-es",
  "Spanish to English": "Helsinki-NLP/opus-mt-es-en",
  "English to Chinese": "Helsinki-NLP/opus-mt-en-zh",
  "Chinese to English": "Helsinki-NLP/opus-mt-zh-en",
}
idiomaseleccionado = st.selectbox("Idiomas: ",list(translation_models.keys()))
traductor = pipeline(task="translation", model = translation_models[idiomaseleccionado])

textoingresado = st.text_area("Ingrese el texto a trauducir", "")

if st.button("Traducir"):
    with st.spinner("Traduciendo..."):
        time.sleep(2)
        if textoingresado:
            textotraducido = traductor(textoingresado, max_length=500)[0]['textotraducido']
            st.succes("Texto Traducido:  {textotraducido}")
        else:
            st.warning("Ingrese un texto, no seas vivo")

if st.button("Limpiar"):
    textoingresado=""
    st.empty()