Update app.py

app.py

@@ -7,14 +7,14 @@ import re
 import os
 import tempfile
 import base64
-from fpdf import FPDF, HTMLMixin
+from fpdf import FPDF, HTMLMixin  # Estos módulos ya no se usan en este ejemplo
 
 # ---------------- Funciones de análisis y grafo -------------------
 
 def get_transaction(tx_id):
     """
-    Intenta obtener los datos de una transacción Bitcoin usando endpoints públicos.
-    Primero se prueba con Blockstream.info y, de fallar, con mempool.space.
+    Obtiene datos de una transacción Bitcoin usando endpoints públicos.
+    Se prueba primero con Blockstream.info y, de fallar, con mempool.space.
     """
     urls = [
         f"https://blockstream.info/api/tx/{tx_id}",
@@ -70,8 +70,8 @@ def generar_grafo(tx_data):
 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 podría necesitar ajustes según la respuesta real.)
+    indica que la transacción ha sido etiquetada como MIXER.
+    (Esta función es un ejemplo y puede requerir ajustes según la respuesta real.)
     """
     url = f"https://blockchain.info/rawtx/{tx_id}?format=json"
     try:
@@ -96,9 +96,9 @@ def analizar_transaccion(tx_id):
       - Calcula totales en BTC y fee, mostrando también su equivalente en USD.
       - 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).
+      - Incorpora información de blockchain.com (etiqueta MIXER).
       - Genera un grafo interactivo.
-      - Retorna un informe en HTML, la figura del grafo y una tupla (reporte, fig) para el PDF.
+      - Retorna un informe en HTML, la figura del grafo y una tupla (reporte, fig) que se guarda en el estado.
     """
     tx_data = get_transaction(tx_id)
     if not tx_data:
@@ -113,6 +113,7 @@ 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)
@@ -246,73 +247,23 @@ def analizar_transaccion(tx_id):
             </p>
         </div>
         """
+        # Se devuelve además la tupla (reporte, fig) para almacenarla en el estado.
         return reporte, fig, (reporte, fig)
     
     except Exception as e:
         return f"⚠️ Error durante el análisis: {str(e)}", None, None
 
-# ---------------- Función para generar el PDF -------------------
+# ---------------- Función para mostrar el modal -------------------
 
-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 (como texto plano)
-      - El grafo (convertido a imagen PNG)
-      - Pie de página: "Generado por José R. Leonett"
-    Retorna el PDF como bytes.
-    """
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
-        tmp_path = tmp.name
-    try:
-        fig.write_image(tmp_path)
-    except Exception as e:
-        print("Error al guardar la imagen del grafo:", e)
-        tmp_path = None
-
-    plain_text = re.sub('<[^<]+?>', '', report)
-    
-    pdf = PDF()
-    pdf.add_page()
-    pdf.set_font("Arial", "B", 16)
-    pdf.cell(0, 10, "ANALISIS FORENSE DE BLOCKCHAIN", ln=True, align="C")
-    pdf.ln(5)
-    pdf.set_font("Arial", "", 12)
-    pdf.multi_cell(0, 10, plain_text)
-    pdf.ln(5)
-    if tmp_path and os.path.exists(tmp_path):
-        pdf.add_page()
-        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")
-    
-    pdf_bytes = pdf.output(dest="S").encode("latin1")
-    return pdf_bytes
-
-def descargar_pdf(analysis_tuple):
+def mostrar_modal(analysis_tuple):
     """
-    Genera el PDF y retorna un HTML que dispara la descarga automática.
+    Función que, a partir de la tupla de análisis, devuelve la actualización para
+    abrir el modal y el contenido HTML del informe.
     """
     if not analysis_tuple:
-        return "<p>No hay análisis generado.</p>"
-    report, fig = analysis_tuple
-    pdf_bytes = generar_pdf(report, fig)
-    pdf_b64 = base64.b64encode(pdf_bytes).decode("utf-8")
-    html_download = f'''
-    <html>
-      <body>
-         <a id="downloadLink" href="data:application/pdf;base64,{pdf_b64}" download="analisis.pdf" style="display:none;"></a>
-         <script>document.getElementById("downloadLink").click();</script>
-         <p>Si no se descarga automáticamente, <a href="data:application/pdf;base64,{pdf_b64}" download="analisis.pdf">haz clic aquí</a>.</p>
-      </body>
-    </html>
-    '''
-    return html_download
+        return gr.update(visible=False), "<p>No hay análisis generado.</p>"
+    report, _ = analysis_tuple
+    return gr.update(visible=True), report
 
 # ---------------- INTERFAZ GRÁFICA CON GRADIO -------------------
 
@@ -321,18 +272,18 @@ with gr.Blocks(
     title="🔍 Detector de Mixers / CoinJoin en Transacciones Bitcoin",
     css=".custom-btn { background-color: #ADD8E6 !important; color: black !important; padding: 5px 10px; font-size: 12px; }"
 ) as demo:
+    # Encabezado
     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)")
     gr.Markdown("**Nota:** Este analizador funciona únicamente con transacciones de Bitcoin. No se pueden analizar transacciones de Ethereum.")
     
     with gr.Row():
+        # Columna izquierda: TXID, Ejemplos y Explicación
         with gr.Column(scale=1):
-            plot_output = gr.Plot()
             tx_input = gr.Textbox(
                 label="TXID de la Transacción", 
                 placeholder="Ej: 9dd51e2d45f4f7bddcc3f0f7a05c3fd60543a11cfc9fbd0e1ca4434668cfa3e1"
             )
-            analyze_btn = gr.Button("Analizar Transacción", elem_classes=["custom-btn"])
             gr.Markdown("### Ejemplos de TXIDs válidos:")
             gr.Examples(
                 examples=[
@@ -341,11 +292,7 @@ with gr.Blocks(
                 ],
                 inputs=tx_input
             )
-        with gr.Column(scale=1):
-            reporte_html = gr.HTML()
-            pdf_download_btn = gr.Button("GENERAR ANALISIS EN PDF", elem_classes=["custom-btn"])
-            download_html = gr.HTML()
-            # Se amplió el scroll de explicación a 350px de alto para ver mejor el contenido
+            # Recuadro de explicación (scroll ampliado a 350px)
             explanation_box = gr.HTML(value="""
             <div style="overflow-y: auto; height: 350px; border: 1px solid #cccccc; padding: 10px;">
                 <h4>Explicación de los campos:</h4>
@@ -357,8 +304,8 @@ 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> Se muestra solo si se detecta posible mixer.</li>
-                    <li><strong>Montos únicos en outputs:</strong> Se muestra solo si es mixer.</li>
+                    <li><strong>Outputs:</strong> Se muestran solo si se detecta posible mixer.</li>
+                    <li><strong>Montos únicos en outputs:</strong> Se muestran solo si es mixer.</li>
                     <li><strong>Detalles de outputs únicos:</strong> Lista de cada monto (en BTC) y las direcciones asociadas (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>
@@ -368,11 +315,24 @@ with gr.Blocks(
                 </ul>
             </div>
             """)
+        # Columna derecha: Grafo y Reporte
+        with gr.Column(scale=1):
+            plot_output = gr.Plot()
+            reporte_html = gr.HTML()
+            analyze_btn = gr.Button("Analizar Transacción", elem_classes=["custom-btn"])
     
     # Estado para almacenar el análisis (tupla: (reporte, fig))
     analysis_state = gr.State()
     
+    # Al hacer clic en "Analizar Transacción", se actualizan el reporte, el grafo y se guarda el análisis en el estado.
     analyze_btn.click(fn=analizar_transaccion, inputs=tx_input, outputs=[reporte_html, plot_output, analysis_state])
-    pdf_download_btn.click(fn=descargar_pdf, inputs=analysis_state, outputs=download_html)
+    
+    # Botón para generar el análisis en un modal emergente.
+    modal_btn = gr.Button("GENERAR ANALISIS", elem_classes=["custom-btn"])
+    with gr.Modal("Resultados del Análisis", visible=False) as result_modal:
+        modal_report = gr.HTML()
+    
+    # La función mostrar_modal abrirá el modal con el contenido del análisis.
+    modal_btn.click(fn=mostrar_modal, inputs=analysis_state, outputs=[result_modal, modal_report])
     
 demo.launch()