Update app.py

app.py

@@ -65,19 +65,26 @@ def display_spectral_embedding():
         grid_size = st.slider("Choose grid size (n x n):", min_value=3, max_value=10, value=6)
         G_custom = nx.grid_2d_graph(grid_size, grid_size)
 
-        # Edge removal logic
-        removed_edges = st.multiselect("Choose edges to remove (e.g., (0, 0) to (0, 1)):", list(G_custom.edges))
+        # List all edges to allow removal
+        all_edges = list(G_custom.edges)
 
-        # Remove selected edges
-        G_custom.remove_edges_from(removed_edges)
-
-        options = {"node_color": "C0", "node_size": 100}
         fig, axs = plt.subplots(3, 3, figsize=(12, 12))
         axs = axs.flatten()
 
-        # Visualize the modified graph
+        # Loop through each subplot and allow edge removal individually
         for i in range(7):
-            nx.draw_spectral(G_custom, **options, ax=axs[i])
+            selected_edges = st.multiselect(f"Select edges to remove for graph {i+1}:",
+                                            options=[str(edge) for edge in all_edges])
+
+            # Convert the selected edges from string to tuple
+            edges_to_remove = [tuple(eval(edge)) for edge in selected_edges]
+
+            # Remove the selected edges
+            G_custom_copy = G_custom.copy()
+            G_custom_copy.remove_edges_from(edges_to_remove)
+
+            # Draw the graph with removed edges
+            nx.draw_spectral(G_custom_copy, **{"node_color": "C0", "node_size": 100}, ax=axs[i])
 
         st.pyplot(fig)