Update app.py

app.py

@@ -20,18 +20,30 @@ def draw_graph(G, pos=None, title="Graph Visualization"):
     nx.draw(G, pos=pos, with_labels=True, node_color='lightblue', node_size=500, font_size=10, font_weight='bold')
     st.pyplot(plt)
 
-# Function for creating and drawing a multipartite graph
+# Function to create a multipartite graph
 def multilayered_graph(*subset_sizes):
-    extents = nx.utils.pairwise(it.accumulate((0,) + subset_sizes))
-    layers = [range(start, end) for start, end in extents]
     G = nx.Graph()
-    for i, layer in enumerate(layers):
-        G.add_nodes_from(layer, layer=i)
-    for layer1, layer2 in nx.utils.pairwise(layers):
-        G.add_edges_from(it.product(layer1, layer2))
+    layers = len(subset_sizes)
+    node_id = 0
+
+    # Create nodes for each subset and add edges between nodes in adjacent layers
+    for i, size in enumerate(subset_sizes):
+        for j in range(size):
+            G.add_node(node_id, layer=i)  # Assign a layer attribute
+            node_id += 1
+
+    # Add edges between nodes in adjacent layers
+    node_ids = list(G.nodes())
+    for i in range(layers - 1):
+        layer_nodes = [node for node in node_ids if G.nodes[node]["layer"] == i]
+        next_layer_nodes = [node for node in node_ids if G.nodes[node]["layer"] == i + 1]
+        for node in layer_nodes:
+            for next_node in next_layer_nodes:
+                G.add_edge(node, next_node)
+
     return G
 
-# Function to display Drawing: Multipartite Layout
+# Function to display Multipartite Layout
 def display_multipartite_layout():
     st.title("Drawing: Multipartite Layout")
 
@@ -64,25 +76,27 @@ def display_multipartite_layout():
         subset_colors_input = st.text_area("Enter subset colors (comma-separated, e.g., gold,violet,green):", value="gold,violet,violet,violet,violet,limegreen,limegreen,darkorange")
         subset_colors = subset_colors_input.split(',')
 
-        # Validate input
-        if len(subset_sizes) == len(subset_colors):
-            # Generate and plot multipartite graph
-            G = multilayered_graph(*subset_sizes)
-            color = [subset_colors[data["layer"]] for v, data in G.nodes(data=True)]
-            pos = nx.multipartite_layout(G, subset_key="layer")
-            
-            plt.figure(figsize=(8, 8))
-            nx.draw(G, pos, node_color=color, with_labels=False)
-            plt.axis("equal")
-            st.pyplot(plt)
-        else:
+        # Check if the number of colors matches the number of subsets
+        if len(subset_sizes) != len(subset_colors):
             st.error("The number of colors should match the number of subsets.")
+        else:
+            # Add a button to generate the graph
+            if st.button("Generate Graph"):
+                # Generate and plot multipartite graph
+                G = multilayered_graph(*subset_sizes)
+                color = [subset_colors[data["layer"]] for v, data in G.nodes(data=True)]
+                pos = nx.multipartite_layout(G, subset_key="layer")
+                
+                plt.figure(figsize=(8, 8))
+                nx.draw(G, pos, node_color=color, with_labels=False)
+                plt.axis("equal")
+                st.pyplot(plt)
 
 # Display Drawing: Multipartite Layout if selected
 if sidebar_option == "Drawing: Multipartite Layout":
     display_multipartite_layout()
 
-# Function to display Drawing: Labels And Colors
+# Function to display Labels and Colors
 def display_labels_and_colors():
     st.title("Drawing: Labels And Colors")
 
@@ -135,29 +149,31 @@ def display_labels_and_colors():
         edges = st.text_area("Enter edges (format: node1-node2, comma-separated, e.g., a-b,b-c):", value="a-b,b-c,c-d")
         edge_list = [tuple(edge.split('-')) for edge in edges.split(',')]
 
-        # Generate graph based on user input
-        G_custom = nx.Graph()
-        G_custom.add_nodes_from(node_labels)
-        G_custom.add_edges_from(edge_list)
-
         # Let user choose colors for nodes and edges
         node_color = st.color_picker("Pick a color for nodes:", "#FF6347")
         edge_color = st.color_picker("Pick a color for edges:", "#4682B4")
-        
-        # Generate layout for the nodes
-        pos_custom = nx.spring_layout(G_custom)
 
-        # Draw the graph
-        nx.draw_networkx_nodes(G_custom, pos_custom, node_color=node_color, node_size=800, edgecolors="gray", alpha=0.9)
-        nx.draw_networkx_edges(G_custom, pos_custom, edge_color=edge_color, width=2, alpha=0.7)
-        
-        # Create custom labels
-        custom_labels = {node: f"${node}$" for node in node_labels}
-        nx.draw_networkx_labels(G_custom, pos_custom, labels=custom_labels, font_size=22, font_color="whitesmoke")
+        # Add a button to generate the graph
+        if st.button("Generate Graph"):
+            # Generate graph based on user input
+            G_custom = nx.Graph()
+            G_custom.add_nodes_from(node_labels)
+            G_custom.add_edges_from(edge_list)
 
-        plt.tight_layout()
-        plt.axis("off")
-        st.pyplot(plt)
+            # Generate layout for the nodes
+            pos_custom = nx.spring_layout(G_custom)
+
+            # Draw the graph
+            nx.draw_networkx_nodes(G_custom, pos_custom, node_color=node_color, node_size=800, edgecolors="gray", alpha=0.9)
+            nx.draw_networkx_edges(G_custom, pos_custom, edge_color=edge_color, width=2, alpha=0.7)
+
+            # Create custom labels
+            custom_labels = {node: f"${node}$" for node in node_labels}
+            nx.draw_networkx_labels(G_custom, pos_custom, labels=custom_labels, font_size=22, font_color="whitesmoke")
+
+            plt.tight_layout()
+            plt.axis("off")
+            st.pyplot(plt)
 
 # Display Drawing: Labels And Colors if selected
 if sidebar_option == "Drawing: Labels And Colors":