Update app.py

app.py

@@ -3,6 +3,11 @@ import gradio as gr
 import networkx as nx
 import plotly.graph_objects as go
 from datetime import datetime
+import re
+import os
+from fpdf import FPDF, HTMLMixin
+
+# ---------------- Funciones de análisis y grafo -------------------
 
 def get_transaction(tx_id):
     """
@@ -45,10 +50,8 @@ def generar_grafo(tx_data):
     Nodos secundarios: direcciones de inputs y outputs.
     """
     G = nx.DiGraph()
-    
     txid = tx_data.get("txid", "desconocido")
     G.add_node(txid, color="#FF6B6B", size=20, tipo="Transacción")
-    
     # Inputs
     for inp in tx_data.get("vin", []):
         prevout = inp.get("prevout", {})
@@ -56,28 +59,25 @@ def generar_grafo(tx_data):
         if addr:
             G.add_node(addr, color="#4ECDC4", size=15, tipo="Input")
             G.add_edge(addr, txid)
-    
     # Outputs
     for out in tx_data.get("vout", []):
         addr = out.get("scriptpubkey_address")
         if addr:
             G.add_node(addr, color="#45B7D1", size=15, tipo="Output")
             G.add_edge(txid, addr)
-    
     return G
 
 def check_blockchain_tags(tx_id):
     """
     Consulta la API pública de blockchain.com para ver si el TXID (o sus metadatos)
     incluye información que indique que la transacción ha sido etiquetada como MIXER.
-    Esta función es un ejemplo y puede requerir ajustes según la estructura real del JSON.
+    (Esta función es un ejemplo y podría necesitar ajustes según la respuesta real.)
     """
     url = f"https://blockchain.info/rawtx/{tx_id}?format=json"
     try:
         response = requests.get(url, timeout=10)
         if response.status_code == 200:
             data = response.json()
-            # Se asume que la API podría incluir un campo 'tags' o 'category'
             tags = data.get("tags", [])
             if "MIXER" in tags:
                 return True
@@ -92,106 +92,75 @@ def check_blockchain_tags(tx_id):
 def analizar_transaccion(tx_id):
     """
     Analiza una transacción Bitcoin:
-      - Obtiene los datos desde Blockstream.info o mempool.space.
+      - Obtiene datos desde Blockstream.info o mempool.space.
       - Calcula totales en BTC y fee, mostrando también su equivalente en USD.
-      - Muestra información adicional: versión, tamaño, peso y fee rate.
-      - Aplica una heurística mejorada para detectar posibles CoinJoin/mixers:
-            * Si tiene >5 inputs y >5 outputs y menos de 3 montos únicos, o
-            * Si tiene >2 outputs y 2 o menos montos únicos.
-      - Incorpora información adicional de blockchain.com (si está disponible) para detectar etiquetas MIXER.
+      - Muestra información adicional (versión, tamaño, peso, fee rate).
+      - Aplica heurística para detectar posibles CoinJoin/mixers.
+      - Incorpora información adicional de blockchain.com (etiqueta MIXER).
       - Genera un grafo interactivo.
-      - Retorna un reporte en HTML y la figura del grafo.
-      
-    Nota: La detección se basa en heurísticas y en la información adicional de blockchain.com.
-    Se recomienda complementar con herramientas especializadas para un análisis forense completo.
+      - Retorna un informe en HTML y la figura del grafo.
     """
     tx_data = get_transaction(tx_id)
     if not tx_data:
-        return ("❌ Transacción no encontrada o error al obtener datos. "
-                "Asegúrate de ingresar un TXID de Bitcoin válido."), None
+        return ("❌ Transacción no encontrada o error al obtener datos. Asegúrate de ingresar un TXID válido."), None, None
     
     try:
-        # Datos básicos
         num_inputs = len(tx_data.get("vin", []))
         num_outputs = len(tx_data.get("vout", []))
         montos = [out.get("value", 0) / 1e8 for out in tx_data.get("vout", [])]
         montos_unicos = len(set(montos))
-        
-        # Cálculos forenses en BTC
         total_input_value = sum(inp.get("prevout", {}).get("value", 0) for inp in tx_data.get("vin", [])) / 1e8
         total_output_value = sum(out.get("value", 0) for out in tx_data.get("vout", [])) / 1e8
         fee = total_input_value - total_output_value
 
-        # Heurística para detectar CoinJoin/mixer
+        # Heurística para detectar mixer
         heuristic_mixer = ((num_inputs > 5 and num_outputs > 5 and montos_unicos < 3) or
                            (num_outputs > 2 and montos_unicos <= 2))
-        
-        # Verificar información adicional vía blockchain.com
         es_mixer_blockchain = check_blockchain_tags(tx_id)
-        
-        # Combinamos ambas señales: si alguna indica mixer, marcamos la transacción.
         es_mixer = heuristic_mixer or es_mixer_blockchain
 
-        # Construir mensaje detallado según la fuente de la alerta
         if es_mixer:
             if heuristic_mixer and es_mixer_blockchain:
                 mixer_message = (
                     "<p style='color: #FF5252;'><strong>Alerta de Mixer / CoinJoin:</strong> "
-                    "La transacción presenta patrones compatibles con un mixer (CoinJoin) según la heurística, "
-                    "y la API de blockchain.com indica que está etiquetada como MIXER. Se recomienda un análisis forense adicional.</p>"
+                    "La transacción cumple criterios heurísticos y la API de blockchain.com la etiqueta como MIXER.</p>"
                 )
             elif heuristic_mixer:
                 mixer_message = (
                     "<p style='color: #FF5252;'><strong>Alerta de Mixer / CoinJoin:</strong> "
-                    "La transacción presenta patrones compatibles con un mixer (CoinJoin) según la heurística. "
-                    "Se recomienda un análisis forense adicional.</p>"
+                    "La transacción cumple criterios heurísticos compatibles con un mixer.</p>"
                 )
             elif es_mixer_blockchain:
                 mixer_message = (
                     "<p style='color: #FF5252;'><strong>Alerta de Mixer / CoinJoin:</strong> "
-                    "La API de blockchain.com indica que esta transacción está etiquetada como MIXER. "
-                    "Se recomienda un análisis forense adicional.</p>"
+                    "La API de blockchain.com etiqueta esta transacción como MIXER.</p>"
                 )
         else:
             mixer_message = (
                 "<p style='color: #4CAF50;'><strong>Sin indicios de mezcla:</strong> "
-                "La transacción no muestra patrones comunes asociados a servicios mixer.</p>"
+                "La transacción no presenta patrones de mixer.</p>"
             )
-        
-        # Obtener precio BTC actual
-        btc_price = get_btc_price()
-        if btc_price is None:
-            btc_price = 0
-        
-        # Datos adicionales de la transacción
+
+        btc_price = get_btc_price() or 0
+
         version = tx_data.get("version", "N/A")
         size = tx_data.get("size", None)
         weight = tx_data.get("weight", "N/A")
-        if size:
-            fee_rate = (fee * 1e8) / size  # en sat/byte
-            fee_rate_str = f"{fee_rate:.2f} sat/byte"
-        else:
-            fee_rate_str = "N/A"
-        
-        # Formatear fecha y hora (si la transacción está confirmada)
+        fee_rate_str = f"{(fee * 1e8) / size:.2f} sat/byte" if size else "N/A"
+
         status = tx_data.get("status", {})
         block_time = status.get("block_time")
-        if block_time:
-            fecha_hora_str = datetime.fromtimestamp(block_time).strftime('%Y-%m-%d %H:%M:%S')
-        else:
-            fecha_hora_str = "Desconocida"
-        
-        # Generar grafo de la transacción
+        fecha_hora_str = datetime.fromtimestamp(block_time).strftime('%Y-%m-%d %H:%M:%S') if block_time else "Desconocida"
+
+        # Generar grafo
         G = generar_grafo(tx_data)
         pos = nx.spring_layout(G, seed=42)
-        
         edge_x, edge_y = [], []
         for edge in G.edges():
             x0, y0 = pos[edge[0]]
             x1, y1 = pos[edge[1]]
             edge_x.extend([x0, x1, None])
             edge_y.extend([y0, y1, None])
-        
         node_x, node_y, hover_texts, node_colors, node_sizes = [], [], [], [], []
         for node in G.nodes():
             x, y = pos[node]
@@ -201,7 +170,6 @@ def analizar_transaccion(tx_id):
             hover_texts.append(f"Tipo: {tipo}<br>ID: {node[:8]}...")
             node_colors.append(G.nodes[node].get("color", "#FFFFFF"))
             node_sizes.append(G.nodes[node].get("size", 10))
-        
         fig = go.Figure(
             data=[
                 go.Scatter(
@@ -228,40 +196,35 @@ def analizar_transaccion(tx_id):
                 template="plotly_dark"
             )
         )
-        
-        # Calcular detalles de outputs: agrupar direcciones por monto
-        unique_outputs_dict = {}
-        for out in tx_data.get("vout", []):
-            amount = out.get("value", 0) / 1e8
-            addr = out.get("scriptpubkey_address")
-            if addr:
-                unique_outputs_dict.setdefault(amount, []).append(addr)
-        
-        unique_outputs_details = "<ul>"
-        for amt, addrs in unique_outputs_dict.items():
-            unique_outputs_details += f"<li>{amt:.8f} BTC: {', '.join(addrs)}</li>"
-        unique_outputs_details += "</ul>"
-        
-        # Si se detecta posible mixer, se muestran métricas adicionales de outputs;
-        # de lo contrario, se omiten.
+
+        # Agrupar detalles de outputs (solo si se detecta mixer)
         if es_mixer:
-            mixer_metrics = f"""
-            <p>📤 <strong>Outputs:</strong> {num_outputs}</p>
-            <p>💰 <strong>Montos únicos en outputs:</strong> {montos_unicos}</p>
-            <p><strong>Detalles de outputs únicos:</strong></p>
-            {unique_outputs_details}
-            """
+            unique_outputs_dict = {}
+            for out in tx_data.get("vout", []):
+                amount = out.get("value", 0) / 1e8
+                addr = out.get("scriptpubkey_address")
+                if addr:
+                    unique_outputs_dict.setdefault(amount, []).append(addr)
+            unique_outputs_details = "<ul>"
+            for amt, addrs in unique_outputs_dict.items():
+                unique_outputs_details += f"<li>{amt:.8f} BTC: {', '.join(addrs)}</li>"
+            unique_outputs_details += "</ul>"
+            mixer_metrics = (
+                f"<p>📤 <strong>Outputs:</strong> {num_outputs}</p>"
+                f"<p>💰 <strong>Montos únicos en outputs:</strong> {montos_unicos}</p>"
+                f"<p><strong>Detalles de outputs únicos:</strong></p>{unique_outputs_details}"
+            )
         else:
             mixer_metrics = ""
-        
-        # Formateo de montos con equivalentes en USD
+
         total_input_str = f"{total_input_value:.8f} BTC"
         total_input_str += f" (${total_input_value * btc_price:,.2f})" if btc_price else ""
         total_output_str = f"{total_output_value:.8f} BTC"
         total_output_str += f" (${total_output_value * btc_price:,.2f})" if btc_price else ""
         fee_str = f"{fee:.8f} BTC"
         fee_str += f" (${fee * btc_price:,.2f})" if btc_price else ""
-        
+
+        # Construir el informe HTML
         reporte = f"""
         <div style="padding: 20px; border-radius: 10px;">
             <h3 style="color: {'#FF5252' if es_mixer else '#4CAF50'};">
@@ -282,21 +245,82 @@ def analizar_transaccion(tx_id):
             <p>🔔 <strong>Estado de Confirmación:</strong> {"Confirmada" if status.get("confirmed", False) else "No confirmada"}</p>
             {mixer_message}
             <p style="font-size: 0.9em;">
-                Nota: La detección se basa en heurísticas y en la información adicional de blockchain.com. Se recomienda usar herramientas especializadas (como CoinJoin explorers) para un análisis forense completo.
+                Nota: La detección se basa en heurísticas y en información adicional de blockchain.com. Se recomienda usar herramientas especializadas para un análisis forense completo.
             </p>
         </div>
         """
-        
-        return reporte, fig
+        # Además devolvemos (report, fig) para almacenarlos en un estado
+        return reporte, fig, (reporte, fig)
     
     except Exception as e:
-        return f"⚠️ Error durante el análisis: {str(e)}", None
+        return f"⚠️ Error durante el análisis: {str(e)}", None, None
+
+# ---------------- Función para generar el PDF -------------------
+
+class PDF(FPDF, HTMLMixin):
+    pass
+
+def generar_pdf(report, fig):
+    """
+    Genera un PDF que incluye:
+      - Título: ANALISIS FORENSE DE BLOCKCHAIN
+      - El contenido del informe (texto plano, obtenido removiendo etiquetas HTML)
+      - El grafo (convertido a imagen PNG)
+      - Pie de página: "Generado por José R. Leonett"
+    Retorna el PDF como bytes.
+    """
+    # Guardar el grafo como imagen PNG temporalmente
+    temp_img_path = "temp_graph.png"
+    try:
+        fig.write_image(temp_img_path)
+    except Exception as e:
+        print("Error al guardar la imagen del grafo:", e)
+        temp_img_path = None
+
+    # Extraer texto plano del informe (quitando etiquetas HTML simples)
+    plain_text = re.sub('<[^<]+?>', '', report)
+    
+    pdf = PDF()
+    pdf.add_page()
+    # Título
+    pdf.set_font("Arial", "B", 16)
+    pdf.cell(0, 10, "ANALISIS FORENSE DE BLOCKCHAIN", ln=True, align="C")
+    pdf.ln(5)
+    # Contenido del informe
+    pdf.set_font("Arial", "", 12)
+    pdf.multi_cell(0, 10, plain_text)
+    pdf.ln(5)
+    # Agregar la imagen del grafo si existe
+    if temp_img_path and os.path.exists(temp_img_path):
+        pdf.add_page()
+        # Ajustar el ancho a 190 (márgenes de 10) para que se centre
+        pdf.image(temp_img_path, x=10, y=20, w=190)
+        os.remove(temp_img_path)
+    # Pie de página
+    pdf.set_font("Arial", "I", 10)
+    pdf.ln(10)
+    pdf.cell(0, 10, "Generado por José R. Leonett", ln=True, align="C")
+    
+    pdf_bytes = pdf.output(dest="S").encode("latin1")
+    return pdf_bytes
+
+def descargar_pdf(analysis_tuple):
+    """
+    Función que se activa al presionar el botón DESCARGAR ANALISIS.
+    Recibe como entrada la tupla (report, fig) generada por analizar_transaccion.
+    Si no hay datos, no devuelve nada.
+    """
+    if not analysis_tuple:
+        return None
+    report, fig = analysis_tuple
+    return generar_pdf(report, fig)
+
+# ---------------- INTERFAZ GRÁFICA CON GRADIO -------------------
 
-# INTERFAZ GRÁFICA CON GRADIO
 with gr.Blocks(
     theme=gr.themes.Soft(), 
     title="🔍 Detector de Mixers / CoinJoin en Transacciones Bitcoin",
-    css=".custom-btn { background-color: #ADD8E6 !important; color: black !important; }"
+    css=".custom-btn { background-color: #ADD8E6 !important; color: black !important; padding: 5px 10px; font-size: 12px; }"
 ) as demo:
     gr.Markdown("# 🔍 Detector de Mixers / CoinJoin en Transacciones Bitcoin")
     gr.Markdown("Desarrollado por José R. Leonett para la comunidad de Peritos Forenses Digitales de Guatemala [www.forensedigital.gt](http://www.forensedigital.gt)")
@@ -305,14 +329,12 @@ with gr.Blocks(
     with gr.Row():
         # Columna Izquierda: Grafo y controles de entrada
         with gr.Column(scale=1):
-            # Espacio para el grafo (se actualizará con la transacción analizada)
             plot_output = gr.Plot()
-            # Controles de entrada debajo del grafo
             tx_input = gr.Textbox(
                 label="TXID de la Transacción", 
                 placeholder="Ej: 9dd51e2d45f4f7bddcc3f0f7a05c3fd60543a11cfc9fbd0e1ca4434668cfa3e1"
             )
-            btn = gr.Button("Analizar Transacción", elem_classes=["custom-btn"])
+            analyze_btn = gr.Button("Analizar Transacción", elem_classes=["custom-btn"])
             gr.Markdown("### Ejemplos de TXIDs válidos:")
             gr.Examples(
                 examples=[
@@ -325,7 +347,7 @@ with gr.Blocks(
         # Columna Derecha: Resultados y recuadro explicativo
         with gr.Column(scale=1):
             reporte_html = gr.HTML()
-            explanation_text = """
+            explanation_box = gr.HTML(value="""
             <div style="overflow-y: scroll; height: 220px; border: 1px solid #cccccc; padding: 10px;">
                 <h4>Explicación de los campos:</h4>
                 <ul>
@@ -343,13 +365,20 @@ with gr.Blocks(
                     <li><strong>Total Salidas:</strong> Suma total de los valores de salida (en BTC y USD).</li>
                     <li><strong>Fee:</strong> Diferencia entre entradas y salidas, que representa la tarifa de la transacción (en BTC y USD).</li>
                     <li><strong>Estado de Confirmación:</strong> Indica si la transacción está confirmada o no.</li>
-                    <li><strong>Alerta de Mixer / CoinJoin:</strong> Muestra si la transacción presenta patrones de mezcla, según la heurística y/o información de blockchain.com.</li>
+                    <li><strong>Alerta de Mixer / CoinJoin:</strong> Muestra si la transacción presenta patrones de mezcla.</li>
                 </ul>
             </div>
-            """
-            explanation_box = gr.HTML(value=explanation_text)
+            """)
+    
+    # Estado para almacenar el análisis generado (tupla: (report, fig))
+    analysis_state = gr.State()
+
+    # Al hacer clic en "Analizar Transacción", se actualizan el informe, el grafo y se guarda el estado.
+    analyze_btn.click(fn=analizar_transaccion, inputs=tx_input, outputs=[reporte_html, plot_output, analysis_state])
     
-    # Configuración de la acción del botón
-    btn.click(fn=analizar_transaccion, inputs=tx_input, outputs=[reporte_html, plot_output])
+    # Botón para descargar el análisis en PDF; se activa sólo si hay datos en analysis_state.
+    download_btn = gr.Button("DESCARGAR ANALISIS", elem_classes=["custom-btn"])
+    pdf_file = gr.File(label="Archivo PDF generado")
+    download_btn.click(fn=descargar_pdf, inputs=analysis_state, outputs=pdf_file)
     
 demo.launch()