Update app.py

app.py

@@ -12,7 +12,9 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Basic: Simple graph Directed", "Drawing: Custom Node Position",
                                   "Drawing: Cluster Layout", "Drawing: Degree Analysis",
                                   "Drawing: Ego Graph", "Drawing: Eigenvalues", "Drawing: Four Grids",
-                                  "Drawing: House With Colors", "Drawing: Labels And Colors", "Drawing: Multipartite Layout"])
+                                  "Drawing: House With Colors", "Drawing: Labels And Colors", 
+                                  "Drawing: Multipartite Layout", "Drawing: Node Colormap"])
+
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -20,6 +22,33 @@ 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)
 
+# Function to display Drawing: Node Colormap
+def display_node_colormap():
+    st.title("Drawing: Node Colormap")
+
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        G = nx.cycle_graph(24)
+        pos = nx.circular_layout(G)
+        nx.draw(G, pos, node_color=range(24), node_size=800, cmap=plt.cm.Blues)
+        st.pyplot(plt)
+
+    elif option == "Create your own":
+        num_nodes = st.number_input("Number of nodes:", min_value=2, max_value=100, value=24)
+        color_map = st.selectbox("Select a colormap:", plt.colormaps(), index=plt.colormaps().index('Blues'))
+
+        if st.button("Generate Graph"):
+            # Create cycle graph with custom number of nodes
+            G_custom = nx.cycle_graph(num_nodes)
+            pos = nx.circular_layout(G_custom)
+            nx.draw(G_custom, pos, node_color=range(num_nodes), node_size=800, cmap=plt.get_cmap(color_map))
+            st.pyplot(plt)
+
+# Display Drawing: Node Colormap if selected
+if sidebar_option == "Drawing: Node Colormap":
+    display_node_colormap()
+
 # Function to create a multipartite graph
 def multilayered_graph(*subset_sizes):
     G = nx.Graph()