added sim

pages/2_Concepts_comparison.py

@@ -105,26 +105,25 @@ with output_col_1:
                 print(np.round(similarity[0][0], 3))
                 edges.append((concept_ids[i], concept_ids[j], np.round(similarity[0][0], 3)))
 
-    # Create an empty graph
-    G = nx.Graph()
-
-    # Add edges with weights to the graph
-    for edge in edges:
-        node1, node2, weight = edge
-        G.add_edge(node1, node2, weight=weight)
-
-
-    # Initiate PyVis network object
-    net = Network(
-                       height='400px',
-                       width='100%',
-                       bgcolor='#222222',
-                       font_color='white'
-                      )
-
-    # Take Networkx graph and translate it to a PyVis graph format
-    net.from_nx(G)
-
+    # # Create an empty graph
+    # G = nx.Graph()
+
+    # # Add edges with weights to the graph
+    # for edge in edges:
+    #     node1, node2, weight = edge
+    #     G.add_edge(node1, node2, weight=weight)
+
+    net = Network(height="750px", width="100%",)
+    for e in edges:
+        src = e[0]
+        dst = e[1]
+        w = e[2]
+
+        net.add_node(src, src, title=src)
+        net.add_node(dst, dst, title=dst)
+        net.add_edge(src, dst, value=w, title=src + ' to ' + dst + ' similarity ' +str(w))
+    print(net)
+    
     # Generate network with specific layout settings
     net.repulsion(
                         node_distance=420,