Update app.py

app.py

@@ -20,7 +20,8 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Drawing: Multipartite Layout", "Drawing: Node Colormap",
                                   "Drawing: Rainbow Coloring", "Drawing: Random Geometric Graph","Drawing: Self-loops",
                                   "Drawing: Simple Path", "Drawing: Spectral Embedding", "Drawing: Traveling Salesman Problem",
-                                  "Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation", "3D Drawing: Basic Matplotlib"])
+                                  "Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation", "3D Drawing: Basic Matplotlib",
+                                  "Graphviz Layout: Attributes"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -28,6 +29,76 @@ 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)
 
+def graphviz_layout_attributes():
+    st.title("Graphviz Layout: Attributes")
+
+    # Sidebar selection for Default Example or Custom Graph
+    graph_mode = st.radio(
+        "Choose a Mode:",
+        ("Default Example", "Create Your Own"),
+        help="Default example shows graph with attributes, or you can create your own custom graph."
+    )
+
+    if graph_mode == "Default Example":
+        # Create a graph and add edges and nodes with attributes
+        G = nx.Graph()
+        G.add_edge(1, 2, color="red")
+        G.add_edge(2, 3, color="red")
+        G.add_node(3)
+        G.add_node(4)
+
+        # Convert NetworkX graph to Graphviz graph
+        A = nx.nx_agraph.to_agraph(G)  # convert to a graphviz graph
+        A.draw("attributes.png", prog="neato")  # Draw with pygraphviz
+
+        # Display the graph image
+        st.image("attributes.png", caption="Graph with Attributes")
+
+        # Convert back to NetworkX graph and display attributes
+        X = nx.nx_agraph.from_agraph(A)
+        
+        st.write("### Edge Data:")
+        st.write(list(X.edges(data=True)))  # Display edges with data
+        
+        st.write("### Default Graph Attributes:")
+        st.write(X.graph)  # Display default graph attributes
+        
+        st.write("### Node Data:")
+        st.write(X.nodes.data(True))  # Display node data with attributes
+
+    elif graph_mode == "Create Your Own":
+        st.write("### Custom Graph Creation")
+
+        # Input for custom graph (create a random graph)
+        num_nodes = st.slider("Number of Nodes", min_value=2, max_value=50, value=10)
+        num_edges = st.slider("Number of Edges", min_value=1, max_value=num_nodes*(num_nodes-1)//2, value=10)
+
+        # Generate a custom graph
+        G_custom = nx.gnm_random_graph(num_nodes, num_edges)
+
+        # Convert to Graphviz and draw
+        A_custom = nx.nx_agraph.to_agraph(G_custom)
+        A_custom.draw("custom_attributes.png", prog="neato")
+
+        # Display the custom graph image
+        st.image("custom_attributes.png", caption="Custom Graph with Attributes")
+
+        # Convert back to NetworkX graph and display custom attributes
+        X_custom = nx.nx_agraph.from_agraph(A_custom)
+
+        st.write("### Edge Data:")
+        st.write(list(X_custom.edges(data=True)))  # Display edges with data
+        
+        st.write("### Default Graph Attributes:")
+        st.write(X_custom.graph)  # Display default graph attributes
+        
+        st.write("### Node Data:")
+        st.write(X_custom.nodes.data(True))  # Display node data with attributes
+
+# Display the corresponding page based on sidebar option
+if sidebar_option == "Graphviz Layout: Attributes":
+    graphviz_layout_attributes()
+
 if sidebar_option == "3D Drawing: Animations of 3D Rotation":
     st.title("3D Drawing: Animations of 3D Rotation")