Update app.py

app.py

@@ -1,54 +1,52 @@
 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())
+# Añadir el idioma base para seleccionar el origen
+source_lang_dict = {'Detectar automáticamente': 'auto'}
+source_lang_dict.update(lang_dict)
 
-# 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
+lang_list = list(lang_dict.keys())
+source_lang_list = list(source_lang_dict.keys())
 
-# Función para dividir texto en fragmentos
 def split_text(text, limit=4000):
     sentences = re.split(r'([;.])', text)
     chunks = []
@@ -63,56 +61,106 @@ def split_text(text, limit=4000):
         chunks.append(chunk)
     return chunks
 
-# 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]
+def protect_html_tags(text):
+    # Identificar y extraer etiquetas HTML
+    html_pattern = re.compile(r'<([^>]+)>')
+    html_matches = html_pattern.finditer(text)
     
-    # Separar texto y etiquetas HTML
-    parts = split_html_text(text)
-    translated_parts = []
+    # Crear un diccionario de reemplazo
+    replacements = {}
     
-    for part in parts:
-        if re.match(r'<[^>]+>', part):
-            # Es una etiqueta HTML, se mantiene intacta
-            translated_parts.append(part)
-        else:
-            # 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))
+    # Crear una versión del texto con marcadores únicos en lugar de las etiquetas HTML
+    protected_text = text
     
-    translated_text = ''.join(translated_parts)
-    return translated_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
 
-# Función principal
-def main(Texto, from_lang, to_lang):
-    translated_text = translate_text(from_lang, to_lang, Texto)
+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
+    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)
+    
+    # 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>
-        {translated_text}
+    <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>
     """
+    
     return html_output
 
-# Interfaz de Gradio
 iface = gr.Interface(
     fn=main,
     inputs=[
-        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")
+        "text",
+        gr.Dropdown(lang_list, label="Idioma de Destino")
     ],
-    outputs="text",
-    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>"
+    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>"""
 )
 
 iface.launch()