Update app.py

app.py

@@ -62,7 +62,7 @@ def triads_graph():
         }
 
         fig, axes = plt.subplots(4, 4, figsize=(10, 10))
-        
+
         for (title, triad), ax in zip(triads.items(), axes.flatten()):
             G = nx.DiGraph()
             G.add_nodes_from([1, 2, 3])
@@ -70,7 +70,7 @@ def triads_graph():
             nx.draw_networkx(
                 G,
                 ax=ax,
-                with_labels=False,
+                with_labels=True,  # Labels the vertices
                 node_color=["green"],
                 node_size=200,
                 arrowsize=20,
@@ -99,47 +99,49 @@ def triads_graph():
             "Enter your triads in the format: {'triad_name': [(edge1), (edge2), ...]}",
             value="{'003': [], '012': [(1, 2)]}"
         )
-        
-        # Try to evaluate the input as a dictionary of triads
-        try:
-            custom_triads = eval(triad_input)
-            if isinstance(custom_triads, dict) and all(isinstance(value, list) and all(isinstance(edge, tuple) and len(edge) == 2 for edge in value) for value in custom_triads.values()):
-                fig, axes = plt.subplots(len(custom_triads), 1, figsize=(10, len(custom_triads) * 5))
-                if len(custom_triads) == 1:  # Handle case where only one triad is entered
-                    axes = [axes]
-
-                for (title, triad), ax in zip(custom_triads.items(), axes):
-                    G = nx.DiGraph()
-                    G.add_nodes_from([1, 2, 3])
-                    G.add_edges_from(triad)
-
-                    nx.draw_networkx(
-                        G,
-                        ax=ax,
-                        with_labels=False,
-                        node_color=["green"],
-                        node_size=200,
-                        arrowsize=20,
-                        width=2,
-                        pos=nx.planar_layout(G),
-                    )
-                    ax.set_xlim(val * 1.2 for val in ax.get_xlim())
-                    ax.set_ylim(val * 1.2 for val in ax.get_ylim())
-                    ax.text(
-                        0,
-                        0,
-                        title,
-                        fontsize=15,
-                        fontweight="extra bold",
-                        horizontalalignment="center",
-                        bbox={"boxstyle": "square,pad=0.3", "fc": "none"},
-                    )
-                fig.tight_layout()
-                st.pyplot(fig)
-            else:
-                st.error("Invalid format. Please enter a dictionary of triads in the format {'triad_name': [(edge1), (edge2), ...]}.")
-        except Exception as e:
-            st.error(f"Error parsing input: {e}")
+
+        # Generate Button
+        if st.button("Generate Graph"):
+            # Try to evaluate the input as a dictionary of triads
+            try:
+                custom_triads = eval(triad_input)
+                if isinstance(custom_triads, dict) and all(isinstance(value, list) and all(isinstance(edge, tuple) and len(edge) == 2 for edge in value) for value in custom_triads.values()):
+                    fig, axes = plt.subplots(len(custom_triads), 1, figsize=(10, len(custom_triads) * 5))
+                    if len(custom_triads) == 1:  # Handle case where only one triad is entered
+                        axes = [axes]
+
+                    for (title, triad), ax in zip(custom_triads.items(), axes):
+                        G = nx.DiGraph()
+                        G.add_nodes_from([1, 2, 3])
+                        G.add_edges_from(triad)
+
+                        nx.draw_networkx(
+                            G,
+                            ax=ax,
+                            with_labels=True,  # Labels the vertices
+                            node_color=["green"],
+                            node_size=200,
+                            arrowsize=20,
+                            width=2,
+                            pos=nx.planar_layout(G),
+                        )
+                        ax.set_xlim(val * 1.2 for val in ax.get_xlim())
+                        ax.set_ylim(val * 1.2 for val in ax.get_ylim())
+                        ax.text(
+                            0,
+                            0,
+                            title,
+                            fontsize=15,
+                            fontweight="extra bold",
+                            horizontalalignment="center",
+                            bbox={"boxstyle": "square,pad=0.3", "fc": "none"},
+                        )
+                    fig.tight_layout()
+                    st.pyplot(fig)
+                else:
+                    st.error("Invalid format. Please enter a dictionary of triads in the format {'triad_name': [(edge1), (edge2), ...]}.")
+            except Exception as e:
+                st.error(f"Error parsing input: {e}")
 
 # Display the corresponding page based on sidebar option
 if sidebar_option == "Graph: Triads":