Update app.py

app.py

@@ -6,78 +6,80 @@ from datetime import datetime
 
 def get_transaction(tx_id):
     """
-    Obtiene los datos de una transacción usando la API pública de Blockstream.
-    Retorna el JSON de la transacción si se obtiene correctamente, o None en caso de error.
+    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.
     """
-    try:
-        url = f"https://blockstream.info/api/tx/{tx_id}"
-        response = requests.get(url, timeout=10)
-        if response.status_code == 200:
-            return response.json()
-        else:
-            print(f"Error: Código de estado {response.status_code}")
-            return None
-    except Exception as e:
-        print("Excepción en get_transaction:", e)
-        return None
+    urls = [
+        f"https://blockstream.info/api/tx/{tx_id}",
+        f"https://mempool.space/api/tx/{tx_id}"
+    ]
+    for url in urls:
+        try:
+            response = requests.get(url, timeout=10)
+            if response.status_code == 200:
+                return response.json()
+            else:
+                print(f"Error en {url}: Código de estado {response.status_code}")
+        except Exception as e:
+            print(f"Excepción al consultar {url}: {e}")
+    return None
 
 def generar_grafo(tx_data):
     """
     Genera un grafo dirigido a partir de los datos de la transacción.
-    - Nodo central: la transacción.
-    - Nodos de entrada (inputs) y salida (outputs) conectados al nodo central.
+    Nodo central: la transacción.
+    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="tx")
+    G.add_node(txid, color="#FF6B6B", size=20, tipo="Transacción")
     
-    # Agregar inputs (direcciones anteriores)
+    # 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_node(addr, color="#4ECDC4", size=15, tipo="Input")
             G.add_edge(addr, txid)
     
-    # Agregar outputs (direcciones destino)
+    # 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_node(addr, color="#45B7D1", size=15, tipo="Output")
             G.add_edge(txid, addr)
     
     return G
 
 def analizar_transaccion(tx_id):
     """
-    Analiza una transacción BTC:
-      - Obtiene los datos de la transacción.
+    Analiza una transacción Bitcoin:
+      - Obtiene los datos.
       - Aplica una heurística simple para detectar posibles mixers.
-      - Genera un grafo interactivo con Plotly.
-      - Retorna un reporte en HTML junto con la figura del grafo.
+      - Genera un grafo interactivo.
+      - Retorna un reporte en HTML y la figura del grafo.
+    Nota: Este analizador solo funciona con transacciones de Bitcoin.
     """
     tx_data = get_transaction(tx_id)
     if not tx_data:
-        return "❌ Transacción no encontrada o error al obtener datos.", None
+        return ("❌ Transacción no encontrada o error al obtener datos. "
+                "Asegúrate de ingresar un TXID de Bitcoin válido."), None
     
     try:
-        # Heurística para detectar mixer
+        # Heurística para detectar mixer: muchos inputs/outputs y pocos montos únicos.
         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))
         
-        # Umbrales de detección (se pueden ajustar)
         es_mixer = num_inputs > 5 and num_outputs > 5 and montos_unicos < 3
         
-        # Generar grafo de la transacción
+        # Generar grafo
         G = generar_grafo(tx_data)
-        
-        # Calcular posición de los nodos (seed para consistencia)
         pos = nx.spring_layout(G, seed=42)
         
-        # Preparar coordenadas para las aristas
+        # Datos de aristas
         edge_x, edge_y = [], []
         for edge in G.edges():
             x0, y0 = pos[edge[0]]
@@ -85,19 +87,17 @@ def analizar_transaccion(tx_id):
             edge_x.extend([x0, x1, None])
             edge_y.extend([y0, y1, None])
         
-        # Preparar datos de los nodos
-        node_x, node_y = [], []
-        hover_texts, node_colors, node_sizes = [], [], []
+        # Datos de nodos
+        node_x, node_y, hover_texts, node_colors, node_sizes = [], [], [], [], []
         for node in G.nodes():
             x, y = pos[node]
             node_x.append(x)
             node_y.append(y)
             tipo = G.nodes[node].get("tipo", "desconocido")
-            hover_texts.append(f"Tipo: {tipo}<br>ID: {node[:6]}...")
+            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))
         
-        # Crear figura interactiva con Plotly
         fig = go.Figure(
             data=[
                 go.Scatter(
@@ -110,13 +110,13 @@ def analizar_transaccion(tx_id):
                     x=node_x, y=node_y,
                     mode="markers+text",
                     marker=dict(color=node_colors, size=node_sizes, line_width=1),
-                    text=[node[:6] for node in G.nodes()],
+                    text=[node[:8] for node in G.nodes()],
                     hovertext=hover_texts,
                     hoverinfo="text"
                 )
             ],
             layout=go.Layout(
-                title="Grafo de Transacción",
+                title="Visualización de Conexiones de la Transacción",
                 showlegend=False,
                 margin=dict(b=0, l=0, r=0, t=40),
                 xaxis=dict(showgrid=False, zeroline=False, visible=False),
@@ -125,49 +125,55 @@ def analizar_transaccion(tx_id):
             )
         )
         
-        # Formatear la fecha del bloque si está disponible
-        block_time = tx_data.get("status", {}).get("block_time")
+        status = tx_data.get("status", {})
+        block_time = status.get("block_time")
         fecha_str = datetime.fromtimestamp(block_time).strftime('%Y-%m-%d') if block_time else "Desconocida"
+        confirmed = status.get("confirmed", False)
+        estado_confirmacion = "Confirmada" if confirmed else "No confirmada"
         
-        # Reporte en HTML con estilos
         reporte = f"""
         <div style="padding: 20px; background: #1a1a1a; border-radius: 10px; color: white;">
-            <h3 style="color: {'#4CAF50' if es_mixer else '#FF5252'};">
-                {'🔮 POSIBLE MIXER' if es_mixer else '✅ TRANSACCIÓN NORMAL'}
+            <h3 style="color: {'#FF5252' if es_mixer else '#4CAF50'};">
+                {'🔮 POSIBLE MIXER' if es_mixer else '✅ Transacción Normal'}
             </h3>
+            <p><strong>TXID:</strong> {tx_id}</p>
             <p>📥 Inputs: {num_inputs}</p>
             <p>📤 Outputs: {num_outputs}</p>
             <p>💰 Montos únicos: {montos_unicos}</p>
             <p>📅 Fecha: {fecha_str}</p>
+            <p>🔔 Estado: {estado_confirmacion}</p>
         </div>
         """
         
         return reporte, fig
-        
+    
     except Exception as e:
         return f"⚠️ Error durante el análisis: {str(e)}", None
 
-# Interfaz de Gradio
-with gr.Blocks(theme=gr.themes.Soft(), title="Detector de Mixers + Grafo") as demo:
-    gr.Markdown("# 🌐 Detector de Mixers con Grafo")
-    gr.Markdown("Analiza transacciones BTC **sin API key** y visualiza conexiones.")
+with gr.Blocks(theme=gr.themes.Soft(), title="🔍 Detector de Mixers en Transacciones Bitcoin") as demo:
+    gr.Markdown("# 🔍 Detector de Mixers en Transacciones Bitcoin")
+    gr.Markdown("**Nota:** Este analizador funciona únicamente con transacciones de Bitcoin. "
+                "No se pueden analizar transacciones de Ethereum.")
     
     with gr.Row():
-        tx_input = gr.Textbox(label="ID de Transacción", placeholder="Ej: c86157c5...")
-        btn = gr.Button("Analizar 🕵️‍♂️")
-    
-    with gr.Row():
-        reporte_html = gr.HTML()
-        plot_output = gr.Plot()
+        # Columna Izquierda: Entrada y Ejemplos
+        with gr.Column(scale=1):
+            tx_input = gr.Textbox(label="TXID de la Transacción", 
+                                  placeholder="Ej: f4184fc596403b9d638783cf57adfe4c75c605f6356fbc91338530e9831e9e16")
+            btn = gr.Button("Analizar Transacción")
+            gr.Markdown("### Ejemplos de TXIDs válidos:")
+            gr.Examples(
+                examples=[
+                    ["f4184fc596403b9d638783cf57adfe4c75c605f6356fbc91338530e9831e9e16"],
+                    ["7b67ce16ad1fd3b58030ebf5a5c21d41ef245a8f9f89db46d7f7df1ac857c2c7"]
+                ],
+                inputs=tx_input
+            )
+        # Columna Derecha: Resultados
+        with gr.Column(scale=2):
+            reporte_html = gr.HTML()
+            plot_output = gr.Plot()
     
     btn.click(fn=analizar_transaccion, inputs=tx_input, outputs=[reporte_html, plot_output])
-    
-    gr.Examples(
-        examples=[
-            ["c86157c5d8f6a96d18d5a5f015a726f8e7a837d00cfe2a7dc4a68b2df671f9d7"],
-            ["f4184fc596403b9d638783cf57adfe4c75c605f6356fbc91338530e9831e9e16"]
-        ],
-        inputs=tx_input
-    )
 
 demo.launch()