Update app.py

app.py

@@ -104,15 +104,15 @@ class Visualizer:
             if d <= thresh and (neighbor == entity2 or neighbor == entity1) and len(neighbors) < num_neighbors + 1:
                 neighbors.add(entity1)
                 neighbors.add(entity2)
-                spring_tension = (thresh - d) * 10  # for vis purposes
+                spring_tension = (thresh - d) * 1000  # for vis purposes
                 G.add_edge(entity1, entity2, weight=spring_tension)
         neighbors.remove(neighbor)
         for entity1, entity2, d in tqdm(normalized_distances):
             if entity2 in neighbors and entity1 in neighbors:
-                spring_tension = thresh - d
+                spring_tension = (thresh - d) * 10
                 G.add_edge(entity1, entity2, weight=spring_tension)
 
-        pos = nx.spring_layout(G, weight="weight")  # Positions for all nodes
+        pos = nx.spring_layout(G, weight="weight", iterations=200, threshold=1e-6)  # Positions for all nodes
         edges = G.edges(data=True)
         nx.draw_networkx_nodes(G, pos, node_size=1, alpha=0.01)
         edges_connected_to_specific_node = [(u, v) for u, v in G.edges() if u == neighbor or v == neighbor]
@@ -121,7 +121,7 @@ class Visualizer:
         # nx.draw_networkx_edges(G, pos, edgelist=edges_not_connected_to_specific_node, edge_color='gray', alpha=0.1, width=1)
         for u, v, d in edges:
             if u == neighbor or v == neighbor:
-                nx.draw_networkx_edge_labels(G, pos, edge_labels={(u, v): round((thresh - (d['weight'] / 10)), 2)}, font_color="red", alpha=0.4)  # reverse modifications
+                nx.draw_networkx_edge_labels(G, pos, edge_labels={(u, v): round((thresh - (d['weight'] / 1000)) * 100, 2)}, font_color="red", alpha=0.4)  # reverse modifications
         nx.draw_networkx_labels(G, pos, font_size=14, font_family='sans-serif', font_color='green')
         nx.draw_networkx_labels(G, pos, labels={neighbor: neighbor}, font_size=14, font_family='sans-serif', font_color='red')
         plt.title(f'Graph of {distance_type}')