Upload app.py

app.py

@@ -332,31 +332,42 @@ def build_entity_to_entity_graph(entities_by_doc: List[List[Tuple[str, str]]]) -
                 pair = tuple(sorted([unique_entities[i], unique_entities[j]], key=str))
                 cooccurrence[pair] += 1
     
-    # Construir nós (entidades com >= 2 docs)
+    # Construir nós (entidades com >= 2 docs, limitado a top 50)
     nodes = []
     entity_to_id = {}
-    node_count = sum(1 for e, c in entity_doc_count.items() if c >= 2)
+    
+    # Filtrar e ordenar por frequência
+    valid_entities = [(e, c) for e, c in entity_doc_count.items() if c >= 2]
+    valid_entities.sort(key=lambda x: x[1], reverse=True)
+    valid_entities = valid_entities[:50]  # Limitar a 50 nós principais
+    
+    node_count = len(valid_entities)
     
     node_idx = 0
-    for entity, count in sorted(entity_doc_count.items(), key=lambda x: x[1], reverse=True):
-        if count >= 2:
-            entity_to_id[entity] = node_idx
-            
-            # Posição em espiral 3D (melhor distribuição)
-            angle = 2 * np.pi * node_idx / max(node_count, 1) * 3  # 3 voltas na espiral
-            radius = 1.5 + (node_idx / max(node_count, 1)) * 2  # Raio aumenta
-            z_pos = (node_idx / max(node_count, 1) - 0.5) * 3  # Altura varia
-            
-            nodes.append({
-                "id": node_idx,
-                "entity": entity[0],
-                "type": entity[1],
-                "docs": count,
-                "x": float(radius * np.cos(angle)),
-                "y": float(radius * np.sin(angle)),
-                "z": float(z_pos)
-            })
-            node_idx += 1
+    for entity, count in valid_entities:
+        entity_to_id[entity] = node_idx
+        
+        # Layout em esfera 3D (melhor para muitos nós)
+        # Golden angle para distribuição uniforme
+        golden_angle = np.pi * (3 - np.sqrt(5))
+        theta = node_idx * golden_angle
+        phi = np.arccos(1 - 2 * (node_idx + 0.5) / max(node_count, 1))
+        
+        radius = 3.0  # Raio fixo maior
+        x = radius * np.sin(phi) * np.cos(theta)
+        y = radius * np.sin(phi) * np.sin(theta)
+        z = radius * np.cos(phi)
+        
+        nodes.append({
+            "id": node_idx,
+            "entity": entity[0],
+            "type": entity[1],
+            "docs": count,
+            "x": float(x),
+            "y": float(y),
+            "z": float(z)
+        })
+        node_idx += 1
     
     # Construir arestas (co-ocorrências)
     entity_edges = []