Upload app.py

app.py

@@ -367,17 +367,53 @@ def build_entity_to_entity_graph(entities_by_doc: List[List[Tuple[str, str]]]) -
                 "target": entity_to_id[ent2],
                 "weight": weight,
                 "source_entity": ent1[0],
-                "target_entity": ent2[0]
+                "target_entity": ent2[0],
+                "reason": f"Aparecem juntos em {weight} documento(s)"
             })
     
     # Ordenar arestas por peso
     entity_edges.sort(key=lambda x: x["weight"], reverse=True)
     
+    # Calcular métricas de grafo
+    # Degree centrality (número de conexões de cada nó)
+    degree = defaultdict(int)
+    for edge in entity_edges:
+        degree[edge["source"]] += edge["weight"]
+        degree[edge["target"]] += edge["weight"]
+    
+    # Calcular max degree para normalização
+    max_degree = max(degree.values()) if degree else 1
+    
+    # Atualizar nós com métricas
+    hubs = []
+    for node in nodes:
+        node_degree = degree.get(node["id"], 0)
+        node["degree"] = node_degree
+        node["centrality"] = round(node_degree / max_degree, 3)
+        
+        # Classificar nó
+        if node["centrality"] > 0.7:
+            node["role"] = "hub"  # Hub central
+            hubs.append(node)
+        elif node["centrality"] > 0.3:
+            node["role"] = "connector"  # Conector
+        else:
+            node["role"] = "peripheral"  # Periférico
+    
+    # Top hubs
+    hubs.sort(key=lambda x: x["degree"], reverse=True)
+    
     return {
         "nodes": nodes,
         "edges": entity_edges[:200],  # Limitar a 200 arestas mais fortes
         "node_count": len(nodes),
-        "edge_count": len(entity_edges)
+        "edge_count": len(entity_edges),
+        "hubs": [{"entity": h["entity"], "type": h["type"], "degree": h["degree"]} for h in hubs[:5]],
+        "insights": {
+            "total_connections": sum(degree.values()) // 2,
+            "avg_degree": round(sum(degree.values()) / len(degree), 1) if degree else 0,
+            "hub_count": len(hubs)
+        }
     }