Update app.py

app.py

@@ -70,27 +70,27 @@ def display_weighted_graph():
         st.pyplot(plt)
 
     elif option == "Create your own":
-        # User can create their own graph
-        nodes = st.text_input("Enter nodes (comma-separated):", "a,b,c,d,e,f").split(',')
-        nodes = [node.strip() for node in nodes]
-
-        # Create a placeholder for weight inputs
-        weight_inputs = {}
+        # User can create their own graph with edges and weights
+        edge_input = st.text_area(
+            "Enter edges with weights (format: node1,node2,weight;node1,node2,weight;...)",
+            "a,b,0.6;a,c,0.2;c,d,0.1;c,e,0.7;c,f,0.9;a,d,0.3"
+        )
 
-        # 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
+        # Parse the input string
+        edges = edge_input.split(";")
+        edge_list = []
+        for edge in edges:
+            node1, node2, weight = edge.split(",")
+            edge_list.append((node1.strip(), node2.strip(), float(weight.strip())))
 
-        # Only show "Generate Graph" after weight inputs are made
+        # Add a button to generate the graph
         generate_button = st.button("Generate Graph")
 
         if generate_button:
             G_custom = nx.Graph()
 
-            # Add edges to the graph based on the input weights
-            for (node1, node2), weight in weight_inputs.items():
+            # Add edges to the graph
+            for node1, node2, weight in edge_list:
                 G_custom.add_edge(node1, node2, weight=weight)
 
             # Create layout for visualization