Update app.py

app.py

@@ -21,7 +21,8 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Drawing: Rainbow Coloring", "Drawing: Random Geometric Graph","Drawing: Self-loops",
                                   "Drawing: Simple Path", "Drawing: Spectral Embedding", "Drawing: Traveling Salesman Problem",
                                   "Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation", "3D Drawing: Basic Matplotlib",
-                                  "Graph: DAG - Topological Layout", "Graph: Erdos Renyi", "Graph: Karate Club", "Graph: Minimum Spanning Tree"])
+                                  "Graph: DAG - Topological Layout", "Graph: Erdos Renyi", "Graph: Karate Club", "Graph: Minimum Spanning Tree",
+                                  "Graph: Triads"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -29,6 +30,117 @@ def draw_graph(G, pos=None, title="Graph Visualization"):
     nx.draw(G, pos=pos, with_labels=True, node_color='lightblue', node_size=500, font_size=10, font_weight='bold')
     st.pyplot(plt)
 
+def triads_graph():
+    st.title("Graph: Triads")
+
+    # Sidebar selection for Default Example or Custom Triads
+    graph_mode = st.radio(
+        "Choose a Mode:",
+        ("Default Example", "Create Your Own"),
+        help="Default example shows predefined triads, or you can create your own triads."
+    )
+
+    if graph_mode == "Default Example":
+        # Define the triads
+        triads = {
+            "003": [],
+            "012": [(1, 2)],
+            "102": [(1, 2), (2, 1)],
+            "021D": [(3, 1), (3, 2)],
+            "021U": [(1, 3), (2, 3)],
+            "021C": [(1, 3), (3, 2)],
+            "111D": [(1, 2), (2, 1), (3, 1)],
+            "111U": [(1, 2), (2, 1), (1, 3)],
+            "030T": [(1, 2), (3, 2), (1, 3)],
+            "030C": [(1, 3), (3, 2), (2, 1)],
+            "201": [(1, 2), (2, 1), (3, 1), (1, 3)],
+            "120D": [(1, 2), (2, 1), (3, 1), (3, 2)],
+            "120U": [(1, 2), (2, 1), (1, 3), (2, 3)],
+            "120C": [(1, 2), (2, 1), (1, 3), (3, 2)],
+            "210": [(1, 2), (2, 1), (1, 3), (3, 2), (2, 3)],
+            "300": [(1, 2), (2, 1), (2, 3), (3, 2), (1, 3), (3, 1)],
+        }
+
+        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])
+            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)
+
+    elif graph_mode == "Create Your Own":
+        st.write("### Create Your Own Triads")
+
+        # Let user define custom triads
+        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)]"
+        )
+        
+        # Try to evaluate the input as a list of edges
+        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"},
+                )
+                st.pyplot(fig)
+            else:
+                st.error("Invalid format. Please enter a list of edges in the format [(1, 2), (2, 1), ...].")
+        except Exception as e:
+            st.error(f"Error parsing input: {e}")
+
+# Display the corresponding page based on sidebar option
+if sidebar_option == "Graph: Triads":
+    triads_graph()
+
 def minimum_spanning_tree_graph():
     st.title("Graph: Minimum Spanning Tree")