Update app.py

app.py

@@ -21,7 +21,7 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Drawing: Rainbow Coloring", "Drawing: Random Geometric Graph","Drawing: Self-loops",
                                   "Drawing: Simple Path", "Drawing: Spectral Embedding", "Drawing: Traveling Salesman Problem",
                                   "Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation", "3D Drawing: Basic Matplotlib",
-                                  "Graph: DAG - Topological Layout", "Graph: Erdos Renyi", "Graph: Karate Club"])
+                                  "Graph: DAG - Topological Layout", "Graph: Erdos Renyi", "Graph: Karate Club", "Graph: Minimum Spanning Tree"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -29,6 +29,89 @@ 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)
 
+def minimum_spanning_tree_graph():
+    st.title("Graph: Minimum Spanning Tree")
+
+    # Sidebar selection for Default Example or Custom Graph
+    graph_mode = st.radio(
+        "Choose a Mode:",
+        ("Default Example", "Create Your Own"),
+        help="Default example shows a graph and its minimum spanning tree, or you can create your own graph."
+    )
+
+    if graph_mode == "Default Example":
+        # Create a default graph
+        G = nx.Graph()
+        G.add_edges_from(
+            [
+                (0, 1, {"weight": 4}),
+                (0, 7, {"weight": 8}),
+                (1, 7, {"weight": 11}),
+                (1, 2, {"weight": 8}),
+                (2, 8, {"weight": 2}),
+                (2, 5, {"weight": 4}),
+                (2, 3, {"weight": 7}),
+                (3, 4, {"weight": 9}),
+                (3, 5, {"weight": 14}),
+                (4, 5, {"weight": 10}),
+                (5, 6, {"weight": 2}),
+                (6, 8, {"weight": 6}),
+                (7, 8, {"weight": 7}),
+            ]
+        )
+
+        # Find the minimum spanning tree
+        T = nx.minimum_spanning_tree(G)
+
+        # Visualize the graph and the minimum spanning tree
+        pos = nx.spring_layout(G)
+        fig, ax = plt.subplots(figsize=(8, 8))
+        nx.draw_networkx_nodes(G, pos, node_color="lightblue", node_size=500, ax=ax)
+        nx.draw_networkx_edges(G, pos, edge_color="grey", ax=ax)
+        nx.draw_networkx_labels(G, pos, font_size=12, font_family="sans-serif", ax=ax)
+        nx.draw_networkx_edge_labels(
+            G, pos, edge_labels={(u, v): d["weight"] for u, v, d in G.edges(data=True)}, ax=ax
+        )
+        nx.draw_networkx_edges(T, pos, edge_color="green", width=2, ax=ax)
+        ax.set_title("Graph and Minimum Spanning Tree")
+        plt.axis("off")
+        st.pyplot(fig)
+
+    elif graph_mode == "Create Your Own":
+        st.write("### Create Your Own Graph")
+
+        # Allow user to input the number of nodes and edges for custom graph
+        num_nodes = st.number_input("Number of nodes", min_value=2, value=5)
+        num_edges = st.number_input("Number of edges", min_value=1, value=6)
+
+        # Create random graph with user input
+        G = nx.gnm_random_graph(num_nodes, num_edges)
+
+        # Add random weights to the edges
+        for u, v in G.edges():
+            G[u][v]["weight"] = st.number_input(f"Weight for edge ({u}, {v})", value=1)
+
+        # Find the minimum spanning tree
+        T = nx.minimum_spanning_tree(G)
+
+        # Visualize the graph and the minimum spanning tree
+        pos = nx.spring_layout(G)
+        fig, ax = plt.subplots(figsize=(8, 8))
+        nx.draw_networkx_nodes(G, pos, node_color="lightblue", node_size=500, ax=ax)
+        nx.draw_networkx_edges(G, pos, edge_color="grey", ax=ax)
+        nx.draw_networkx_labels(G, pos, font_size=12, font_family="sans-serif", ax=ax)
+        nx.draw_networkx_edge_labels(
+            G, pos, edge_labels={(u, v): d["weight"] for u, v, d in G.edges(data=True)}, ax=ax
+        )
+        nx.draw_networkx_edges(T, pos, edge_color="green", width=2, ax=ax)
+        ax.set_title("Custom Graph and Minimum Spanning Tree")
+        plt.axis("off")
+        st.pyplot(fig)
+
+# Display the corresponding page based on sidebar option
+if sidebar_option == "Graph: Minimum Spanning Tree":
+    minimum_spanning_tree_graph()
+
 def karate_club_graph():
     st.title("Graph: Karate Club")