Update app.py

app.py

@@ -66,27 +66,31 @@ def display_spectral_embedding():
         G_custom = nx.grid_2d_graph(grid_size, grid_size)
 
         # List all edges to allow removal
-        all_edges = list(G_custom.edges)
+        all_edges = list(G_custom.edges())
 
-        fig, axs = plt.subplots(3, 3, figsize=(12, 12))
-        axs = axs.flatten()
+        # Add "Generate" button
+        generate_button = st.button("Generate Graph")
 
-        # Loop through each subplot and allow edge removal individually
-        for i in range(7):
-            selected_edges = st.multiselect(f"Select edges to remove for graph {i+1}:",
-                                            options=[str(edge) for edge in all_edges])
+        if generate_button:
+            fig, axs = plt.subplots(3, 3, figsize=(12, 12))
+            axs = axs.flatten()
 
-            # Convert the selected edges from string to tuple
-            edges_to_remove = [tuple(eval(edge)) for edge in selected_edges]
+            # Loop through each subplot and allow edge removal individually
+            for i in range(7):
+                selected_edges = st.multiselect(f"Select edges to remove for graph {i+1}:",
+                                                options=[str(edge) for edge in all_edges])
 
-            # Remove the selected edges
-            G_custom_copy = G_custom.copy()
-            G_custom_copy.remove_edges_from(edges_to_remove)
+                # Convert the selected edges from string to tuple
+                edges_to_remove = [tuple(eval(edge)) for edge in selected_edges]
 
-            # Draw the graph with removed edges
-            nx.draw_spectral(G_custom_copy, **{"node_color": "C0", "node_size": 100}, ax=axs[i])
+                # Remove the selected edges
+                G_custom_copy = G_custom.copy()
+                G_custom_copy.remove_edges_from(edges_to_remove)
 
-        st.pyplot(fig)
+                # Draw the graph with removed edges
+                nx.draw_spectral(G_custom_copy, **{"node_color": "C0", "node_size": 100}, ax=axs[i])
+
+            st.pyplot(fig)
 
 # Display Drawing: Spectral Embedding if selected
 if sidebar_option == "Drawing: Spectral Embedding":