Update app.py

app.py

@@ -25,6 +25,66 @@ def extraer_texto_pdf(pdf_file):
     except Exception as e:
         return f"Error: {str(e)}"
 
+# ============= GENERAR AUDIO CON EMOCIÓN MEJORADO =============
+def generar_audio_respuesta(texto, client):
+    """Convierte la respuesta de texto a audio usando modelos TTS con emoción"""
+    
+    # Lista de modelos TTS con mejor calidad emocional
+    modelos_tts = [
+        "facebook/mms-tts-spa",  # Mejor para español
+        "espnet/kan-bayashi_ljspeech_vits",  # VITS con emoción
+        "suno/bark-small",  # Bark tiene mejor expresividad
+        "microsoft/speecht5_tts",
+    ]
+    
+    # Limitar y optimizar texto para TTS
+    texto_limpio = texto.replace("*", "").replace("#", "").replace("`", "")
+    texto_corto = texto_limpio[:500] if len(texto_limpio) > 500 else texto_limpio
+    
+    # Agregar puntuación para mejor prosodia
+    if not texto_corto.endswith(('.', '!', '?')):
+        texto_corto += "."
+    
+    print(f"🎤 Generando audio: {texto_corto[:100]}...")
+    
+    for modelo in modelos_tts:
+        try:
+            print(f"🔊 Intentando con: {modelo}")
+            
+            # Generar audio con el modelo
+            audio_bytes = client.text_to_speech(
+                text=texto_corto,
+                model=modelo
+            )
+            
+            # Guardar audio
+            timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
+            audio_path = f"respuesta_audio_{timestamp}.wav"
+            
+            with open(audio_path, "wb") as f:
+                if isinstance(audio_bytes, bytes):
+                    f.write(audio_bytes)
+                else:
+                    # Si es un generador, escribir el contenido
+                    for chunk in audio_bytes:
+                        if isinstance(chunk, bytes):
+                            f.write(chunk)
+            
+            # Verificar que el archivo se creó correctamente
+            if os.path.exists(audio_path) and os.path.getsize(audio_path) > 0:
+                print(f"✅ Audio generado exitosamente con {modelo}: {audio_path}")
+                return audio_path
+            else:
+                print(f"⚠️ Archivo vacío o no creado con {modelo}")
+                continue
+            
+        except Exception as e:
+            print(f"❌ Error con {modelo}: {str(e)}")
+            continue
+    
+    print("⚠️ No se pudo generar audio con ningún modelo")
+    return None
+
 # ============= ASISTENTE IA CONVERSACIONAL =============
 def asistente_ia_factura(texto, pregunta_usuario):
     """Asistente IA que explica conceptos, responde preguntas y da consejos sobre facturas"""
@@ -88,52 +148,6 @@ Responde ahora:"""
     
     return "❌ No se pudo obtener respuesta del asistente IA", None
 
-# ============= GENERAR AUDIO DE LA RESPUESTA CON MODELO MEJORADO =============
-def generar_audio_respuesta(texto, client):
-    """Convierte la respuesta de texto a audio usando TTS avanzado"""
-    
-    modelos_tts = [
-        "facebook/mms-tts-spa",
-        "microsoft/speecht5_tts",
-        "suno/bark",
-        "facebook/fastspeech2-en-200_speaker-cv4",
-    ]
-    
-    # Limitar texto para TTS
-    texto_corto = texto[:400] if len(texto) > 400 else texto
-    
-    for modelo in modelos_tts:
-        try:
-            print(f"🔊 Generando audio con: {modelo}")
-            
-            # Intentar generar audio
-            audio_bytes = client.text_to_speech(
-                text=texto_corto,
-                model=modelo
-            )
-            
-            # Guardar audio
-            timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
-            audio_path = f"respuesta_audio_{timestamp}.wav"
-            
-            with open(audio_path, "wb") as f:
-                if isinstance(audio_bytes, bytes):
-                    f.write(audio_bytes)
-                else:
-                    # Si es un generador, escribir el contenido
-                    for chunk in audio_bytes:
-                        f.write(chunk)
-            
-            print(f"✅ Audio generado exitosamente: {audio_path}")
-            return audio_path
-            
-        except Exception as e:
-            print(f"❌ Error con {modelo}: {str(e)}")
-            continue
-    
-    print("⚠️ No se pudo generar audio con ningún modelo")
-    return None
-
 # ============= ANÁLISIS DE SENTIMIENTO DE FACTURA =============
 def analizar_sentimiento_factura(texto, client):
     """Analiza si la factura tiene alertas, urgencias o problemas"""
@@ -233,9 +247,9 @@ Responde solo con el nombre de la categoría:"""
     except:
         return "📁 **Categoría:** No clasificada"
 
-# ============= TRADUCTOR MULTIIDIOMA =============
-def traducir_factura(texto, idioma_destino, client):
-    """Traduce el contenido de la factura a otro idioma"""
+# ============= TRADUCTOR MULTIIDIOMA CON CSV TABULAR =============
+def traducir_factura_con_csv(datos_json, texto, idioma_destino, client):
+    """Traduce la factura y genera tanto texto como CSV tabular"""
     
     idiomas = {
         "Inglés": "English",
@@ -247,23 +261,280 @@ def traducir_factura(texto, idioma_destino, client):
     
     idioma = idiomas.get(idioma_destino, "English")
     
-    prompt = f"""Traduce este resumen de factura al {idioma}. Mantén el formato y estructura:
+    # 1. Traducir el texto completo
+    prompt_texto = f"""Traduce este resumen de factura al {idioma}. Mantén el formato y estructura:
 
 {texto[:2000]}
 
 Traducción:"""
     
     try:
-        response = client.chat.completions.create(
+        response_texto = client.chat.completions.create(
             model="Qwen/Qwen2.5-72B-Instruct",
-            messages=[{"role": "user", "content": prompt}],
+            messages=[{"role": "user", "content": prompt_texto}],
             max_tokens=1000,
             temperature=0.3
         )
-        
-        return response.choices[0].message.content
+        texto_traducido = response_texto.choices[0].message.content
     except:
-        return "❌ Error en la traducción"
+        texto_traducido = "❌ Error en la traducción del texto"
+    
+    # 2. Crear DataFrame traducido
+    if not datos_json:
+        return texto_traducido, None, None
+    
+    # Traducir etiquetas según el idioma
+    traducciones = {
+        "Inglés": {
+            "seccion": "Section",
+            "campo": "Field",
+            "valor": "Value",
+            "tipo": "Type",
+            "info_general": "GENERAL INFORMATION",
+            "numero_factura": "Invoice Number",
+            "fecha": "Date",
+            "identificador": "Identifier",
+            "emisor": "ISSUER",
+            "nombre": "Name",
+            "nif": "Tax ID",
+            "direccion": "Address",
+            "cliente": "CLIENT",
+            "productos": "PRODUCTS",
+            "producto": "Product",
+            "cantidad": "Quantity",
+            "precio_unitario": "Unit Price",
+            "total_producto": "Total",
+            "descripcion": "Description",
+            "numerico": "Numeric",
+            "monetario": "Monetary",
+            "totales": "TOTALS",
+            "base_imponible": "Taxable Base",
+            "iva": "VAT",
+            "total": "TOTAL",
+            "informacion": "Information"
+        },
+        "Francés": {
+            "seccion": "Section",
+            "campo": "Champ",
+            "valor": "Valeur",
+            "tipo": "Type",
+            "info_general": "INFORMATIONS GÉNÉRALES",
+            "numero_factura": "Numéro de Facture",
+            "fecha": "Date",
+            "identificador": "Identifiant",
+            "emisor": "ÉMETTEUR",
+            "nombre": "Nom",
+            "nif": "NIF",
+            "direccion": "Adresse",
+            "cliente": "CLIENT",
+            "productos": "PRODUITS",
+            "producto": "Produit",
+            "cantidad": "Quantité",
+            "precio_unitario": "Prix Unitaire",
+            "total_producto": "Total",
+            "descripcion": "Description",
+            "numerico": "Numérique",
+            "monetario": "Monétaire",
+            "totales": "TOTAUX",
+            "base_imponible": "Base Imposable",
+            "iva": "TVA",
+            "total": "TOTAL",
+            "informacion": "Information"
+        },
+        "Alemán": {
+            "seccion": "Abschnitt",
+            "campo": "Feld",
+            "valor": "Wert",
+            "tipo": "Typ",
+            "info_general": "ALLGEMEINE INFORMATIONEN",
+            "numero_factura": "Rechnungsnummer",
+            "fecha": "Datum",
+            "identificador": "Kennung",
+            "emisor": "AUSSTELLER",
+            "nombre": "Name",
+            "nif": "Steuernummer",
+            "direccion": "Adresse",
+            "cliente": "KUNDE",
+            "productos": "PRODUKTE",
+            "producto": "Produkt",
+            "cantidad": "Menge",
+            "precio_unitario": "Stückpreis",
+            "total_producto": "Gesamt",
+            "descripcion": "Beschreibung",
+            "numerico": "Numerisch",
+            "monetario": "Monetär",
+            "totales": "SUMMEN",
+            "base_imponible": "Steuerbemessungsgrundlage",
+            "iva": "MwSt",
+            "total": "GESAMT",
+            "informacion": "Information"
+        },
+        "Italiano": {
+            "seccion": "Sezione",
+            "campo": "Campo",
+            "valor": "Valore",
+            "tipo": "Tipo",
+            "info_general": "INFORMAZIONI GENERALI",
+            "numero_factura": "Numero Fattura",
+            "fecha": "Data",
+            "identificador": "Identificatore",
+            "emisor": "EMITTENTE",
+            "nombre": "Nome",
+            "nif": "Partita IVA",
+            "direccion": "Indirizzo",
+            "cliente": "CLIENTE",
+            "productos": "PRODOTTI",
+            "producto": "Prodotto",
+            "cantidad": "Quantità",
+            "precio_unitario": "Prezzo Unitario",
+            "total_producto": "Totale",
+            "descripcion": "Descrizione",
+            "numerico": "Numerico",
+            "monetario": "Monetario",
+            "totales": "TOTALI",
+            "base_imponible": "Imponibile",
+            "iva": "IVA",
+            "total": "TOTALE",
+            "informacion": "Informazione"
+        },
+        "Portugués": {
+            "seccion": "Seção",
+            "campo": "Campo",
+            "valor": "Valor",
+            "tipo": "Tipo",
+            "info_general": "INFORMAÇÃO GERAL",
+            "numero_factura": "Número da Fatura",
+            "fecha": "Data",
+            "identificador": "Identificador",
+            "emisor": "EMISSOR",
+            "nombre": "Nome",
+            "nif": "NIF",
+            "direccion": "Endereço",
+            "cliente": "CLIENTE",
+            "productos": "PRODUTOS",
+            "producto": "Produto",
+            "cantidad": "Quantidade",
+            "precio_unitario": "Preço Unitário",
+            "total_producto": "Total",
+            "descripcion": "Descrição",
+            "numerico": "Numérico",
+            "monetario": "Monetário",
+            "totales": "TOTAIS",
+            "base_imponible": "Base Tributável",
+            "iva": "IVA",
+            "total": "TOTAL",
+            "informacion": "Informação"
+        }
+    }
+    
+    t = traducciones.get(idioma_destino, traducciones["Inglés"])
+    
+    filas = []
+    
+    # Información general
+    filas.append({
+        t["seccion"]: t["info_general"],
+        t["campo"]: t["numero_factura"],
+        t["valor"]: datos_json.get('numero_factura', 'N/A'),
+        t["tipo"]: t["identificador"]
+    })
+    filas.append({
+        t["seccion"]: t["info_general"],
+        t["campo"]: t["fecha"],
+        t["valor"]: datos_json.get('fecha', 'N/A'),
+        t["tipo"]: t["fecha"]
+    })
+    
+    # Emisor
+    if 'emisor' in datos_json:
+        emisor = datos_json['emisor']
+        if isinstance(emisor, dict):
+            for key, value in emisor.items():
+                campo_traducido = t.get(key, key.replace('_', ' ').title())
+                filas.append({
+                    t["seccion"]: t["emisor"],
+                    t["campo"]: campo_traducido,
+                    t["valor"]: str(value),
+                    t["tipo"]: t["informacion"]
+                })
+    
+    # Cliente
+    if 'cliente' in datos_json:
+        cliente = datos_json['cliente']
+        if isinstance(cliente, dict):
+            for key, value in cliente.items():
+                campo_traducido = t.get(key, key.replace('_', ' ').title())
+                filas.append({
+                    t["seccion"]: t["cliente"],
+                    t["campo"]: campo_traducido,
+                    t["valor"]: str(value),
+                    t["tipo"]: t["informacion"]
+                })
+    
+    # Productos
+    productos = datos_json.get('productos', datos_json.get('conceptos', datos_json.get('items', [])))
+    if productos and len(productos) > 0:
+        for i, prod in enumerate(productos, 1):
+            filas.append({
+                t["seccion"]: t["productos"],
+                t["campo"]: f'{t["producto"]} {i}',
+                t["valor"]: prod.get('descripcion', 'N/A'),
+                t["tipo"]: t["descripcion"]
+            })
+            filas.append({
+                t["seccion"]: t["productos"],
+                t["campo"]: f'{t["cantidad"]} P{i}',
+                t["valor"]: str(prod.get('cantidad', '')),
+                t["tipo"]: t["numerico"]
+            })
+            filas.append({
+                t["seccion"]: t["productos"],
+                t["campo"]: f'{t["precio_unitario"]} P{i}',
+                t["valor"]: f"{prod.get('precio_unitario', 0)}",
+                t["tipo"]: t["monetario"]
+            })
+            filas.append({
+                t["seccion"]: t["productos"],
+                t["campo"]: f'{t["total_producto"]} P{i}',
+                t["valor"]: f"{prod.get('total', 0)}",
+                t["tipo"]: t["monetario"]
+            })
+    
+    # Totales
+    totales = datos_json.get('totales', {})
+    if totales or 'base_imponible' in datos_json or 'total' in datos_json:
+        base = totales.get('base_imponible', datos_json.get('base_imponible', 0))
+        iva = totales.get('iva', datos_json.get('iva', 0))
+        porcentaje_iva = totales.get('porcentaje_iva', datos_json.get('porcentaje_iva', 0))
+        total = totales.get('total', datos_json.get('total', 0))
+        
+        filas.append({
+            t["seccion"]: t["totales"],
+            t["campo"]: t["base_imponible"],
+            t["valor"]: f"{base}",
+            t["tipo"]: t["monetario"]
+        })
+        filas.append({
+            t["seccion"]: t["totales"],
+            t["campo"]: f'{t["iva"]} ({porcentaje_iva}%)',
+            t["valor"]: f"{iva}",
+            t["tipo"]: t["monetario"]
+        })
+        filas.append({
+            t["seccion"]: t["totales"],
+            t["campo"]: t["total"],
+            t["valor"]: f"{total}",
+            t["tipo"]: t["monetario"]
+        })
+    
+    df_traducido = pd.DataFrame(filas)
+    
+    # Guardar CSV
+    timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
+    csv_filename = f"factura_traducida_{idioma_destino}_{timestamp}.csv"
+    df_traducido.to_csv(csv_filename, index=False, encoding='utf-8-sig', sep=',')
+    
+    return texto_traducido, df_traducido, csv_filename
 
 # ============= DETECTOR DE FRAUDE =============
 def detectar_fraude_factura(datos_json, texto, client):
@@ -636,6 +907,7 @@ Responde en JSON:
         pass
     
     return {"es_valida": True, "errores": [], "advertencias": [], "nivel_cumplimiento": "completo"}
+
 def extraer_gastos_deducibles(datos_json, texto, client):
     """Identifica qué parte de la factura es deducible fiscalmente"""
     
@@ -895,109 +1167,6 @@ def json_a_csv(datos_json):
     
     return pd.DataFrame(filas)
 
-# ============= GENERAR CSV TABULAR DE TRADUCCIÓN =============
-def traduccion_a_csv(datos_json, traduccion_texto, idioma):
-    """Convierte la factura traducida en un CSV tabular estructurado"""
-    
-    if not datos_json:
-        return None
-    
-    timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
-    csv_filename = f"factura_traducida_{idioma}_{timestamp}.csv"
-    
-    filas = []
-    
-    # Intentar traducir cada campo
-    token = os.getenv("aa")
-    if token:
-        client = InferenceClient(token=token)
-        
-        # Traducir secciones
-        try:
-            # Información general
-            filas.append({
-                'Sección': 'INFORMACIÓN GENERAL' if idioma == 'Español' else 'GENERAL INFORMATION',
-                'Campo': 'Número de Factura' if idioma == 'Español' else 'Invoice Number',
-                'Valor': datos_json.get('numero_factura', 'N/A'),
-                'Tipo': 'Identificador' if idioma == 'Español' else 'Identifier'
-            })
-            filas.append({
-                'Sección': 'INFORMACIÓN GENERAL' if idioma == 'Español' else 'GENERAL INFORMATION',
-                'Campo': 'Fecha' if idioma == 'Español' else 'Date',
-                'Valor': datos_json.get('fecha', 'N/A'),
-                'Tipo': 'Fecha' if idioma == 'Español' else 'Date'
-            })
-            
-            # Emisor
-            if 'emisor' in datos_json:
-                emisor = datos_json['emisor']
-                if isinstance(emisor, dict):
-                    for key, value in emisor.items():
-                        filas.append({
-                            'Sección': 'EMISOR' if idioma == 'Español' else 'ISSUER',
-                            'Campo': key.replace('_', ' ').title(),
-                            'Valor': str(value),
-                            'Tipo': 'Información' if idioma == 'Español' else 'Information'
-                        })
-            
-            # Cliente
-            if 'cliente' in datos_json:
-                cliente = datos_json['cliente']
-                if isinstance(cliente, dict):
-                    for key, value in cliente.items():
-                        filas.append({
-                            'Sección': 'CLIENTE' if idioma == 'Español' else 'CLIENT',
-                            'Campo': key.replace('_', ' ').title(),
-                            'Valor': str(value),
-                            'Tipo': 'Información' if idioma == 'Español' else 'Information'
-                        })
-            
-            # Productos
-            productos = datos_json.get('productos', [])
-            if productos:
-                for i, prod in enumerate(productos, 1):
-                    filas.append({
-                        'Sección': 'PRODUCTOS' if idioma == 'Español' else 'PRODUCTS',
-                        'Campo': f'Producto {i}' if idioma == 'Español' else f'Product {i}',
-                        'Valor': prod.get('descripcion', 'N/A'),
-                        'Tipo': 'Descripción' if idioma == 'Español' else 'Description'
-                    })
-                    filas.append({
-                        'Sección': 'PRODUCTOS' if idioma == 'Español' else 'PRODUCTS',
-                        'Campo': f'Cantidad P{i}' if idioma == 'Español' else f'Quantity P{i}',
-                        'Valor': str(prod.get('cantidad', '')),
-                        'Tipo': 'Numérico' if idioma == 'Español' else 'Numeric'
-                    })
-            
-            # Totales
-            totales = datos_json.get('totales', {})
-            if totales:
-                filas.append({
-                    'Sección': 'TOTALES' if idioma == 'Español' else 'TOTALS',
-                    'Campo': 'Base Imponible' if idioma == 'Español' else 'Taxable Base',
-                    'Valor': str(totales.get('base_imponible', 0)),
-                    'Tipo': 'Monetario' if idioma == 'Español' else 'Monetary'
-                })
-                filas.append({
-                    'Sección': 'TOTALES' if idioma == 'Español' else 'TOTALS',
-                    'Campo': 'IVA' if idioma == 'Español' else 'VAT',
-                    'Valor': str(totales.get('iva', 0)),
-                    'Tipo': 'Monetario' if idioma == 'Español' else 'Monetary'
-                })
-                filas.append({
-                    'Sección': 'TOTALES' if idioma == 'Español' else 'TOTALS',
-                    'Campo': 'TOTAL',
-                    'Valor': str(totales.get('total', 0)),
-                    'Tipo': 'Monetario' if idioma == 'Español' else 'Monetary'
-                })
-        except:
-            pass
-    
-    df = pd.DataFrame(filas)
-    df.to_csv(csv_filename, index=False, encoding='utf-8-sig', sep=',')
-    
-    return csv_filename
-
 # ============= GENERAR PDF TEMPLATES =============
 def generar_pdf_clasico(csv_file, datos_json):
     timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
@@ -1391,11 +1560,11 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
                     - ✅ Recomendaciones de negociación
                     """)
         
-        # ============= TAB 4: TRADUCCIÓN MULTIIDIOMA =============
+        # ============= TAB 4: TRADUCCIÓN MULTIIDIOMA CON TABLA =============
         with gr.Tab("🌍 Traducción Internacional"):
             gr.Markdown("""
             # 🌐 Traductor Profesional de Facturas
-            ### Traduce tu factura a 5 idiomas y exporta a CSV
+            ### Traduce tu factura a 5 idiomas con vista tabular y exporta a CSV
             """)
             
             with gr.Row():
@@ -1410,23 +1579,29 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
                     
                     gr.Markdown("---")
                     gr.Markdown("### Exportar Traducción")
-                    btn_csv_traduccion = gr.Button("📥 Descargar CSV Tabular Traducido", variant="secondary")
-                    csv_traduccion_output = gr.File(label="📊 CSV Traducido (Formato Tabular)")
+                    csv_traduccion_output = gr.File(label="📊 Descargar CSV Tabular Traducido")
                 
                 with gr.Column():
-                    gr.Markdown("### 📝 Resultado de la Traducción")
+                    gr.Markdown("### 📊 Vista Tabular Traducida")
+                    tabla_traduccion = gr.DataFrame(
+                        label="Factura traducida en formato tabular",
+                        wrap=True
+                    )
+                    
+                    gr.Markdown("---")
+                    gr.Markdown("### 📝 Texto Traducido")
                     resultado_traduccion = gr.Textbox(
-                        label="Texto traducido",
-                        lines=15,
+                        label="Resumen en texto",
+                        lines=10,
                         placeholder="La traducción aparecerá aquí..."
                     )
                     
                     gr.Markdown("""
                     💡 **Características:**
                     - 🌍 Traducción profesional automática
-                    - 📊 Exporta la traducción a formato CSV
+                    - 📊 Vista tabular igual que la factura original
+                    - 📥 Exporta a CSV con el mismo formato
                     - ✅ Mantiene la estructura original
-                    - 🚀 Traducción en segundos
                     """)
     
     gr.Markdown("---")
@@ -1435,7 +1610,7 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
     
     1. **📄 Extracción Automática:** Sube tu PDF y obtén datos estructurados en CSV tabular (separado por comas)
     2. **🤖 Asistente IA con Avatar:** Pregunta cualquier duda y escucha la respuesta con voz robótica
-    3. **🔬 Análisis Avanzado:** 7 herramientas profesionales de análisis
+    3. **🔬 Análisis Avanzado:** 12 herramientas profesionales de análisis
     4. **🌍 Traducción:** Traduce y exporta a CSV en 5 idiomas
     
     ---
@@ -1468,6 +1643,7 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
     
     **🌐 Traducción Avanzada:**
     - Traduce a 5 idiomas
+    - Vista tabular traducida
     - Exporta traducciones a CSV
     - Mantiene formato profesional
     
@@ -1617,28 +1793,16 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
         client = InferenceClient(token=token)
         return f"### 📊 Resumen Ejecutivo - Dashboard\n\n{generar_resumen_ejecutivo(datos_json, client)}"
     
-    def ejecutar_traduccion(texto, idioma):
+    # Traducción completa con tabla
+    def ejecutar_traduccion_completa(texto, datos_json, idioma):
         if not texto:
-            return "❌ Procesa una factura primero"
+            return "❌ Procesa una factura primero", None, None
         token = os.getenv("aa")
         if not token:
-            return "❌ Error de configuración"
+            return "❌ Error de configuración", None, None
         client = InferenceClient(token=token)
-        return traducir_factura(texto, idioma, client)
-    
-    # Variable de estado para la traducción
-    traduccion_texto_state = gr.State()
-    idioma_actual_state = gr.State()
-    
-    def ejecutar_traduccion_con_estado(texto, idioma):
-        resultado = ejecutar_traduccion(texto, idioma)
-        return resultado, resultado, idioma
-    
-    def generar_csv_traduccion(traduccion, idioma):
-        if not traduccion or traduccion.startswith("❌"):
-            return None
-        csv_file = traduccion_a_csv(traduccion, idioma)
-        return csv_file
+        texto_traducido, df_traducido, csv_filename = traducir_factura_con_csv(datos_json, texto, idioma, client)
+        return texto_traducido, df_traducido, csv_filename
     
     def ejecutar_duplicados(datos_json):
         if not datos_json:
@@ -1749,7 +1913,7 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
 **Advertencias:**
 {chr(10).join([f"- ⚠️ {a}" for a in resultado.get('advertencias', [])]) if resultado.get('advertencias') else '- No hay advertencias'}"""
     
-    # Conectar nuevas funcionalidades
+    # Conectar funcionalidades
     btn_duplicados.click(fn=ejecutar_duplicados, inputs=[datos_json_state], outputs=[resultado_duplicados])
     btn_impacto.click(fn=ejecutar_impacto, inputs=[datos_json_state], outputs=[resultado_impacto])
     btn_recordatorios.click(fn=ejecutar_recordatorios, inputs=[datos_json_state], outputs=[resultado_recordatorios])
@@ -1764,16 +1928,11 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
     btn_prediccion.click(fn=ejecutar_prediccion, inputs=[datos_json_state], outputs=[resultado_prediccion])
     btn_ejecutivo.click(fn=ejecutar_ejecutivo, inputs=[datos_json_state], outputs=[resultado_ejecutivo])
     
+    # Traducción con tabla
     btn_traducir.click(
-        fn=ejecutar_traduccion_con_estado,
-        inputs=[texto_extraido, idioma_selector],
-        outputs=[resultado_traduccion, traduccion_texto_state, idioma_actual_state]
-    )
-    
-    btn_csv_traduccion.click(
-        fn=generar_csv_traduccion,
-        inputs=[traduccion_texto_state, idioma_actual_state],
-        outputs=[csv_traduccion_output]
+        fn=ejecutar_traduccion_completa,
+        inputs=[texto_extraido, datos_json_state, idioma_selector],
+        outputs=[resultado_traduccion, tabla_traduccion, csv_traduccion_output]
     )
 
 if __name__ == "__main__":