Update app.py

app.py

@@ -1,47 +1,28 @@
 import streamlit as st
 from mtranslate import translate
+from bs4 import BeautifulSoup, NavigableString
 import re
+import time
 import html
 
 # Diccionario de idiomas y sus códigos
 lang_dict = {
+    'Auto': 'auto',
     'Español': 'es',
     'English': 'en',
-    'Mandarín': 'zh',
-    'Hindi': 'hi',
-    'Árabe': 'ar',
-    'Portugués': 'pt',
-    'Bengalí': 'bn',
-    'Ruso': 'ru',
-    'Japonés': 'ja',
-    'Panyabí': 'pa',
-    'Alemán': 'de',
-    'Javanés': 'jw',
-    'Coreano': 'ko',
     'Francés': 'fr',
-    'Vietnamita': 'vi',
-    'Turco': 'tr',
+    'Alemán': 'de',
     'Italiano': 'it',
-    'Ucraniano': 'uk',
-    'Tailandés': 'th',
-    'Guyaratí': 'gu',
-    'Polaco': 'pl',
-    'Griego': 'el',
-    'Neerlandés': 'nl',
-    'Sueco': 'sv',
-    'Rumano': 'ro',
-    'Checo': 'cs',
-    'Húngaro': 'hu',
-    'Hebreo': 'he',
-    'Indonesio': 'id',
-    'Nepalí': 'ne',
-    'Gallego': 'gl',
-    'Catalán': 'ca',
-    'Vasco': 'eu'
+    'Portugués': 'pt',
+    'Chino': 'zh',
+    'Japonés': 'ja',
+    'Ruso': 'ru',
 }
 
 lang_list = list(lang_dict.keys())
+source_lang_list = ['Auto'] + lang_list
 
+# --- FUNCIONES ---
 def split_text(text, limit=4000):
     sentences = re.split(r'([;.])', text)
     chunks = []
@@ -56,50 +37,73 @@ def split_text(text, limit=4000):
         chunks.append(chunk)
     return chunks
 
-def translate_text(lang, text):
-    # Obtener el código del idioma
-    lang_code = lang_dict[lang]
-    
-    # Dividir el texto en fragmentos que no superen el límite de 4000 caracteres
-    chunks = split_text(text)
-    
-    translated_chunks = [translate(chunk, lang_code) for chunk in chunks]
-    translated_text = ''.join(translated_chunks)
-    
-    return translated_text
+def translate_html_content(text, source_lang, target_lang):
+    soup = BeautifulSoup(text, 'html.parser')
+
+    for node in soup.find_all(text=True):
+        if isinstance(node, NavigableString) and node.strip():
+            original_text = node.strip()
+            chunks = split_text(original_text)
+            translated_chunks = [translate(chunk, target_lang, source_lang) for chunk in chunks]
+            translated_text = ''.join(translated_chunks)
+
+            # Asegurar espacios alrededor del texto
+            if node.previous_sibling and not node.previous_sibling.endswith(' '):
+                translated_text = ' ' + translated_text
+            if node.next_sibling and not node.next_sibling.startswith(' '):
+                translated_text += ' '
+
+            node.replace_with(translated_text)
+
+    return str(soup)
 
 def main():
-    st.title("Traducción sin límites")
-    
-    text = st.text_area("Texto a traducir", height=200)
-    lang = st.selectbox("Idioma", lang_list)
-    
-    if st.button("Traducir"):
-        # Realizar la traducción
-        translated_text = translate_text(lang, text)
-        
-        # Mostrar el resultado con el botón de copia
-        st.session_state.translated_text = translated_text
-        st.write("Texto traducido:")
-        st.write(translated_text)
-        
-        # Mostrar el botón de copiar
-        copy_button = f"""
-        <button onclick="copyToClipboard()">Copiar</button>
-        <script>
-        function copyToClipboard() {{
-            var text = `{html.escape(translated_text)}`;
-            navigator.clipboard.writeText(text).then(function() {{
-                alert('Texto copiado al portapapeles');
-            }}, function(err) {{
-                console.error('No se pudo copiar el texto: ', err);
-            }});
-        }}
-        </script>
-        """
-        st.markdown(copy_button, unsafe_allow_html=True)
+    st.title("🌍 Traductor HTML Inteligente")
+
+    with st.expander("ℹ️ Instrucciones"):
+        st.write(
+            "- Introduce tu texto o HTML en la caja de texto.\n"
+            "- Selecciona el idioma de origen y el idioma de destino.\n"
+            "- Haz clic en 'Traducir' para obtener el texto traducido."
+        )
+
+    text = st.text_area("✍️ Introduce el texto o HTML aquí:", height=200)
+    col1, col2 = st.columns(2)
+    with col1:
+        source_lang = st.selectbox("🌐 Idioma de origen:", source_lang_list)
+    with col2:
+        target_lang = st.selectbox("🎯 Idioma de destino:", lang_list, index=1)
+
+    if st.button("🚀 Traducir"):
+        if text.strip():
+            with st.spinner("⏳ Traduciendo..."):
+                translated_text = translate_html_content(text, lang_dict[source_lang], lang_dict[target_lang])
+                st.session_state.translated_text = translated_text
+                time.sleep(2)  # Simulación de carga
+
+            st.subheader("📝 Resultado de la traducción:")
+            st.write(translated_text)
+
+            # Botón de copiar
+            copy_button = f"""
+            <button onclick="copyToClipboard()">📋 Copiar</button>
+            <script>
+            function copyToClipboard() {{
+                var text = `{html.escape(translated_text)}`;
+                navigator.clipboard.writeText(text).then(function() {{
+                    alert('Texto copiado al portapapeles');
+                }}, function(err) {{
+                    console.error('No se pudo copiar el texto: ', err);
+                }});
+            }}
+            </script>
+            """
+            st.markdown(copy_button, unsafe_allow_html=True)
+        else:
+            st.warning("⚠️ Por favor, introduce un texto antes de traducir.")
 
 if "translated_text" not in st.session_state:
     st.session_state.translated_text = ""
 
-main()
+# Ejecutar la app
+main()