Update app.py

app.py

@@ -94,46 +94,50 @@ def triads_graph():
     elif graph_mode == "Create Your Own":
         st.write("### Create Your Own Triads")
 
-        # Let user define custom triads
+        # Input: Enter triads as a dictionary (e.g., {'triad_name': [(1, 2), (2, 1)]})
         triad_input = st.text_area(
-            "Enter the edges for your triad (e.g., [(1, 2), (2, 1), (3, 1)])",
-            value="[(1, 2), (2, 1), (3, 1)]"
+            "Enter your triads in the format: {'triad_name': [(edge1), (edge2), ...]}",
+            value="{'003': [], '012': [(1, 2)]}"
         )
         
-        # Try to evaluate the input as a list of edges
+        # Try to evaluate the input as a dictionary of triads
         try:
-            custom_triad = eval(triad_input)
-            if isinstance(custom_triad, list) and all(isinstance(edge, tuple) and len(edge) == 2 for edge in custom_triad):
-                G = nx.DiGraph()
-                G.add_nodes_from([1, 2, 3])
-                G.add_edges_from(custom_triad)
-
-                # Plot the graph
-                fig, ax = plt.subplots(figsize=(5, 5))
-                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,
-                    "Custom Triad",
-                    fontsize=15,
-                    fontweight="extra bold",
-                    horizontalalignment="center",
-                    bbox={"boxstyle": "square,pad=0.3", "fc": "none"},
-                )
+            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 list of edges in the format [(1, 2), (2, 1), ...].")
+                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}")