Update graph

app.py

@@ -15,16 +15,18 @@ from network import analysis
 with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
   # with gr.Tab("📝 About"):
   #     # gr.Markdown(ABOUT_TEXT)
+
+  with gr.Row():
+    with gr.Column():
+      gr.Label("👪 Visualize witches family connections", show_label=False)
+      gr.Markdown("You can see the various parent-child-sibling-spouse relationships of the accused witches!")
+      gr.HTML(analysis)
+  
   with gr.Row():
     with gr.Column():
-      gr.Label("📊 Visually explore witches family data")
+      gr.Label("📊 Visually explore witches family data", show_label=False)
       gr.Markdown("You can use drag-and-drop operations to explore the data, start your analysis now! In the given query, we see where the accused witches reside after filtering on the gender label (here, Female) and we show a breakdown of the class!")
       pyg_app = get_html_on_gradio(df, spec="./config.json")
       gr.HTML(pyg_app)
-  with gr.Row():
-    with gr.Column():
-      gr.Label("👪 Visualize witches family connections")
-      gr.Markdown("You can see the various parent-child-sibling-spouse relationships of the accused witches!")
-      gr.HTML(analysis)
-            
+
 demo.launch(app_kwargs={"routes": [PYGWALKER_ROUTE]}, share=True).queue()

network.py

@@ -9,19 +9,33 @@ from data import df
 def analysis():
     G = nx.DiGraph()
 
-    # Define the columns to consider for relationships
     relationship_columns = ['father', 'sibling', 'spouse', 'mother', 'child']
 
-    G.add_nodes_from(df["itemLabel"])
-
+    # G.add_nodes_from(df["itemLabel"])
+    inf_labels = ["residence", "occupation", "class"]
     for index, row in df.iterrows():
-        main_entity = row['itemLabel']
-        for relationship in relationship_columns:
-            if pd.notna(row[relationship]):
-                G.add_edge(main_entity, row[relationship], relationship=str(relationship), label=relationship)
+        if row["itemLabel"] not in G.nodes():
+            G.add_node(row["itemLabel"])
+            information = ""
+            for i in inf_labels:
+                if pd.notna(row[i]):
+                    information+=f"{i[:-5]}: {row[i]}\n"
+            G.nodes[row["itemLabel"]]["attr"]=information
+
+        for index, row in df.iterrows():
+            main_entity = row['itemLabel']
+            for relationship in relationship_columns:
+                if pd.notna(row[relationship]) and not G.has_edge(main_entity, row[relationship]) and not G.has_edge(row[relationship], main_entity):
+                    # if pd.notna(row[relationship]):
+
+                    G.add_edge(row[relationship], main_entity, label=str(relationship)+ "_of")
 
-    plt.figure(figsize=(20, 10))  # Set the size of the plot
-    # pos = nx.kamada_kawai_layout(G)  # Compute the positions of the nodes
+    for x in G.nodes():
+        if "attr" not in  G.nodes[x]:
+            G.nodes[x]["attr"]=""
+
+    # plt.figure(figsize=(50, 20))  # Set the size of the plot
+    pos = nx.kamada_kawai_layout(G)  # Compute the positions of the nodes
 
     # # Draw the nodes and edges with labels
     # nx.draw(G, pos, with_labels=True, node_size=20, node_color='skyblue', font_size=10, font_color='black', font_weight='bold', alpha=0.7)
@@ -35,27 +49,28 @@ def analysis():
     # plt.show()
 
 
-    net = Network(width="1200px", height="800px", bgcolor="#222222", font_color="white", directed=True)
+    net = Network(width="1500px", height="1000px", bgcolor="#FFFFFF", font_color="black", directed=True)
     net.from_nx(G)
 
     for node in net.nodes:
-        node["title"] = node["id"]
+        node["title"] = G.nodes[node["id"]]["attr"]
         node["value"] = len(G[node["id"]])
-
+        node["font"]={"size": 20}
 
     for edge in net.edges:
         # Set edge title to the relationship from your graph
-        relationship = G[edge["from"]][edge["to"]].get("relationship", "Unknown")
+        relationship = G[edge["from"]][edge["to"]].get("label", "Unknown")
         edge["title"] = relationship  # This will show as a tooltip
         edge["color"] = "blue"
         edge["width"] = 2 if relationship != "Unknown" else 1
 
+
     html = net.generate_html()
     #need to remove ' from HTML
     html = html.replace("'", "\"")
     
-    return f"""<iframe style="width: 100%; height: 800px;margin:0 auto" name="result" allow="midi; geolocation; microphone; camera; 
+    return f"""<iframe style="width: 100%; height: 800px; margin:0 auto" name="result" allow="midi; geolocation; microphone; camera; 
     display-capture; encrypted-media;" sandbox="allow-modals allow-forms 
     allow-scripts allow-same-origin allow-popups 
     allow-top-navigation-by-user-activation allow-downloads" allowfullscreen="" 
-    allowpaymentrequest="" frameborder="0" srcdoc='{html}'></iframe>"""
+    allowpaymentrequest="" frameBorder="0" srcdoc='{html}'></iframe>"""

requirements.txt

@@ -4,4 +4,5 @@ gradio
 pygwalker
 datasets
 pyvis
-networkx
+networkx
+scipy