Update app.py

app.py

@@ -1,48 +1,54 @@
 import gradio as gr
 from mtranslate import translate
 import re
-import html
 
 # Diccionario de idiomas y sus códigos
 lang_dict = {
     'Automático': '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', 
+    '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'
 }
 
 lang_list = list(lang_dict.keys())
 
+# Función para separar texto y etiquetas HTML
+def split_html_text(text):
+    pattern = r'(<[^>]+>)'
+    parts = re.split(pattern, text)
+    return [part for part in parts if part]  # Eliminar partes vacías
+
+# Función para dividir texto en fragmentos
 def split_text(text, limit=4000):
     sentences = re.split(r'([;.])', text)
     chunks = []
@@ -57,111 +63,56 @@ 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 = {}
-    
-    # Crear una versión del texto con marcadores únicos en lugar de las etiquetas HTML
-    protected_text = text
-    
-    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
-
-def translate_text(source_lang, target_lang, text):
-    # Proteger etiquetas HTML
-    protected_text, replacements = protect_html_tags(text)
+# Función de traducción mejorada
+def translate_text(from_lang, to_lang, text):
+    from_code = lang_dict[from_lang]
+    to_code = lang_dict[to_lang]
     
-    # Obtener códigos de idioma
-    source_code = lang_dict[source_lang]
-    target_code = lang_dict[target_lang]
+    # Separar texto y etiquetas HTML
+    parts = split_html_text(text)
+    translated_parts = []
     
-    # Dividir el texto en fragmentos para no superar el límite
-    chunks = split_text(protected_text)
-    
-    # Traducir cada fragmento respetando el idioma de origen
-    translated_chunks = []
-    for chunk in chunks:
-        if source_lang == 'Automático':
-            # Si es automático, no especificamos el idioma de origen
-            translated_chunk = translate(chunk, target_code)
+    for part in parts:
+        if re.match(r'<[^>]+>', part):
+            # Es una etiqueta HTML, se mantiene intacta
+            translated_parts.append(part)
         else:
-            # Si se especificó un idioma de origen, lo usamos
-            translated_chunk = translate(chunk, target_code, source=source_code)
-        translated_chunks.append(translated_chunk)
-    
-    translated_text = ''.join(translated_chunks)
+            # Es texto, se divide en fragmentos si es necesario y se traduce
+            chunks = split_text(part)
+            translated_chunks = []
+            for chunk in chunks:
+                if from_code == 'auto':
+                    translated_chunk = translate(chunk, to_code)
+                else:
+                    translated_chunk = translate(chunk, to_code, from_code)
+                translated_chunks.append(translated_chunk)
+            translated_parts.append(''.join(translated_chunks))
     
-    # Restaurar etiquetas HTML
-    final_text = restore_html_tags(translated_text, replacements)
-    
-    return final_text
+    translated_text = ''.join(translated_parts)
+    return translated_text
 
-def main(texto, source_lang, 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(source_lang, 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
+# Función principal
+def main(Texto, from_lang, to_lang):
+    translated_text = translate_text(from_lang, to_lang, Texto)
     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
-        </button>
+    <div>
+        {translated_text}
     </div>
     """
-    
     return html_output
 
+# Interfaz de Gradio
 iface = gr.Interface(
     fn=main,
     inputs=[
-        "text",
-        gr.Dropdown(lang_list, value="Automático", label="Idioma de Origen"),
-        gr.Dropdown(lang_list, label="Idioma de Destino")
+        gr.Textbox(label="Texto a traducir"),
+        gr.Dropdown(lang_list, label="Idioma de origen", value="Automático"),
+        gr.Dropdown(lang_list, label="Idioma de 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, selecciona el idioma de origen (o déjalo en "Automático") 
-        y selecciona el idioma de destino. Las etiquetas HTML serán preservadas. ¡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, selecciona el idioma de origen (o deja 'Automático') y el idioma de destino. ¡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()