Update app.py

app.py

@@ -5,6 +5,7 @@ import plotly.graph_objects as go
 from datetime import datetime
 import re
 import os
+import tempfile
 from fpdf import FPDF, HTMLMixin
 
 # ---------------- Funciones de análisis y grafo -------------------
@@ -52,14 +53,12 @@ def generar_grafo(tx_data):
     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", {})
         addr = prevout.get("scriptpubkey_address")
         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:
@@ -98,7 +97,7 @@ def analizar_transaccion(tx_id):
       - Aplica heurística para detectar posibles CoinJoin/mixers.
       - Incorpora información adicional de blockchain.com (etiqueta MIXER).
       - Genera un grafo interactivo.
-      - Retorna un informe en HTML y la figura del grafo.
+      - Retorna un informe en HTML, la figura del grafo y una tupla (interna) para el PDF.
     """
     tx_data = get_transaction(tx_id)
     if not tx_data:
@@ -113,7 +112,6 @@ def analizar_transaccion(tx_id):
         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 mixer
         heuristic_mixer = ((num_inputs > 5 and num_outputs > 5 and montos_unicos < 3) or
                            (num_outputs > 2 and montos_unicos <= 2))
         es_mixer_blockchain = check_blockchain_tags(tx_id)
@@ -197,7 +195,7 @@ def analizar_transaccion(tx_id):
             )
         )
 
-        # Agrupar detalles de outputs (solo si se detecta mixer)
+        # Si se detecta mixer, se muestran métricas adicionales de outputs.
         if es_mixer:
             unique_outputs_dict = {}
             for out in tx_data.get("vout", []):
@@ -224,7 +222,6 @@ def analizar_transaccion(tx_id):
         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'};">
@@ -249,7 +246,7 @@ def analizar_transaccion(tx_id):
             </p>
         </div>
         """
-        # Además devolvemos (report, fig) para almacenarlos en un estado
+        # La tupla interna que usaremos para el PDF es (reporte, fig)
         return reporte, fig, (reporte, fig)
     
     except Exception as e:
@@ -264,39 +261,35 @@ 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 contenido del informe (como texto plano)
       - 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"
+    # Guardar el grafo a imagen PNG en un archivo temporal
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
+        tmp_path = tmp.name
     try:
-        fig.write_image(temp_img_path)
+        fig.write_image(tmp_path)
     except Exception as e:
         print("Error al guardar la imagen del grafo:", e)
-        temp_img_path = None
+        tmp_path = None
 
-    # Extraer texto plano del informe (quitando etiquetas HTML simples)
+    # Extraer texto plano del informe (remover etiquetas HTML)
     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):
+    if tmp_path and os.path.exists(tmp_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.image(tmp_path, x=10, y=20, w=190)
+        os.remove(tmp_path)
     pdf.set_font("Arial", "I", 10)
     pdf.ln(10)
     pdf.cell(0, 10, "Generado por José R. Leonett", ln=True, align="C")
@@ -306,9 +299,8 @@ def generar_pdf(report, fig):
 
 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.
+    Función para generar y retornar el PDF.
+    Si no hay datos en la tupla, retorna None.
     """
     if not analysis_tuple:
         return None
@@ -327,7 +319,6 @@ with gr.Blocks(
     gr.Markdown("**Nota:** Este analizador funciona únicamente con transacciones de Bitcoin. No se pueden analizar transacciones de Ethereum.")
     
     with gr.Row():
-        # Columna Izquierda: Grafo y controles de entrada
         with gr.Column(scale=1):
             plot_output = gr.Plot()
             tx_input = gr.Textbox(
@@ -343,8 +334,6 @@ with gr.Blocks(
                 ],
                 inputs=tx_input
             )
-            
-        # Columna Derecha: Resultados y recuadro explicativo
         with gr.Column(scale=1):
             reporte_html = gr.HTML()
             explanation_box = gr.HTML(value="""
@@ -358,27 +347,25 @@ with gr.Blocks(
                     <li><strong>Peso:</strong> Peso de la transacción en unidades de peso.</li>
                     <li><strong>Fee rate:</strong> Tarifa pagada por byte (sat/byte).</li>
                     <li><strong>Inputs:</strong> Número de entradas de la transacción.</li>
-                    <li><strong>Outputs:</strong> Número de salidas (se muestra solo si se detecta posible mixer).</li>
-                    <li><strong>Montos únicos en outputs:</strong> Número de valores únicos en las salidas (se muestra solo si se detecta mixer).</li>
-                    <li><strong>Detalles de outputs únicos:</strong> Lista de cada monto (en BTC) junto a las direcciones que lo reciben (se muestra solo si se detecta mixer).</li>
+                    <li><strong>Outputs:</strong> (Se muestra solo si se detecta posible mixer)</li>
+                    <li><strong>Montos únicos en outputs:</strong> (Solo si es mixer)</li>
+                    <li><strong>Detalles de outputs únicos:</strong> (Solo si es mixer)</li>
                     <li><strong>Total Entradas:</strong> Suma total de los valores de entrada (en BTC y USD).</li>
                     <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>Fee:</strong> Diferencia entre entradas y salidas (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.</li>
                 </ul>
             </div>
             """)
     
-    # Estado para almacenar el análisis generado (tupla: (report, fig))
+    # Estado para almacenar el análisis (tupla: (reporte, 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])
     
-    # 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)
+    # Botón de descarga (utilizamos gr.Download para que al hacer clic se inicie la descarga automáticamente)
+    download_btn = gr.Download(label="GENERAR ANALISIS EN PDF", elem_classes=["custom-btn"])
+    download_btn.click(fn=descargar_pdf, inputs=analysis_state, outputs="file")
     
 demo.launch()