Update app.py

app.py

@@ -14,7 +14,7 @@ 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: Random Geometric Graph"])
+                                  "Drawing: Rainbow Coloring", "Drawing: Random Geometric Graph","Drawing: Self-loops"])
 
 
 
@@ -24,6 +24,57 @@ 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: Self-loops
+def display_self_loops():
+    st.title("Drawing: Self-loops")
+
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        # Default example of a graph with self-loops
+        G = nx.complete_graph(3, create_using=nx.DiGraph)
+        G.add_edge(0, 0)  # Add a self-loop to node 0
+        pos = nx.circular_layout(G)
+
+        # Draw the graph
+        nx.draw(G, pos, with_labels=True)
+
+        # Add self-loops to the remaining nodes
+        edgelist = [(1, 1), (2, 2)]
+        G.add_edges_from(edgelist)
+
+        # Draw the newly added self-loops with different formatting
+        nx.draw_networkx_edges(G, pos, edgelist=edgelist, arrowstyle="<|-", style="dashed")
+        st.pyplot(plt)
+
+    elif option == "Create your own":
+        # User can create their own graph with self-loops
+        num_nodes = st.number_input("Number of nodes:", min_value=2, max_value=20, value=3)
+        add_self_loops = st.checkbox("Add self-loops to all nodes?", value=True)
+
+        if st.button("Generate Graph"):
+            # Generate a complete graph
+            G = nx.complete_graph(num_nodes, create_using=nx.DiGraph)
+            
+            # Optionally add self-loops to all nodes
+            if add_self_loops:
+                for node in G.nodes():
+                    G.add_edge(node, node)
+
+            pos = nx.circular_layout(G)
+            
+            # Draw the graph with self-loops
+            nx.draw(G, pos, with_labels=True)
+
+            # Style self-loops differently
+            edgelist = [(node, node) for node in G.nodes()]
+            nx.draw_networkx_edges(G, pos, edgelist=edgelist, arrowstyle="<|-", style="dashed")
+            st.pyplot(plt)
+
+# Display Drawing: Self-loops if selected
+if sidebar_option == "Drawing: Self-loops":
+    display_self_loops()
+
 # Function to display Drawing: Random Geometric Graph
 def display_random_geometric_graph():
     st.title("Drawing: Random Geometric Graph")