Update app.py

app.py

@@ -1,51 +1,48 @@
 import gradio as gr
 from mtranslate import translate
 import re
-import html
+import json
 
 # Diccionario de idiomas y sus códigos
 lang_dict = {
-    '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', 
-    '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', 
+    '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',
+    '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'
 }
 
-# Añadir el idioma base para seleccionar el origen
-source_lang_dict = {'Detectar automáticamente': 'auto'}
-source_lang_dict.update(lang_dict)
-
-lang_list = list(lang_dict.keys())
-source_lang_list = list(source_lang_dict.keys())
+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)
@@ -61,106 +58,51 @@ def split_text(text, limit=4000):
         chunks.append(chunk)
     return chunks
 
-def protect_html_tags(text):
-    # Identificar y extraer etiquetas HTML
-    html_pattern = re.compile(r'<([^>]+)>')
-    html_matches = html_pattern.finditer(text)
-    
-    # Crear un diccionario de reemplazo
-    replacements = {}
+def translate_html_content(text, source_lang, target_lang):
+    elementor_pattern = re.compile(r'(<[^>]*class="[^"]*elementor[^"]*"[^>]*>.*?</[^>]*>)', re.DOTALL|re.IGNORECASE)
+    parts = elementor_pattern.split(text)
     
-    # Crear una versión del texto con marcadores únicos en lugar de las etiquetas HTML
-    protected_text = text
+    translated_parts = []
+    for i, part in enumerate(parts):
+        if i % 2 == 1:  # Elementor element
+            translated_parts.append(part)
+            continue
+            
+        chunks = split_text(part)
+        translated_chunks = [translate(chunk, target_lang, source_lang) for chunk in chunks]
+        translated_parts.append(''.join(translated_chunks))
     
-    for i, match in enumerate(html_matches):
-        placeholder = f"__HTML_TAG_{i}__"
-        replacements[placeholder] = match.group(0)
-        protected_text = protected_text.replace(match.group(0), placeholder, 1)
-    
-    return protected_text, replacements
-
-def restore_html_tags(text, replacements):
-    # Restaurar las etiquetas HTML originales
-    restored_text = text
-    for placeholder, original in replacements.items():
-        restored_text = restored_text.replace(placeholder, original)
-    return restored_text
+    return ''.join(translated_parts)
 
-def translate_text(source_lang, target_lang, text):
-    # Proteger etiquetas HTML
-    protected_text, replacements = protect_html_tags(text)
-    
-    # Obtener código del idioma de destino
+def main(Texto, source_lang, target_lang):
+    source_code = lang_dict[source_lang]
     target_code = lang_dict[target_lang]
     
-    # Dividir el texto en fragmentos para no superar el límite
-    chunks = split_text(protected_text)
-    
-    # Traducir cada fragmento
-    translated_chunks = []
-    for chunk in chunks:
-        # mtranslate no acepta el parámetro 'source', solo podemos especificar el destino
-        # La API de Google detecta automáticamente el idioma de origen
-        translated_chunk = translate(chunk, target_code)
-        translated_chunks.append(translated_chunk)
-    
-    translated_text = ''.join(translated_chunks)
+    translated_text = translate_html_content(Texto, source_code, target_code)
+    escaped_text = json.dumps(translated_text)[1:-1]
     
-    # Restaurar etiquetas HTML
-    final_text = restore_html_tags(translated_text, replacements)
-    
-    return final_text
-
-def main(texto, target_lang):
-    # Verificar si hay texto para traducir
-    if not texto.strip():
-        return "<div>Por favor, ingresa un texto para traducir.</div>"
-    
-    # Realizar la traducción
-    translated_text = translate_text('Detectar automáticamente', target_lang, texto)
-    
-    # Escapar el texto traducido para JavaScript (para el botón de copia)
-    escaped_text = html.escape(translated_text).replace("'", "\\'").replace('"', '\\"')
-    
-    # Generar HTML con botón de copia
     html_output = f"""
-    <div style="position: relative;">
-        <div>{translated_text}</div>
-        <button 
-            style="position: absolute; top: 0; right: 0; font-size: small; padding: 5px; background-color: #f0f0f0; color: #333; border: 1px solid #ccc; border-radius: 4px; cursor: pointer;" 
-            onclick='navigator.clipboard.writeText("{escaped_text}").then(() => alert("Texto copiado al portapapeles"))'>
-            Copiar
+    <div style="position: relative; min-height: 100px;">
+        <div id="translated-content">{translated_text}</div>
+        <button onclick="navigator.clipboard.writeText('{escaped_text}').then(() => alert('Texto copiado'))" 
+            style="position: absolute; top: 0; right: 0; padding: 5px; background: #ffffff; border: 1px solid #e0e0e0; border-radius: 4px; cursor: pointer;">
+            📋 Copiar
         </button>
     </div>
     """
-    
     return html_output
 
 iface = gr.Interface(
     fn=main,
     inputs=[
         "text",
-        gr.Dropdown(lang_list, label="Idioma de Destino")
+        gr.Dropdown(source_lang_list, value="Auto", label="Idioma origen"),
+        gr.Dropdown(lang_list, value="Español", label="Idioma destino")
     ],
     outputs="html",
-    title="""<div style='margin:0 auto;text-align:center'>
-        <div style='margin:0 auto;text-align:center'>
-            <img style='width:100px;display: inline-table;margin-bottom:-10px' src='https://artxeweb.com/media/files/idioma.jpg'>
-            <p>Traducción sin límites</p>
-        </div>
-    </div>""",
-    description="""<p style='margin-bottom:10px;text-align:center;background: #ffffff; padding: 8px; border-radius: 8px; 
-        border-width: 1px; border: solid 1px #e5e7eb;'>
-        Ingresa el texto que deseas traducir y selecciona el idioma de destino. Las etiquetas HTML serán preservadas.
-        El idioma de origen se detecta automáticamente. ¡No hay límites!
-    </p>""",
-    article="""<div style='margin-top:10px'>
-        <p style='text-align: center !important; background: #ffffff; padding: 5px 30px; 
-        border-radius: 8px; border-width: 1px; border: solid 1px #e5e7eb; width: fit-content; margin: auto;'>
-            Desarrollada por <a style='text-decoration: none !important; color: #e12a31 !important;' 
-            title='Artxe Web' href='https://artxeweb.com'>© Artxe Web</a>
-        </p>
-    </div>"""
+    title="<div style='margin:0 auto;text-align:center'><div style='margin:0 auto;text-align:center'><img style='width:100px;display: inline-table;margin-bottom:-10px' src='https://artxeweb.com/media/files/idioma.jpg'><p>Traducción sin límites</p></div>",
+    description="<p style='margin-bottom:10px;text-align:center;background: #ffffff; padding: 8px; border-radius: 8px; border-width: 1px; border: solid 1px #e5e7eb;'>Ingresa el texto que deseas traducir y selecciona los idiomas. ¡No hay límites!</p>",
+    article="<div style='margin-top:10px'><p style='text-align: center !important; background: #ffffff; padding: 5px 30px; border-radius: 8px; border-width: 1px; border: solid 1px #e5e7eb; width: fit-content; margin: auto;'>Desarrollada por <a style='text-decoration: none !important; color: #e12a31 !important;' title='Artxe Web' href='https://artxeweb.com'>© Artxe Web</a></p></div>"
 )
 
 iface.launch()