Update app.py

app.py

@@ -7,7 +7,8 @@ import numpy as np
 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"])
+                                  "Basic: Simple graph Directed", "Drawing: Custom Node Position", 
+                                  "Drawing: Chess Masters"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -56,7 +57,7 @@ def display_read_write_graph(G):
     
     # Write the graph's edge list to a file
     st.write("### Writing Edge List to 'grid.edgelist' file:")
-    nx.write_edgelist(G, path="grid.edgelist", delimiter=":")
+    nx.write_edgelist(G, path="grid.edgelist", delimiter=":")  # Save edge list
     st.write("Edge list written to 'grid.edgelist'")
 
     # Read the graph from the edge list
@@ -124,6 +125,45 @@ def display_custom_node_position():
     # Draw the graph
     draw_graph(G, pos)
 
+# Function to display Chess Masters Graphs for Drawing: Chess Masters
+def display_chess_masters():
+    st.title("Drawing: Chess Masters")
+
+    # Define the directed graph
+    G = nx.DiGraph([(0, 3), (1, 3), (2, 4), (3, 5), (3, 6), (4, 6), (5, 6)])
+
+    # Group nodes by column
+    left_nodes = [0, 1, 2]
+    middle_nodes = [3, 4]
+    right_nodes = [5, 6]
+
+    # Set the position according to column (x-coord)
+    pos = {n: (0, i) for i, n in enumerate(left_nodes)}
+    pos.update({n: (1, i + 0.5) for i, n in enumerate(middle_nodes)})
+    pos.update({n: (2, i + 0.5) for i, n in enumerate(right_nodes)})
+
+    # Define plot options
+    options = {
+        "node_size": 500,
+        "node_color": "lightblue",
+        "arrowsize": 20,
+        "width": 2,
+        "edge_color": "gray",
+    }
+
+    # Draw the graph with the specified positions and options
+    nx.draw_networkx(G, pos, **options)
+
+    # Set margins for the axes so that nodes aren't clipped
+    ax = plt.gca()
+    ax.margins(0.20)
+
+    # Turn off the axis for better visualization
+    plt.axis("off")
+
+    # Show the plot
+    st.pyplot(plt)
+
 # Display Basic: Properties if selected
 if sidebar_option == "Basic: Properties":
     st.title("Basic: Properties")
@@ -227,4 +267,8 @@ elif sidebar_option == "Basic: Simple graph Directed":
 
 # Display Drawing: Custom Node Position if selected
 elif sidebar_option == "Drawing: Custom Node Position":
-    display_custom_node_position()
+    display_custom_node_position()
+
+# Display Drawing: Chess Masters if selected
+elif sidebar_option == "Drawing: Chess Masters":
+    display_chess_masters()