Update app.py

app.py

@@ -14,7 +14,8 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Drawing: Ego Graph", "Drawing: Eigenvalues", "Drawing: Four Grids",
                                   "Drawing: House With Colors", "Drawing: Labels And Colors", 
                                   "Drawing: Multipartite Layout", "Drawing: Node Colormap",
-                                  "Drawing: Rainbow Coloring"])
+                                  "Drawing: Rainbow Coloring", "Drawing: Random Geometric Graph"])
+
 
 
 # Helper function to draw and display graph
@@ -23,6 +24,89 @@ 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: Random Geometric Graph
+def display_random_geometric_graph():
+    st.title("Drawing: Random Geometric Graph")
+
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        # Default random geometric graph example
+        G = nx.random_geometric_graph(200, 0.125, seed=896803)
+        pos = nx.get_node_attributes(G, "pos")
+
+        # Find node near the center (0.5, 0.5)
+        dmin = 1
+        ncenter = 0
+        for n in pos:
+            x, y = pos[n]
+            d = (x - 0.5) ** 2 + (y - 0.5) ** 2
+            if d < dmin:
+                ncenter = n
+                dmin = d
+
+        # Color by path length from node near center
+        p = dict(nx.single_source_shortest_path_length(G, ncenter))
+
+        plt.figure(figsize=(8, 8))
+        nx.draw_networkx_edges(G, pos, alpha=0.4)
+        nx.draw_networkx_nodes(
+            G,
+            pos,
+            nodelist=list(p.keys()),
+            node_size=80,
+            node_color=list(p.values()),
+            cmap=plt.cm.Reds_r,
+        )
+
+        plt.xlim(-0.05, 1.05)
+        plt.ylim(-0.05, 1.05)
+        plt.axis("off")
+        st.pyplot(plt)
+
+    elif option == "Create your own":
+        # User can create their own random geometric graph
+        num_nodes = st.number_input("Number of nodes:", min_value=2, max_value=500, value=200)
+        distance = st.slider("Edge distance threshold (between 0 and 1):", 0.01, 1.0, 0.125)
+
+        if st.button("Generate Graph"):
+            # Generate the graph with user input
+            G = nx.random_geometric_graph(num_nodes, distance, seed=896803)
+            pos = nx.get_node_attributes(G, "pos")
+
+            # Find node near the center (0.5, 0.5)
+            dmin = 1
+            ncenter = 0
+            for n in pos:
+                x, y = pos[n]
+                d = (x - 0.5) ** 2 + (y - 0.5) ** 2
+                if d < dmin:
+                    ncenter = n
+                    dmin = d
+
+            # Color by path length from node near center
+            p = dict(nx.single_source_shortest_path_length(G, ncenter))
+
+            plt.figure(figsize=(8, 8))
+            nx.draw_networkx_edges(G, pos, alpha=0.4)
+            nx.draw_networkx_nodes(
+                G,
+                pos,
+                nodelist=list(p.keys()),
+                node_size=80,
+                node_color=list(p.values()),
+                cmap=plt.cm.Reds_r,
+            )
+
+            plt.xlim(-0.05, 1.05)
+            plt.ylim(-0.05, 1.05)
+            plt.axis("off")
+            st.pyplot(plt)
+
+# Display Drawing: Random Geometric Graph if selected
+if sidebar_option == "Drawing: Random Geometric Graph":
+    display_random_geometric_graph()
+
 # Function to display Drawing: Rainbow Coloring
 def display_rainbow_coloring():
     st.title("Drawing: Rainbow Coloring")