Update app.py

app.py

@@ -2,23 +2,36 @@ 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")
+st.title("Traducción Instantánea: Inglés - Español y Español - Inglés")
 
-# Cargar el modelo de traducción
-Nombre_Modelo = 'Helsinki-NLP/opus-mt-en-es'
-TraductorEN_ES = pipeline('translation', model=Nombre_Modelo)
+# Cargar los modelos de traducción
+modelo_en_es = 'Helsinki-NLP/opus-mt-en-es'  # Inglés a Español
+modelo_es_en = 'Helsinki-NLP/opus-mt-es-en'  # Español a Inglés
+
+traductor_en_es = pipeline('translation', model=modelo_en_es)
+traductor_es_en = pipeline('translation', model=modelo_es_en)
+
+# Selección del modo de traducción
+modo_traduccion = st.selectbox("Selecciona el modo de traducción:", 
+                                 ["Inglés a Español", "Español a Inglés"])
 
 # Entrada de texto del usuario
-texto_a_traducir = st.text_area("Introduce el texto en inglés que deseas traducir:", "")
+texto_a_traducir = st.text_area("Introduce el texto 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'])
+        if modo_traduccion == "Inglés a Español":
+            # Realizar la traducción de Inglés a Español
+            resultado = traductor_en_es([texto_a_traducir])
+            st.write("Traducción al español:")
+            st.write(resultado[0]['translation_text'])
+        else:
+            # Realizar la traducción de Español a Inglés
+            resultado = traductor_es_en([texto_a_traducir])
+            st.write("Traducción al inglés:")
+            st.write(resultado[0]['translation_text'])
     else:
         st.warning("Por favor, introduce un texto para traducir.")
 
 
+