Update app.py

app.py

@@ -13,7 +13,8 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "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: Node Colormap"])
+                                  "Drawing: Multipartite Layout", "Drawing: Node Colormap",
+                                  "Drawing: Rainbow Coloring"])
 
 
 # Helper function to draw and display graph
@@ -22,6 +23,105 @@ 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: Rainbow Coloring
+def display_rainbow_coloring():
+    st.title("Drawing: Rainbow Coloring")
+
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        # Rainbow Coloring with default parameters
+        node_dist_to_color = {
+            1: "tab:red",
+            2: "tab:orange",
+            3: "tab:olive",
+            4: "tab:green",
+            5: "tab:blue",
+            6: "tab:purple",
+        }
+
+        nnodes = 13
+        G = nx.complete_graph(nnodes)
+
+        n = (nnodes - 1) // 2
+        ndist_iter = list(range(1, n + 1))
+        ndist_iter += ndist_iter[::-1]
+
+        def cycle(nlist, n):
+            return nlist[-n:] + nlist[:-n]
+
+        nodes = list(G.nodes())
+        for i, nd in enumerate(ndist_iter):
+            for u, v in zip(nodes, cycle(nodes, i + 1)):
+                G[u][v]["color"] = node_dist_to_color[nd]
+
+        pos = nx.circular_layout(G)
+        # Create a figure with 1:1 aspect ratio to preserve the circle.
+        fig, ax = plt.subplots(figsize=(8, 8))
+        node_opts = {"node_size": 500, "node_color": "w", "edgecolors": "k", "linewidths": 2.0}
+        nx.draw_networkx_nodes(G, pos, **node_opts)
+        nx.draw_networkx_labels(G, pos, font_size=14)
+        # Extract color from edge data
+        edge_colors = [edgedata["color"] for _, _, edgedata in G.edges(data=True)]
+        nx.draw_networkx_edges(G, pos, width=2.0, edge_color=edge_colors)
+
+        ax.set_axis_off()
+        fig.tight_layout()
+        st.pyplot(plt)
+
+    elif option == "Create your own":
+        nnodes = st.number_input("Number of nodes:", min_value=2, max_value=50, value=13)
+        
+        # Allow users to create their own color map
+        red = st.color_picker("Select a color for Red (1)", "#ff0000")
+        orange = st.color_picker("Select a color for Orange (2)", "#ff7f00")
+        olive = st.color_picker("Select a color for Olive (3)", "#808000")
+        green = st.color_picker("Select a color for Green (4)", "#008000")
+        blue = st.color_picker("Select a color for Blue (5)", "#0000ff")
+        purple = st.color_picker("Select a color for Purple (6)", "#800080")
+        
+        node_dist_to_color = {
+            1: red,
+            2: orange,
+            3: olive,
+            4: green,
+            5: blue,
+            6: purple,
+        }
+
+        if st.button("Generate Graph"):
+            G = nx.complete_graph(nnodes)
+
+            n = (nnodes - 1) // 2
+            ndist_iter = list(range(1, n + 1))
+            ndist_iter += ndist_iter[::-1]
+
+            def cycle(nlist, n):
+                return nlist[-n:] + nlist[:-n]
+
+            nodes = list(G.nodes())
+            for i, nd in enumerate(ndist_iter):
+                for u, v in zip(nodes, cycle(nodes, i + 1)):
+                    G[u][v]["color"] = node_dist_to_color[nd]
+
+            pos = nx.circular_layout(G)
+            # Create a figure with 1:1 aspect ratio to preserve the circle.
+            fig, ax = plt.subplots(figsize=(8, 8))
+            node_opts = {"node_size": 500, "node_color": "w", "edgecolors": "k", "linewidths": 2.0}
+            nx.draw_networkx_nodes(G, pos, **node_opts)
+            nx.draw_networkx_labels(G, pos, font_size=14)
+            # Extract color from edge data
+            edge_colors = [edgedata["color"] for _, _, edgedata in G.edges(data=True)]
+            nx.draw_networkx_edges(G, pos, width=2.0, edge_color=edge_colors)
+
+            ax.set_axis_off()
+            fig.tight_layout()
+            st.pyplot(plt)
+
+# Display Drawing: Rainbow Coloring if selected
+if sidebar_option == "Drawing: Rainbow Coloring":
+    display_rainbow_coloring()
+
 # Function to display Drawing: Node Colormap
 def display_node_colormap():
     st.title("Drawing: Node Colormap")