Update app.py

app.py

@@ -119,7 +119,7 @@ def default_example():
         ax.grid(False)
         # Suppress tick labels
         for dim in (ax.xaxis, ax.yaxis, ax.zaxis):
-            dim.set_ticks([])
+            dim.set_ticks([]) 
         # Set axes labels
         ax.set_xlabel("x")
         ax.set_ylabel("y")
@@ -135,37 +135,41 @@ def create_own_graph():
     nodes = st.number_input("Number of nodes", min_value=2, max_value=50, value=20)
     seed = st.number_input("Seed for layout", value=779)
     
-    # Generate graph and layout
-    G = nx.cycle_graph(nodes)
-    pos = nx.spring_layout(G, dim=3, seed=seed)
-    
-    # Extract node and edge positions
-    node_xyz = np.array([pos[v] for v in sorted(G)])
-    edge_xyz = np.array([(pos[u], pos[v]) for u, v in G.edges()])
+    # Add a button to generate the graph
+    generate_button = st.button("Generate Graph")
 
-    # Create the 3D figure
-    fig = plt.figure()
-    ax = fig.add_subplot(111, projection="3d")
+    if generate_button:
+        # Generate graph and layout
+        G = nx.cycle_graph(nodes)
+        pos = nx.spring_layout(G, dim=3, seed=seed)
+        
+        # Extract node and edge positions
+        node_xyz = np.array([pos[v] for v in sorted(G)])
+        edge_xyz = np.array([(pos[u], pos[v]) for u, v in G.edges()])
 
-    # Plot the nodes
-    ax.scatter(*node_xyz.T, s=100, ec="w")
+        # Create the 3D figure
+        fig = plt.figure()
+        ax = fig.add_subplot(111, projection="3d")
 
-    # Plot the edges
-    for vizedge in edge_xyz:
-        ax.plot(*vizedge.T, color="tab:gray")
+        # Plot the nodes
+        ax.scatter(*node_xyz.T, s=100, ec="w")
 
-    def _format_axes(ax):
-        """Visualization options for the 3D axes."""
-        ax.grid(False)
-        for dim in (ax.xaxis, ax.yaxis, ax.zaxis):
-            dim.set_ticks([])
-        ax.set_xlabel("x")
-        ax.set_ylabel("y")
-        ax.set_zlabel("z")
+        # Plot the edges
+        for vizedge in edge_xyz:
+            ax.plot(*vizedge.T, color="tab:gray")
 
-    _format_axes(ax)
-    fig.tight_layout()
-    st.pyplot(fig)
+        def _format_axes(ax):
+            """Visualization options for the 3D axes."""
+            ax.grid(False)
+            for dim in (ax.xaxis, ax.yaxis, ax.zaxis):
+                dim.set_ticks([])
+            ax.set_xlabel("x")
+            ax.set_ylabel("y")
+            ax.set_zlabel("z")
+
+        _format_axes(ax)
+        fig.tight_layout()
+        st.pyplot(fig)
 
 if sidebar_option == "3D Drawing: Basic Matplotlib":
     st.title("3D Drawing: Basic Matplotlib")
@@ -174,7 +178,7 @@ if sidebar_option == "3D Drawing: Basic Matplotlib":
     graph_mode = st.radio(
         "Choose a Mode:",
         ("Default Example", "Create Your Own"),
-        help="Default example shows a dodecahedral graph, or you can create your own custom graph."
+        help="Default example shows a cycle graph, or you can create your own custom graph."
     )
 
     # Display the chosen option