Update app.py

app.py

@@ -11,7 +11,7 @@ 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: House With Colors", "Drawing: Labels And Colors"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -19,6 +19,87 @@ 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: Labels And Colors
+def display_labels_and_colors():
+    st.title("Drawing: Labels And Colors")
+
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        # Create a cubical graph
+        G = nx.cubical_graph()
+        pos = nx.spring_layout(G, seed=3113794652)  # positions for all nodes
+
+        # Draw nodes with different colors
+        options = {"edgecolors": "tab:gray", "node_size": 800, "alpha": 0.9}
+        nx.draw_networkx_nodes(G, pos, nodelist=[0, 1, 2, 3], node_color="tab:red", **options)
+        nx.draw_networkx_nodes(G, pos, nodelist=[4, 5, 6, 7], node_color="tab:blue", **options)
+
+        # Draw edges
+        nx.draw_networkx_edges(G, pos, width=1.0, alpha=0.5)
+        nx.draw_networkx_edges(
+            G,
+            pos,
+            edgelist=[(0, 1), (1, 2), (2, 3), (3, 0)],
+            width=8,
+            alpha=0.5,
+            edge_color="tab:red",
+        )
+        nx.draw_networkx_edges(
+            G,
+            pos,
+            edgelist=[(4, 5), (5, 6), (6, 7), (7, 4)],
+            width=8,
+            alpha=0.5,
+            edge_color="tab:blue",
+        )
+
+        # Add labels for nodes
+        labels = {0: r"$a$", 1: r"$b$", 2: r"$c$", 3: r"$d$", 4: r"$\alpha$", 5: r"$\beta$", 6: r"$\gamma$", 7: r"$\delta$"}
+        nx.draw_networkx_labels(G, pos, labels, font_size=22, font_color="whitesmoke")
+
+        plt.tight_layout()
+        plt.axis("off")
+        st.pyplot(plt)
+
+    elif option == "Create your own":
+        # Let the user input the nodes and edges of the graph
+        st.write("Enter the nodes and edges to create your own labeled graph.")
+
+        nodes = st.text_area("Enter node labels (comma-separated, e.g., a,b,c,d):", value="a,b,c,d")
+        node_labels = nodes.split(',')
+
+        edges = st.text_area("Enter edges (format: node1-node2, comma-separated, e.g., a-b,b-c):", value="a-b,b-c,c-d")
+        edge_list = [tuple(edge.split('-')) for edge in edges.split(',')]
+
+        # Generate graph based on user input
+        G_custom = nx.Graph()
+        G_custom.add_nodes_from(node_labels)
+        G_custom.add_edges_from(edge_list)
+
+        # Let user choose colors for nodes and edges
+        node_color = st.color_picker("Pick a color for nodes:", "#FF6347")
+        edge_color = st.color_picker("Pick a color for edges:", "#4682B4")
+        
+        # Generate layout for the nodes
+        pos_custom = nx.spring_layout(G_custom)
+
+        # Draw the graph
+        nx.draw_networkx_nodes(G_custom, pos_custom, node_color=node_color, node_size=800, edgecolors="gray", alpha=0.9)
+        nx.draw_networkx_edges(G_custom, pos_custom, edge_color=edge_color, width=2, alpha=0.7)
+        
+        # Create custom labels
+        custom_labels = {node: f"${node}$" for node in node_labels}
+        nx.draw_networkx_labels(G_custom, pos_custom, labels=custom_labels, font_size=22, font_color="whitesmoke")
+
+        plt.tight_layout()
+        plt.axis("off")
+        st.pyplot(plt)
+
+# Display Drawing: Labels And Colors if selected
+if sidebar_option == "Drawing: Labels And Colors":
+    display_labels_and_colors()
+
 # Function to display Drawing: House With Colors
 def display_house_with_colors():
     st.title("Drawing: House With Colors")