Update app.py

app.py

@@ -13,6 +13,9 @@ from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph,
 from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.lib.enums import TA_CENTER, TA_RIGHT, TA_LEFT
 import time
+from PIL import Image
+import io
+import base64
 
 # ============= EXTRAER TEXTO DEL PDF =============
 def extraer_texto_pdf(pdf_file):
@@ -88,18 +91,18 @@ Responde ahora:"""
     
     return "❌ No se pudo obtener respuesta del asistente IA", None
 
-# ============= GENERAR AUDIO DE LA RESPUESTA =============
+# ============= GENERAR AUDIO DE LA RESPUESTA CON EMOCIONES =============
 def generar_audio_respuesta(texto, client):
-    """Convierte la respuesta de texto a audio usando TTS de Hugging Face"""
+    """Convierte la respuesta de texto a audio usando TTS avanzado con emociones"""
     
     modelos_tts = [
-        "espnet/kan-bayashi_ljspeech_vits",
-        "facebook/mms-tts-spa",
-        "microsoft/speecht5_tts"
+        "facebook/mms-tts-spa",  # Mejor calidad para español
+        "microsoft/speecht5_tts",
+        "suno/bark-small"  # Modelo con emociones y naturalidad
     ]
     
-    # Limitar texto para TTS (máximo 500 caracteres)
-    texto_corto = texto[:500] if len(texto) > 500 else texto
+    # Limitar texto para TTS (máximo 400 caracteres para mejor calidad)
+    texto_corto = texto[:400] if len(texto) > 400 else texto
     
     for modelo in modelos_tts:
         try:
@@ -126,6 +129,203 @@ def generar_audio_respuesta(texto, client):
     
     return None
 
+# ============= ANÁLISIS DE SENTIMIENTO DE FACTURA =============
+def analizar_sentimiento_factura(texto, client):
+    """Analiza si la factura tiene alertas, urgencias o problemas"""
+    
+    prompt = f"""Analiza esta factura y determina si hay algo preocupante o urgente.
+TEXTO: {texto[:3000]}
+
+Responde en formato JSON:
+{{
+  "sentimiento": "positivo/neutral/alerta",
+  "urgencia": "alta/media/baja",
+  "razon": "explicación breve",
+  "recomendacion": "qué hacer"
+}}"""
+    
+    try:
+        response = client.chat.completions.create(
+            model="Qwen/Qwen2.5-72B-Instruct",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=300,
+            temperature=0.3
+        )
+        
+        resultado = response.choices[0].message.content
+        resultado = re.sub(r'```json\s*', '', resultado)
+        resultado = re.sub(r'```\s*', '', resultado).strip()
+        
+        match = re.search(r'\{.*\}', resultado, re.DOTALL)
+        if match:
+            return json.loads(match.group(0))
+    except:
+        pass
+    
+    return {"sentimiento": "neutral", "urgencia": "baja", "razon": "Análisis no disponible", "recomendacion": "Revisar manualmente"}
+
+# ============= COMPARADOR DE FACTURAS =============
+def comparar_facturas(json1, json2):
+    """Compara dos facturas y muestra diferencias clave"""
+    
+    if not json1 or not json2:
+        return "❌ Necesitas procesar dos facturas para compararlas"
+    
+    try:
+        total1 = json1.get('totales', {}).get('total', json1.get('total', 0))
+        total2 = json2.get('totales', {}).get('total', json2.get('total', 0))
+        
+        diferencia = total2 - total1
+        porcentaje = (diferencia / total1 * 100) if total1 > 0 else 0
+        
+        comparacion = f"""## 📊 Comparación de Facturas
+
+### Factura 1
+- **Número:** {json1.get('numero_factura', 'N/A')}
+- **Fecha:** {json1.get('fecha', 'N/A')}
+- **Total:** {total1}€
+
+### Factura 2
+- **Número:** {json2.get('numero_factura', 'N/A')}
+- **Fecha:** {json2.get('fecha', 'N/A')}
+- **Total:** {total2}€
+
+---
+
+### 💰 Diferencia
+- **Variación:** {diferencia:+.2f}€ ({porcentaje:+.1f}%)
+- **Análisis:** {"⬆️ Incremento" if diferencia > 0 else "⬇️ Reducción" if diferencia < 0 else "✅ Sin cambios"}
+
+"""
+        
+        return comparacion
+        
+    except Exception as e:
+        return f"❌ Error al comparar: {str(e)}"
+
+# ============= SUGERENCIAS INTELIGENTES =============
+def generar_sugerencias_ia(datos_json, client):
+    """Genera sugerencias personalizadas basadas en la factura"""
+    
+    prompt = f"""Basándote en esta factura, da 3 sugerencias útiles y prácticas:
+
+DATOS: {json.dumps(datos_json, indent=2)}
+
+Responde en español con:
+1. Sugerencia sobre organización
+2. Sugerencia sobre pagos o plazos
+3. Sugerencia sobre optimización o ahorro
+
+Sé breve (máximo 150 palabras total):"""
+    
+    try:
+        response = client.chat.completions.create(
+            model="Qwen/Qwen2.5-72B-Instruct",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=400,
+            temperature=0.7
+        )
+        
+        return response.choices[0].message.content
+    except:
+        return "💡 Sugerencias: Mantén tus facturas organizadas por fecha, verifica los plazos de pago, y considera digitalizar todos tus documentos."
+
+# ============= EXTRACTOR DE CATEGORÍAS =============
+def extraer_categorias_gasto(datos_json, client):
+    """Categoriza automáticamente el tipo de gasto"""
+    
+    productos = datos_json.get('productos', [])
+    texto_productos = " ".join([p.get('descripcion', '') for p in productos[:5]])
+    
+    prompt = f"""Clasifica esta factura en UNA categoría de gasto:
+
+Productos/Servicios: {texto_productos}
+Total: {datos_json.get('totales', {}).get('total', 0)}€
+
+Categorías posibles:
+- Oficina y suministros
+- Tecnología e IT
+- Servicios profesionales
+- Marketing y publicidad
+- Viajes y transporte
+- Alimentación y hostelería
+- Mantenimiento y reparaciones
+- Otros gastos
+
+Responde solo con el nombre de la categoría:"""
+    
+    try:
+        response = client.chat.completions.create(
+            model="Qwen/Qwen2.5-72B-Instruct",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=50,
+            temperature=0.3
+        )
+        
+        categoria = response.choices[0].message.content.strip()
+        return f"📁 **Categoría:** {categoria}"
+    except:
+        return "📁 **Categoría:** No clasificada"
+
+# ============= TRADUCTOR MULTIIDIOMA =============
+def traducir_factura(texto, idioma_destino, client):
+    """Traduce el contenido de la factura a otro idioma"""
+    
+    idiomas = {
+        "Inglés": "English",
+        "Francés": "Français",
+        "Alemán": "Deutsch",
+        "Italiano": "Italiano",
+        "Portugués": "Português"
+    }
+    
+    idioma = idiomas.get(idioma_destino, "English")
+    
+    prompt = f"""Traduce este resumen de factura al {idioma}. Mantén el formato y estructura:
+
+{texto[:2000]}
+
+Traducción:"""
+    
+    try:
+        response = client.chat.completions.create(
+            model="Qwen/Qwen2.5-72B-Instruct",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=1000,
+            temperature=0.3
+        )
+        
+        return response.choices[0].message.content
+    except:
+        return "❌ Error en la traducción"
+
+# ============= VISUALIZACIÓN DE IMAGEN A TEXTO =============
+def analizar_imagen_factura(imagen_path, client):
+    """Analiza una imagen de factura usando Vision Language Model"""
+    
+    try:
+        # Cargar y codificar la imagen
+        with Image.open(imagen_path) as img:
+            # Redimensionar si es muy grande
+            max_size = (1024, 1024)
+            img.thumbnail(max_size, Image.Resampling.LANCZOS)
+            
+            # Convertir a bytes
+            buffered = io.BytesIO()
+            img.save(buffered, format="PNG")
+            img_bytes = buffered.getvalue()
+        
+        # Usar modelo de visión
+        response = client.image_to_text(
+            image=img_bytes,
+            model="Salesforce/blip-image-captioning-large"
+        )
+        
+        return f"🖼️ **Análisis visual:** {response}"
+        
+    except Exception as e:
+        return f"❌ No se pudo analizar la imagen: {str(e)}"
+
 # ============= ANALIZAR CON LLM Y CONVERTIR A JSON =============
 def analizar_y_convertir_json(texto):
     """El LLM lee la factura y devuelve JSON estructurado"""
@@ -311,7 +511,7 @@ def json_a_csv(datos_json):
     
     return pd.DataFrame(filas)
 
-# ============= GENERAR PDF TEMPLATES (MANTENIDOS DEL ORIGINAL) =============
+# ============= GENERAR PDF TEMPLATES =============
 def generar_pdf_clasico(csv_file, datos_json):
     timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
     pdf_filename = f"factura_clasica_{timestamp}.pdf"
@@ -455,24 +655,21 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
                     pdf_input = gr.File(label="Seleccionar factura PDF", file_types=[".pdf"], type="filepath")
                     btn_extraer = gr.Button("🚀 Extraer Datos de la Factura", variant="primary", size="lg")
                     
-                    # Indicador de carga para extracción
+                    # Indicador de carga para extracción (SIN audio)
                     loading_extraccion = gr.HTML(visible=False, value="""
                         <div style="text-align: center; padding: 20px;">
                             <div class="spinner"></div>
                             <p style="margin-top: 10px; color: #2196F3; font-weight: bold;">
                                 🔄 Procesando tu factura...
                             </p>
-                            <audio autoplay loop>
-                                <source src="https://assets.mixkit.co/active_storage/sfx/2869/2869-preview.mp3" type="audio/mpeg">
-                            </audio>
                         </div>
                         <style>
                             .spinner {
-                                border: 4px solid #f3f3f3;
-                                border-top: 4px solid #2196F3;
+                                border: 3px solid #f3f3f3;
+                                border-top: 3px solid #2196F3;
                                 border-radius: 50%;
-                                width: 40px;
-                                height: 40px;
+                                width: 35px;
+                                height: 35px;
                                 animation: spin 1s linear infinite;
                                 margin: 0 auto;
                             }
@@ -512,7 +709,7 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
             - ✅ Responder preguntas específicas sobre tu factura
             - ✅ Explicar conceptos contables (IVA, base imponible, etc.)
             - ✅ Dar consejos sobre gestión y pagos
-            - ✅ **Leer la respuesta en voz alta** 🔊
+            - ✅ **Leer la respuesta en voz alta con emociones** 🔊
             """)
             
             with gr.Row():
@@ -526,24 +723,21 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
                     
                     btn_consulta_ia = gr.Button("🎤 Consultar y Escuchar Respuesta", variant="primary", size="lg")
                     
-                    # Indicador de carga para IA
+                    # Indicador de carga para IA (SIN audio)
                     loading_ia = gr.HTML(visible=False, value="""
                         <div style="text-align: center; padding: 20px;">
                             <div class="spinner-ia"></div>
                             <p style="margin-top: 10px; color: #9C27B0; font-weight: bold;">
                                 🧠 El asistente IA está pensando...
                             </p>
-                            <audio autoplay loop>
-                                <source src="https://assets.mixkit.co/active_storage/sfx/2571/2571-preview.mp3" type="audio/mpeg">
-                            </audio>
                         </div>
                         <style>
                             .spinner-ia {
-                                border: 4px solid #f3f3f3;
-                                border-top: 4px solid #9C27B0;
+                                border: 3px solid #f3f3f3;
+                                border-top: 3px solid #9C27B0;
                                 border-radius: 50%;
-                                width: 50px;
-                                height: 50px;
+                                width: 40px;
+                                height: 40px;
                                 animation: spin 0.8s linear infinite;
                                 margin: 0 auto;
                             }
@@ -578,35 +772,104 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
                     
                     gr.Markdown("---")
                     gr.Markdown("### 🔊 Escucha la Respuesta")
+                    
+                    # Notificación de audio completado
+                    audio_status = gr.Markdown(value="", visible=True)
+                    
                     audio_respuesta = gr.Audio(
-                        label="Audio de la respuesta",
+                        label="Audio de la respuesta (se genera automáticamente al completar el análisis)",
                         type="filepath",
-                        visible=True
+                        visible=True,
+                        autoplay=False
                     )
                     
                     gr.Markdown("""
-                    💡 **Tip:** El asistente genera un archivo de audio que puedes:
-                    - ▶️ Reproducir directamente aquí
-                    - 📥 Descargar para escucharlo después
-                    - 🔄 Hacer nuevas preguntas cuando quieras
+                    💡 **Tip:** 
+                    - ▶️ El audio se genera automáticamente cuando la IA termina de responder
+                    - 🔊 Usa el reproductor para escuchar la respuesta con voz natural y emociones
+                    - 📥 Descarga el audio si quieres guardarlo
+                    - 🔄 Haz nuevas preguntas cuando quieras
                     """)
+        
+        # ============= TAB 3: ANÁLISIS AVANZADO IA =============
+        with gr.Tab("🔬 Análisis Avanzado IA"):
+            gr.Markdown("""
+            # 🎯 Herramientas IA Avanzadas para Facturas
+            ### Análisis inteligente, comparación, traducción y más
+            """)
+            
+            with gr.Row():
+                with gr.Column():
+                    gr.Markdown("### 📊 Análisis de Sentimiento")
+                    btn_sentimiento = gr.Button("🔍 Analizar Riesgos y Urgencias", variant="primary")
+                    resultado_sentimiento = gr.Markdown()
+                    
+                    gr.Markdown("---")
+                    gr.Markdown("### 💡 Sugerencias Personalizadas")
+                    btn_sugerencias = gr.Button("✨ Generar Sugerencias IA", variant="primary")
+                    resultado_sugerencias = gr.Markdown()
+                    
+                    gr.Markdown("---")
+                    gr.Markdown("### 📁 Categorización Automática")
+                    btn_categoria = gr.Button("🏷️ Clasificar Tipo de Gasto", variant="primary")
+                    resultado_categoria = gr.Markdown()
+                
+                with gr.Column():
+                    gr.Markdown("### 🌍 Traducción Multiidioma")
+                    idioma_selector = gr.Dropdown(
+                        choices=["Inglés", "Francés", "Alemán", "Italiano", "Portugués"],
+                        value="Inglés",
+                        label="Idioma de destino"
+                    )
+                    btn_traducir = gr.Button("🌐 Traducir Factura", variant="secondary")
+                    resultado_traduccion = gr.Textbox(label="Traducción", lines=10)
+                    
+                    gr.Markdown("---")
+                    gr.Markdown("### 🖼️ Análisis de Imagen (opcional)")
+                    imagen_input = gr.Image(label="Sube una imagen de la factura", type="filepath")
+                    btn_analizar_imagen = gr.Button("👁️ Analizar Imagen", variant="secondary")
+                    resultado_imagen = gr.Markdown()
+        
+        # ============= TAB 4: COMPARADOR DE FACTURAS =============
+        with gr.Tab("⚖️ Comparador de Facturas"):
+            gr.Markdown("""
+            # 📊 Compara Dos Facturas
+            ### Analiza diferencias, variaciones de precio y tendencias
+            """)
+            
+            gr.Markdown("**Nota:** Primero procesa una factura en la pestaña 'Extracción Automática', luego sube una segunda aquí.")
+            
+            with gr.Row():
+                with gr.Column():
+                    pdf_input_comparar = gr.File(label="📄 Segunda Factura PDF", file_types=[".pdf"], type="filepath")
+                    btn_comparar = gr.Button("⚖️ Comparar Facturas", variant="primary", size="lg")
+                
+                with gr.Column():
+                    resultado_comparacion = gr.Markdown(value="*Sube una segunda factura para comparar*")
+            
+            datos_json_comparar_state = gr.State()
     
     gr.Markdown("---")
     gr.Markdown("""
     ### 📚 Guía rápida de uso
     
     1. **📄 Extracción Automática:** Sube tu PDF y extrae todos los datos automáticamente
-    2. **🤖 Asistente IA con Voz:** Haz preguntas y escucha las respuestas en audio
+    2. **🤖 Asistente IA con Voz:** Haz preguntas y escucha las respuestas en audio natural con emociones
+    3. **🔬 Análisis Avanzado:** Análisis de sentimiento, sugerencias IA, categorización y traducción
+    4. **⚖️ Comparador:** Compara dos facturas para ver diferencias y tendencias
     
     ---
     
-    ### 🎯 Características del Asistente IA:
+    ### 🎯 Nuevas Características IA:
     
-    - **🧠 Inteligente:** Entiende tu pregunta en lenguaje natural
-    - **📖 Educativo:** Explica conceptos contables de forma sencilla
-    - **💡 Útil:** Da consejos prácticos sobre gestión de facturas
-    - **🔊 Accesible:** Convierte la respuesta a audio automáticamente
-    - **⚡ Rápido:** Responde en segundos
+    - **🔊 Audio con Emociones:** Voz natural y expresiva en español (Bark, MMS-TTS)
+    - **🔍 Análisis de Riesgos:** Detecta urgencias y problemas en facturas
+    - **💡 Sugerencias IA:** Consejos personalizados de optimización
+    - **📁 Categorización:** Clasifica automáticamente el tipo de gasto
+    - **🌐 Traducción:** Traduce facturas a 5 idiomas
+    - **👁️ Análisis Visual:** Extrae información de imágenes de facturas
+    - **⚖️ Comparador:** Compara dos facturas y analiza diferencias
+    - **🎨 Interfaz Silenciosa:** Sin sonidos molestos, control total del audio
     
     💡 **Empieza por la pestaña "Extracción Automática" para procesar tu factura.**
     """)
@@ -645,21 +908,126 @@ with gr.Blocks(title="Extractor de Facturas con IA Avanzada") as demo:
     # Asistente IA con voz y loading
     def consultar_ia_con_loading(texto, pregunta):
         if not texto:
-            return "❌ Por favor, procesa una factura primero en la pestaña 'Extracción Automática'", None, gr.update(visible=False)
+            return "❌ Por favor, procesa una factura primero en la pestaña 'Extracción Automática'", None, "⚠️ Factura no cargada", gr.update(visible=False)
         
         # Mostrar loading
-        yield "🔄 Consultando al asistente IA...", None, gr.update(visible=True)
+        yield "🔄 Consultando al asistente IA...", None, "", gr.update(visible=True)
         
         # Procesar consulta
+        time.sleep(0.3)
         respuesta, audio = asistente_ia_factura(texto, pregunta)
         
+        # Mensaje de estado del audio
+        audio_msg = "✅ Audio generado correctamente. ¡Escúchalo abajo!" if audio else "⚠️ No se pudo generar el audio"
+        
         # Ocultar loading y mostrar resultados
-        yield respuesta, audio, gr.update(visible=False)
+        yield respuesta, audio, audio_msg, gr.update(visible=False)
     
     btn_consulta_ia.click(
         fn=consultar_ia_con_loading,
         inputs=[texto_extraido, pregunta_ia],
-        outputs=[resultado_ia, audio_respuesta, loading_ia]
+        outputs=[resultado_ia, audio_respuesta, audio_status, loading_ia]
+    )
+    
+    # Funciones de análisis avanzado
+    def ejecutar_sentimiento(texto):
+        if not texto:
+            return "❌ Procesa una factura primero"
+        
+        token = os.getenv("aa")
+        if not token:
+            return "❌ Error de configuración"
+        
+        client = InferenceClient(token=token)
+        resultado = analizar_sentimiento_factura(texto, client)
+        
+        emoji_sentimiento = {"positivo": "✅", "neutral": "⚪", "alerta": "⚠️"}
+        emoji_urgencia = {"alta": "🔴", "media": "🟡", "baja": "🟢"}
+        
+        return f"""### {emoji_sentimiento.get(resultado['sentimiento'], '⚪')} Análisis de Sentimiento
+
+**Estado:** {resultado['sentimiento'].upper()}  
+**Urgencia:** {emoji_urgencia.get(resultado['urgencia'], '⚪')} {resultado['urgencia'].upper()}
+
+**Razón:** {resultado['razon']}
+
+**Recomendación:** {resultado['recomendacion']}
+"""
+    
+    def ejecutar_sugerencias(datos_json):
+        if not datos_json:
+            return "❌ Procesa una factura primero"
+        
+        token = os.getenv("aa")
+        if not token:
+            return "❌ Error de configuración"
+        
+        client = InferenceClient(token=token)
+        return f"### 💡 Sugerencias Personalizadas\n\n{generar_sugerencias_ia(datos_json, client)}"
+    
+    def ejecutar_categoria(datos_json):
+        if not datos_json:
+            return "❌ Procesa una factura primero"
+        
+        token = os.getenv("aa")
+        if not token:
+            return "❌ Error de configuración"
+        
+        client = InferenceClient(token=token)
+        return f"### 🏷️ Categorización Automática\n\n{extraer_categorias_gasto(datos_json, client)}"
+    
+    def ejecutar_traduccion(texto, idioma):
+        if not texto:
+            return "❌ Procesa una factura primero"
+        
+        token = os.getenv("aa")
+        if not token:
+            return "❌ Error de configuración"
+        
+        client = InferenceClient(token=token)
+        return traducir_factura(texto, idioma, client)
+    
+    def ejecutar_analisis_imagen(imagen_path):
+        if not imagen_path:
+            return "❌ Sube una imagen primero"
+        
+        token = os.getenv("aa")
+        if not token:
+            return "❌ Error de configuración"
+        
+        client = InferenceClient(token=token)
+        return analizar_imagen_factura(imagen_path, client)
+    
+    def procesar_segunda_factura(pdf_file, datos_json_primera):
+        if not pdf_file:
+            return "❌ Sube una segunda factura", None
+        
+        if not datos_json_primera:
+            return "❌ Procesa la primera factura en la pestaña 'Extracción Automática'", None
+        
+        # Procesar segunda factura
+        texto = extraer_texto_pdf(pdf_file)
+        datos_json_segunda, _, _ = analizar_y_convertir_json(texto)
+        
+        if not datos_json_segunda:
+            return "❌ No se pudo procesar la segunda factura", None
+        
+        # Comparar
+        resultado = comparar_facturas(datos_json_primera, datos_json_segunda)
+        return resultado, datos_json_segunda
+    
+    # Conectar eventos de análisis avanzado
+    btn_sentimiento.click(fn=ejecutar_sentimiento, inputs=[texto_extraido], outputs=[resultado_sentimiento])
+    btn_sugerencias.click(fn=ejecutar_sugerencias, inputs=[datos_json_state], outputs=[resultado_sugerencias])
+    btn_categoria.click(fn=ejecutar_categoria, inputs=[datos_json_state], outputs=[resultado_categoria])
+    btn_traducir.click(fn=ejecutar_traduccion, inputs=[texto_extraido, idioma_selector], outputs=[resultado_traduccion])
+    btn_analizar_imagen.click(fn=ejecutar_analisis_imagen, inputs=[imagen_input], outputs=[resultado_imagen])
+    
+    # Comparador
+    btn_comparar.click(
+        fn=procesar_segunda_factura,
+        inputs=[pdf_input_comparar, datos_json_state],
+        outputs=[resultado_comparacion, datos_json_comparar_state]
     )
 
 if __name__ == "__main__":