Update app.py

app.py

@@ -157,6 +157,53 @@ def display_cluster_layout():
         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
 
+        # Input for the number of nodes
+        num_nodes = st.number_input("Number of nodes:", min_value=2, max_value=100, value=5)
+        
+        # Input for edges: users can input edges manually
+        edge_input = st.text_area("Enter edges (format: u,v):", value="0,1\n1,2\n2,3")
+        
+        # Button to generate the graph
+        if st.button("Generate Graph"):
+            edges = []
+            if edge_input:
+                # Parse the input string to create the edges
+                edge_list = edge_input.split("\n")
+                for edge in edge_list:
+                    try:
+                        u, v = map(int, edge.split(","))
+                        edges.append((u, v))
+                    except ValueError:
+                        st.error("Invalid edge format. Please use the format 'u,v'.")
+            
+            # Create a graph and add the edges
+            G_custom = nx.Graph()
+            G_custom.add_edges_from(edges)
+
+            if len(G_custom.nodes) == 0:
+                st.error("No nodes found. Please ensure you have specified edges correctly.")
+            else:
+                # Compute clusters (communities) and layout
+                communities = list(nx.community.greedy_modularity_communities(G_custom))
+
+                # 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_custom, 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_custom, 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_custom, pos=pos, nodelist=nodes, node_color=clr, node_size=100)
+                nx.draw_networkx_edges(G_custom, pos=pos)
+
+                plt.tight_layout()
+                st.pyplot(plt)
+
 # Display Basic: Properties if selected
 if sidebar_option == "Basic: Properties":
     st.title("Basic: Properties")