Update app.py

app.py

@@ -1,75 +1,52 @@
 import gradio as gr
 from mtranslate import translate
-import re
+from bs4 import BeautifulSoup
 import json
-import uuid
 
-# Diccionario de idiomas (sin cambios)
+# Diccionario de idiomas
 lang_dict = {
     'Auto': 'auto',
     'Español': 'es',
     'English': 'en',
-    # ... resto del diccionario ...
+    # Agrega más idiomas según necesites
 }
 
 lang_list = [k for k in lang_dict.keys() if k != 'Auto']
 source_lang_list = ['Auto'] + lang_list
 
-def split_text(text, limit=4000):
-    sentences = re.split(r'([;.])', text)
-    chunks = []
-    chunk = ''
-    for i in range(0, len(sentences), 2):
-        sentence = sentences[i] + (sentences[i+1] if i+1 < len(sentences) else '')
-        if len(chunk) + len(sentence) > limit:
-            chunks.append(chunk)
-            chunk = ''
-        chunk += sentence
-    if chunk:
-        chunks.append(chunk)
-    return chunks
-
 def translate_html_content(text, source_lang, target_lang):
-    # Patrón para identificar etiquetas HTML completas
-    html_tag_pattern = re.compile(r'<[^>]+>')
-    
-    # Encontrar todas las etiquetas HTML
-    tags = html_tag_pattern.findall(text)
-    
-    # Reemplazar cada etiqueta con un marcador único
-    markers = {f"{{{uuid.uuid4()}}}": tag for tag in tags}
-    for marker, tag in markers.items():
-        text = text.replace(tag, marker)
-    
-    # Dividir el texto en chunks respetando los marcadores
-    chunks = split_text(text)
-    
-    # Traducir cada chunk
-    translated_chunks = []
-    for chunk in chunks:
-        # Traducir el chunk
-        translated_chunk = translate(chunk, target_lang, source_lang)
-        # Limpiar caracteres escapados no deseados
-        translated_chunk = translated_chunk.replace('\\"', '"').replace('\\n', '\n').replace('\\\\', '\\')
-        translated_chunks.append(translated_chunk)
-    
-    # Unir los chunks traducidos
-    translated_text = ''.join(translated_chunks)
+    """
+    Traduce el contenido de texto dentro de un HTML, preservando las etiquetas.
+    """
+    # Parsear el HTML con BeautifulSoup
+    soup = BeautifulSoup(text, 'html.parser')
     
-    # Restaurar las etiquetas originales
-    for marker, tag in markers.items():
-        translated_text = translated_text.replace(marker, tag)
+    # Recorrer todos los nodos de texto
+    for node in soup.find_all(text=True):
+        text_to_translate = node.strip()
+        if text_to_translate:
+            # Traducir el texto si no está vacío
+            translated_text = translate(text_to_translate, target_lang, source_lang)
+            # Reemplazar el nodo original con el texto traducido
+            node.replace_with(translated_text)
     
-    return translated_text
+    # Devolver el HTML como string
+    return str(soup)
 
 def main(Texto, source_lang, target_lang):
+    """
+    Función principal que procesa la entrada y genera la salida con un botón de copiar.
+    """
     source_code = lang_dict[source_lang]
     target_code = lang_dict[target_lang]
     
+    # Traducir el contenido HTML
     translated_text = translate_html_content(Texto, source_code, target_code)
-    # Escapar solo para el botón de copiar, no para la salida visual
+    
+    # Escapar el texto solo para el botón de copiar
     escaped_text = json.dumps(translated_text)[1:-1]
     
+    # Generar la salida HTML con el botón de copiar
     html_output = f"""
     <div style="position: relative; min-height: 100px;">
         <div id="translated-content">{translated_text}</div>
@@ -81,15 +58,15 @@ def main(Texto, source_lang, target_lang):
     """
     return html_output
 
-# Configuración de la interfaz (sin cambios)
+# Configuración de la interfaz de Gradio
 iface = gr.Interface(
     fn=main,
     inputs=[
-        "text",
+        gr.Textbox(label="Texto a traducir", lines=10),
         gr.Dropdown(source_lang_list, value="Auto", label="Idioma origen"),
         gr.Dropdown(lang_list, value="Español", label="Idioma destino")
     ],
-    outputs="text",
+    outputs=gr.HTML(label="Texto traducido"),
     title="Traducción sin límites",
     description="Ingresa el texto que deseas traducir y selecciona los idiomas. ¡No hay límites!",
     article="Desarrollada por © Artxe Web"