Update app.py

app.py

@@ -36,7 +36,8 @@ def display_spectral_embedding():
 
         fig, axs = plt.subplots(3, 3, figsize=(12, 12))
         axs = axs.flatten()
-        for i in range(7):
+
+        for i in range(7):  # Looping over 7 images
             if i == 0:
                 nx.draw_spectral(G, **options, ax=axs[i])
             elif i == 1:
@@ -58,6 +59,10 @@ def display_spectral_embedding():
                 G.remove_edge((4, 2), (4, 3))
                 nx.draw_spectral(G, **options, ax=axs[i])
 
+        # Hide the last two subplots (8th and 9th)
+        for j in range(7, 9):
+            fig.delaxes(axs[j])  # Delete the extra axes
+
         st.pyplot(fig)
 
     elif option == "Create your own":
@@ -76,7 +81,7 @@ def display_spectral_embedding():
             axs = axs.flatten()
 
             # Loop through each subplot and allow edge removal individually
-            for i in range(7):
+            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])
 
@@ -90,6 +95,10 @@ def display_spectral_embedding():
                 # Draw the graph with removed edges
                 nx.draw_spectral(G_custom_copy, **{"node_color": "C0", "node_size": 100}, ax=axs[i])
 
+            # Hide the last two subplots (8th and 9th)
+            for j in range(7, 9):
+                fig.delaxes(axs[j])  # Delete the extra axes
+
             st.pyplot(fig)
 
 # Display Drawing: Spectral Embedding if selected