Update app.py

app.py

@@ -5,6 +5,9 @@ import networkx as nx
 import numpy as np
 from operator import itemgetter
 import math  # Import the math module
+from matplotlib import animation
+from mpl_toolkits.mplot3d import Axes3D
+from streamlit.components.v1 import html
 
 # Sidebar for selecting an option
 sidebar_option = st.sidebar.radio("Select an option", 
@@ -17,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"])
+                                  "Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -25,6 +28,70 @@ def draw_graph(G, pos=None, title="Graph Visualization"):
     nx.draw(G, pos=pos, with_labels=True, node_color='lightblue', node_size=500, font_size=10, font_weight='bold')
     st.pyplot(plt)
 
+if sidebar_option == "3D Drawing: Animations of 3D Rotation":
+    st.title("3D Drawing: Animations of 3D Rotation")
+
+    # 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."
+    )
+
+    # Define the function to create animation
+    def generate_animation(G, pos, frames=100):
+        nodes = np.array([pos[v] for v in G])
+        edges = np.array([(pos[u], pos[v]) for u, v in G.edges()])
+
+        fig = plt.figure()
+        ax = fig.add_subplot(111, projection="3d")
+
+        def init():
+            ax.scatter(*nodes.T, alpha=0.2, s=100, color="blue")
+            for vizedge in edges:
+                ax.plot(*vizedge.T, color="gray")
+            ax.grid(False)
+            ax.set_axis_off()
+            plt.tight_layout()
+            return
+
+        def _frame_update(index):
+            ax.view_init(index * 0.2, index * 0.5)
+            return
+
+        ani = animation.FuncAnimation(
+            fig,
+            _frame_update,
+            init_func=init,
+            interval=50,
+            cache_frame_data=False,
+            frames=frames,
+        )
+
+        return ani
+
+    # Default Example
+    if graph_mode == "Default Example":
+        G = nx.dodecahedral_graph()
+        pos = nx.spectral_layout(G, dim=3)
+        ani = generate_animation(G, pos)
+
+    # Create Your Own Example
+    else:
+        st.write("### Customize Your Graph")
+        num_nodes = st.slider("Number of Nodes", min_value=5, max_value=50, value=20)
+        edge_prob = st.slider("Edge Probability", min_value=0.1, max_value=1.0, value=0.3)
+        
+        # Generate custom graph
+        G = nx.erdos_renyi_graph(num_nodes, edge_prob)
+        pos = nx.spectral_layout(G, dim=3)
+        ani = generate_animation(G, pos)
+
+    # Display animation in Streamlit
+    with st.spinner("Rendering animation..."):
+        ani.save("animation.gif", writer="imagemagick")
+        st.image("animation.gif", caption="3D Graph Rotation", use_column_width=True)
+
 # Function to display Weighted Graph
 def display_weighted_graph():
     st.title("Drawing: Weighted Graph")