Update app.py

app.py

@@ -8,7 +8,7 @@ sidebar_option = st.sidebar.radio("Select an option",
                                  ["Select an option", "Basic: Properties", 
                                   "Basic: Read and write graphs", "Basic: Simple graph", 
                                   "Basic: Simple graph Directed", "Drawing: Custom Node Position",
-                                  "Drawing: Cluster Layout"])
+                                  "Drawing: Cluster Layout", "Drawing: Degree Analysis"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -149,8 +149,82 @@ def display_cluster_layout():
 
     plt.tight_layout()
     st.pyplot(plt)
-    
+
+# Function to display Degree Analysis for Drawing: Degree Analysis
+def display_degree_analysis():
+    st.title("Drawing: Degree Analysis")
+
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        G = nx.gnp_random_graph(100, 0.02, seed=10374196)
         
+        degree_sequence = sorted((d for n, d in G.degree()), reverse=True)
+        dmax = max(degree_sequence)
+
+        fig = plt.figure("Degree of a random graph", figsize=(8, 8))
+        # Create a gridspec for adding subplots of different sizes
+        axgrid = fig.add_gridspec(5, 4)
+
+        ax0 = fig.add_subplot(axgrid[0:3, :])
+        Gcc = G.subgraph(sorted(nx.connected_components(G), key=len, reverse=True)[0])
+        pos = nx.spring_layout(Gcc, seed=10396953)
+        nx.draw_networkx_nodes(Gcc, pos, ax=ax0, node_size=20)
+        nx.draw_networkx_edges(Gcc, pos, ax=ax0, alpha=0.4)
+        ax0.set_title("Connected components of G")
+        ax0.set_axis_off()
+
+        ax1 = fig.add_subplot(axgrid[3:, :2])
+        ax1.plot(degree_sequence, "b-", marker="o")
+        ax1.set_title("Degree Rank Plot")
+        ax1.set_ylabel("Degree")
+        ax1.set_xlabel("Rank")
+
+        ax2 = fig.add_subplot(axgrid[3:, 2:])
+        ax2.bar(*np.unique(degree_sequence, return_counts=True))
+        ax2.set_title("Degree histogram")
+        ax2.set_xlabel("Degree")
+        ax2.set_ylabel("# of Nodes")
+
+        fig.tight_layout()
+        st.pyplot(fig)
+
+    elif option == "Create your own":
+        n_nodes = st.number_input("Number of nodes:", min_value=2, max_value=500, value=100)
+        p_edge = st.slider("Edge probability:", min_value=0.0, max_value=1.0, value=0.02)
+        
+        if st.button("Generate"):
+            if n_nodes >= 2:
+                G_custom = nx.gnp_random_graph(n_nodes, p_edge, seed=10374196)
+                degree_sequence = sorted((d for n, d in G_custom.degree()), reverse=True)
+                dmax = max(degree_sequence)
+
+                fig = plt.figure("Degree of a random graph", figsize=(8, 8))
+                # Create a gridspec for adding subplots of different sizes
+                axgrid = fig.add_gridspec(5, 4)
+
+                ax0 = fig.add_subplot(axgrid[0:3, :])
+                Gcc = G_custom.subgraph(sorted(nx.connected_components(G_custom), key=len, reverse=True)[0])
+                pos = nx.spring_layout(Gcc, seed=10396953)
+                nx.draw_networkx_nodes(Gcc, pos, ax=ax0, node_size=20)
+                nx.draw_networkx_edges(Gcc, pos, ax=ax0, alpha=0.4)
+                ax0.set_title("Connected components of G")
+                ax0.set_axis_off()
+
+                ax1 = fig.add_subplot(axgrid[3:, :2])
+                ax1.plot(degree_sequence, "b-", marker="o")
+                ax1.set_title("Degree Rank Plot")
+                ax1.set_ylabel("Degree")
+                ax1.set_xlabel("Rank")
+
+                ax2 = fig.add_subplot(axgrid[3:, 2:])
+                ax2.bar(*np.unique(degree_sequence, return_counts=True))
+                ax2.set_title("Degree histogram")
+                ax2.set_xlabel("Degree")
+                ax2.set_ylabel("# of Nodes")
+
+                fig.tight_layout()
+                st.pyplot(fig)
 
 # Display Basic: Properties if selected
 if sidebar_option == "Basic: Properties":
@@ -260,3 +334,7 @@ elif sidebar_option == "Drawing: Custom Node Position":
 # Display Drawing: Cluster Layout if selected
 elif sidebar_option == "Drawing: Cluster Layout":
     display_cluster_layout()
+
+# Display Drawing: Degree Analysis if selected
+elif sidebar_option == "Drawing: Degree Analysis":
+    display_degree_analysis()