Update app.py

app.py

@@ -11,7 +11,8 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Basic: Simple graph Directed", "Drawing: Custom Node Position",
                                   "Drawing: Cluster Layout", "Drawing: Degree Analysis",
                                   "Drawing: Ego Graph", "Drawing: Eigenvalues", "Drawing: Four Grids",
-                                  "Drawing: House With Colors", "Drawing: Labels And Colors"])
+                                  "Drawing: House With Colors", "Drawing: Labels And Colors",
+                                  "Drawing: Plotting MultiDiGraph Edges and Labels"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -19,6 +20,83 @@ 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)
 
+# Function to display Drawing: Plotting MultiDiGraph Edges and Labels
+def draw_labeled_multigraph(G, attr_name, ax=None):
+    """
+    Length of connectionstyle must be at least that of a maximum number of edges
+    between pair of nodes. This number is maximum one-sided connections
+    for directed graph and maximum total connections for undirected graph.
+    """
+    # Works with arc3 and angle3 connectionstyles
+    connectionstyle = [f"arc3,rad={r}" for r in it.accumulate([0.15] * 4)]
+    pos = nx.shell_layout(G)
+    nx.draw_networkx_nodes(G, pos, ax=ax)
+    nx.draw_networkx_labels(G, pos, font_size=20, ax=ax)
+    nx.draw_networkx_edges(
+        G, pos, edge_color="grey", connectionstyle=connectionstyle, ax=ax
+    )
+
+    labels = {
+        tuple(edge): f"{attr_name}={attrs[attr_name]}"
+        for *edge, attrs in G.edges(keys=True, data=True)
+    }
+    nx.draw_networkx_edge_labels(
+        G,
+        pos,
+        labels,
+        connectionstyle=connectionstyle,
+        label_pos=0.3,
+        font_color="blue",
+        bbox={"alpha": 0},
+        ax=ax,
+    )
+
+# Function to display Drawing: Plotting MultiDiGraph Edges and Labels
+def display_multidigraph_edges_labels():
+    st.title("Drawing: Plotting MultiDiGraph Edges and Labels")
+
+    option = st.radio("Choose a graph type:", ("Default Example", "Create your own"))
+
+    if option == "Default Example":
+        # Generate a multi directed graph as an example
+        nodes = "ABC"
+        prod = list(it.product(nodes, repeat=2))
+        pair_dict = {f"Product x {i}": prod * i for i in range(1, 5)}
+
+        fig, axes = plt.subplots(2, 2)
+        for (name, pairs), ax in zip(pair_dict.items(), np.ravel(axes)):
+            G = nx.MultiDiGraph()
+            for i, (u, v) in enumerate(pairs):
+                G.add_edge(u, v, w=round(i / 3, 2))
+            draw_labeled_multigraph(G, "w", ax)
+            ax.set_title(name)
+        fig.tight_layout()
+        st.pyplot(plt)
+
+    elif option == "Create your own":
+        # Let the user input the nodes and edges of the graph
+        st.write("Enter the nodes and edges to create your own labeled MultiDiGraph.")
+
+        nodes = st.text_area("Enter node labels (comma-separated, e.g., a,b,c,d):", value="a,b,c,d")
+        node_labels = nodes.split(',')
+
+        edges = st.text_area("Enter edges (format: node1-node2, comma-separated, e.g., a-b,b-c):", value="a-b,b-c,c-d")
+        edge_list = [tuple(edge.split('-')) for edge in edges.split(',')]
+
+        # Generate graph based on user input
+        G_custom = nx.MultiDiGraph()
+        for i, (u, v) in enumerate(edge_list):
+            G_custom.add_edge(u, v, w=round(i / 3, 2))
+
+        # Draw the graph
+        fig, ax = plt.subplots()
+        draw_labeled_multigraph(G_custom, "w", ax)
+        st.pyplot(plt)
+
+# Display Drawing: Plotting MultiDiGraph Edges and Labels if selected
+if sidebar_option == "Drawing: Plotting MultiDiGraph Edges and Labels":
+    display_multidigraph_edges_labels()
+
 # Function to display Drawing: Labels And Colors
 def display_labels_and_colors():
     st.title("Drawing: Labels And Colors")