Update app.py

app.py

@@ -74,17 +74,24 @@ def display_weighted_graph():
         nodes = st.text_input("Enter nodes (comma-separated):", "a,b,c,d,e,f").split(',')
         nodes = [node.strip() for node in nodes]
 
-        # Add a button to generate a new graph
+        # Create a placeholder for weight inputs
+        weight_inputs = {}
+
+        # Collecting weight inputs for each edge
+        for i in range(len(nodes)):
+            for j in range(i + 1, len(nodes)):
+                weight = st.number_input(f"Weight between {nodes[i]} and {nodes[j]}:", min_value=0.0, value=1.0, key=f"weight_{nodes[i]}_{nodes[j]}")
+                weight_inputs[(nodes[i], nodes[j])] = weight
+
+        # Only show "Generate Graph" after weight inputs are made
         generate_button = st.button("Generate Graph")
 
         if generate_button:
             G_custom = nx.Graph()
 
-            # Allow user to input weights for the edges between nodes
-            for i in range(len(nodes)):
-                for j in range(i + 1, len(nodes)):
-                    weight = st.number_input(f"Weight between {nodes[i]} and {nodes[j]}:", min_value=0.0, value=1.0)
-                    G_custom.add_edge(nodes[i], nodes[j], weight=weight)
+            # Add edges to the graph based on the input weights
+            for (node1, node2), weight in weight_inputs.items():
+                G_custom.add_edge(node1, node2, weight=weight)
 
             # Create layout for visualization
             pos = nx.spring_layout(G_custom, seed=7)