Update app.py

app.py

@@ -12,7 +12,7 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Basic: Simple graph Directed", "Drawing: Custom Node Position",
                                   "Drawing: Cluster Layout", "Drawing: Degree Analysis",
                                   "Drawing: Ego Graph", "Drawing: Eigenvalues", "Drawing: Four Grids",
-                                  "Drawing: House With Colors", "Drawing: Labels And Colors"])
+                                  "Drawing: House With Colors", "Drawing: Labels And Colors", "Drawing: Multipartite Layout"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -20,6 +20,68 @@ 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
+def multilayered_graph(*subset_sizes):
+    extents = nx.utils.pairwise(itertools.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(itertools.product(layer1, layer2))
+    return G
+
+# Function to display Drawing: Multipartite Layout
+def display_multipartite_layout():
+    st.title("Drawing: Multipartite Layout")
+
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        subset_sizes = [5, 5, 4, 3, 2, 4, 4, 3]
+        subset_color = [
+            "gold", "violet", "violet", "violet", "violet", 
+            "limegreen", "limegreen", "darkorange"
+        ]
+        
+        # Generate and plot multipartite graph
+        G = multilayered_graph(*subset_sizes)
+        color = [subset_color[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)
+
+    elif option == "Create your own":
+        # Let the user input the subset sizes and colors
+        st.write("Enter the subset sizes and colors to create your own multipartite graph.")
+
+        subset_sizes_input = st.text_area("Enter subset sizes (comma-separated, e.g., 5,5,4,3):", value="5,5,4,3,2,4,4,3")
+        subset_sizes = list(map(int, subset_sizes_input.split(',')))
+
+        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:
+            st.error("The number of colors should match the number of subsets.")
+
+# Display Drawing: Multipartite Layout if selected
+if sidebar_option == "Drawing: Multipartite Layout":
+    display_multipartite_layout()
+
 # Function to display Drawing: Labels And Colors
 def display_labels_and_colors():
     st.title("Drawing: Labels And Colors")