Update app.py

app.py

@@ -20,7 +20,7 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Drawing: Multipartite Layout", "Drawing: Node Colormap",
                                   "Drawing: Rainbow Coloring", "Drawing: Random Geometric Graph","Drawing: Self-loops",
                                   "Drawing: Simple Path", "Drawing: Spectral Embedding", "Drawing: Traveling Salesman Problem",
-                                  "Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation"])
+                                  "Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation", "3D Drawing: Basic Matplotlib"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -92,7 +92,96 @@ if sidebar_option == "3D Drawing: Animations of 3D Rotation":
         ani.save("animation.gif", writer="imagemagick")
         st.image("animation.gif", caption="3D Graph Rotation", use_container_width=True)
 
+# Default example code
+def default_example():
+    G = nx.cycle_graph(20)
+
+    # 3d spring layout
+    pos = nx.spring_layout(G, dim=3, seed=779)
+    # Extract node and edge positions from the layout
+    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()])
+
+    # Create the 3D figure
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection="3d")
+
+    # Plot the nodes - alpha is scaled by "depth" automatically
+    ax.scatter(*node_xyz.T, s=100, ec="w")
+
+    # Plot the edges
+    for vizedge in edge_xyz:
+        ax.plot(*vizedge.T, color="tab:gray")
+
+    def _format_axes(ax):
+        """Visualization options for the 3D axes."""
+        # Turn gridlines off
+        ax.grid(False)
+        # Suppress tick labels
+        for dim in (ax.xaxis, ax.yaxis, ax.zaxis):
+            dim.set_ticks([])
+        # Set axes labels
+        ax.set_xlabel("x")
+        ax.set_ylabel("y")
+        ax.set_zlabel("z")
+
+    _format_axes(ax)
+    fig.tight_layout()
+    st.pyplot(fig)
+
+# Create your own graph option
+def create_own_graph():
+    # Input fields to customize the 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()])
+
+    # Create the 3D figure
+    fig = plt.figure()
+    ax = fig.add_subplot(111, projection="3d")
+
+    # Plot the nodes
+    ax.scatter(*node_xyz.T, s=100, ec="w")
+
+    # Plot the edges
+    for vizedge in edge_xyz:
+        ax.plot(*vizedge.T, color="tab:gray")
+
+    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")
 
+    # Provide options for Default Example or Custom Graph
+    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."
+    )
+
+    # Display the chosen option
+    if graph_mode == "Default Example":
+        default_example()
+    elif graph_mode == "Create Your Own":
+        create_own_graph()
 
 # Function to display Weighted Graph
 def display_weighted_graph():