Update app.py

app.py

@@ -73,7 +73,14 @@ def display_spectral_embedding():
         # List all edges to allow removal
         all_edges = list(G_custom.edges())
 
-        # Add "Generate" button
+        # Collect user input for edges to remove (before showing the "Generate" button)
+        selected_edges_per_graph = []
+        for i in range(7):  # Loop over 7 graphs
+            selected_edges = st.multiselect(f"Select edges to remove for graph {i+1}:",
+                                            options=[str(edge) for edge in all_edges])
+            selected_edges_per_graph.append(selected_edges)
+
+        # Add "Generate" button after edge selection
         generate_button = st.button("Generate Graph")
 
         if generate_button:
@@ -81,12 +88,8 @@ def display_spectral_embedding():
             axs = axs.flatten()
 
             # Loop through each subplot and allow edge removal individually
-            for i in range(7):  # Looping over 7 images
-                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]
+            for i in range(7):  # Loop over 7 graphs
+                edges_to_remove = [tuple(eval(edge)) for edge in selected_edges_per_graph[i]]
 
                 # Remove the selected edges
                 G_custom_copy = G_custom.copy()