Update app.py

app.py

@@ -40,7 +40,7 @@ def dag_topological_layout():
     )
 
     if graph_mode == "Default Example":
-        # Create a Directed Acyclic Graph (DAG)
+        # Default DAG example
         G = nx.DiGraph(
             [
                 ("f", "a"),
@@ -73,14 +73,22 @@ def dag_topological_layout():
     elif graph_mode == "Create Your Own":
         st.write("### Custom DAG Creation")
 
-        # Inputs for custom DAG creation (nodes and edges)
-        num_nodes = st.slider("Number of Nodes", min_value=2, max_value=10, value=5)
-        edges = []
+        # Allow the user to input the number of nodes
+        num_nodes = st.number_input("Enter the number of nodes", min_value=2, value=5)
+
+        # Create node names based on the number of nodes
+        nodes = [str(i) for i in range(num_nodes)]
 
+        st.write(f"### Nodes: {nodes}")
+        st.write("#### Add Edges between Nodes")
+
+        # Allow the user to select pairs of nodes to add edges
+        edges = []
         for i in range(num_nodes):
             for j in range(i + 1, num_nodes):
-                if st.checkbox(f"Add edge from {i} to {j}", value=False):
-                    edges.append((i, j))
+                edge = (nodes[i], nodes[j])
+                if st.checkbox(f"Add edge from {edge[0]} to {edge[1]}", value=False):
+                    edges.append(edge)
 
         # Create the custom DAG
         G_custom = nx.DiGraph()