Update app.py

app.py

@@ -74,35 +74,40 @@ 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]
 
-        G_custom = nx.Graph()
+        # Add a button to generate a new graph
+        generate_button = st.button("Generate Graph")
 
-        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)
+        if generate_button:
+            G_custom = nx.Graph()
 
-        # Create layout for visualization
-        pos = nx.spring_layout(G_custom, seed=7)
-        
-        # Determine edges based on weight
-        elarge = [(u, v) for (u, v, d) in G_custom.edges(data=True) if d["weight"] > 0.5]
-        esmall = [(u, v) for (u, v, d) in G_custom.edges(data=True) if d["weight"] <= 0.5]
+            # 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)
 
-        # Draw the graph
-        nx.draw_networkx_nodes(G_custom, pos, node_size=700)
-        nx.draw_networkx_edges(G_custom, pos, edgelist=elarge, width=6)
-        nx.draw_networkx_edges(
-            G_custom, pos, edgelist=esmall, width=6, alpha=0.5, edge_color="b", style="dashed"
-        )
-        nx.draw_networkx_labels(G_custom, pos, font_size=20, font_family="sans-serif")
-        edge_labels = nx.get_edge_attributes(G_custom, "weight")
-        nx.draw_networkx_edge_labels(G_custom, pos, edge_labels)
+            # Create layout for visualization
+            pos = nx.spring_layout(G_custom, seed=7)
 
-        ax = plt.gca()
-        ax.margins(0.08)
-        plt.axis("off")
-        plt.tight_layout()
-        st.pyplot(plt)
+            # Determine edges based on weight
+            elarge = [(u, v) for (u, v, d) in G_custom.edges(data=True) if d["weight"] > 0.5]
+            esmall = [(u, v) for (u, v, d) in G_custom.edges(data=True) if d["weight"] <= 0.5]
+
+            # Draw the graph
+            nx.draw_networkx_nodes(G_custom, pos, node_size=700)
+            nx.draw_networkx_edges(G_custom, pos, edgelist=elarge, width=6)
+            nx.draw_networkx_edges(
+                G_custom, pos, edgelist=esmall, width=6, alpha=0.5, edge_color="b", style="dashed"
+            )
+            nx.draw_networkx_labels(G_custom, pos, font_size=20, font_family="sans-serif")
+            edge_labels = nx.get_edge_attributes(G_custom, "weight")
+            nx.draw_networkx_edge_labels(G_custom, pos, edge_labels)
+
+            ax = plt.gca()
+            ax.margins(0.08)
+            plt.axis("off")
+            plt.tight_layout()
+            st.pyplot(plt)
 
 # Display Drawing: Weighted Graph if selected
 if sidebar_option == "Drawing: Weighted Graph":