Update app.py

app.py

@@ -7,7 +7,8 @@ import numpy as np
 sidebar_option = st.sidebar.radio("Select an option", 
                                  ["Select an option", "Basic: Properties", 
                                   "Basic: Read and write graphs", "Basic: Simple graph", 
-                                  "Basic: Simple graph Directed", "Drawing: Custom Node Position"])
+                                  "Basic: Simple graph Directed", "Drawing: Custom Node Position",
+                                  "Drawing: Cluster Layout"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -124,6 +125,38 @@ def display_custom_node_position():
     # Draw the graph
     draw_graph(G, pos)
 
+# Function to display Cluster Layout for Drawing: Cluster Layout
+def display_cluster_layout():
+    st.title("Drawing: Cluster Layout")
+    
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        G = nx.davis_southern_women_graph()  # Example graph
+        communities = nx.community.greedy_modularity_communities(G)
+
+        # Compute positions for the node clusters as if they were themselves nodes in a supergraph using a larger scale factor
+        supergraph = nx.cycle_graph(len(communities))
+        superpos = nx.spring_layout(G, scale=50, seed=429)
+
+        # Use the "supernode" positions as the center of each node cluster
+        centers = list(superpos.values())
+        pos = {}
+        for center, comm in zip(centers, communities):
+            pos.update(nx.spring_layout(nx.subgraph(G, comm), center=center, seed=1430))
+
+        # Nodes colored by cluster
+        for nodes, clr in zip(communities, ("tab:blue", "tab:orange", "tab:green")):
+            nx.draw_networkx_nodes(G, pos=pos, nodelist=nodes, node_color=clr, node_size=100)
+        nx.draw_networkx_edges(G, pos=pos)
+
+        plt.tight_layout()
+        st.pyplot(plt)
+
+    elif option == "Create your own":
+        st.write("For custom graph creation, you can specify the number of nodes and edges.")
+        # Here you can add functionality for custom graph creation if required
+
 # Display Basic: Properties if selected
 if sidebar_option == "Basic: Properties":
     st.title("Basic: Properties")
@@ -228,3 +261,7 @@ elif sidebar_option == "Basic: Simple graph Directed":
 # Display Drawing: Custom Node Position if selected
 elif sidebar_option == "Drawing: Custom Node Position":
     display_custom_node_position()
+
+# Display Drawing: Cluster Layout if selected
+elif sidebar_option == "Drawing: Cluster Layout":
+    display_cluster_layout()