Update app.py

app.py

@@ -20,7 +20,8 @@ sidebar_option = st.sidebar.radio("Select an option",
                                   "Drawing: Multipartite Layout", "Drawing: Node Colormap",
                                   "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"])
+                                  "Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation", "3D Drawing: Basic Matplotlib",
+                                  "Graphviz Layout: Circular Tree"])
 
 # Helper function to draw and display graph
 def draw_graph(G, pos=None, title="Graph Visualization"):
@@ -28,6 +29,51 @@ 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)
 
+# Default example code
+def default_example_circular_tree():
+    G = nx.balanced_tree(3, 5)
+    pos = nx.nx_agraph.graphviz_layout(G, prog="twopi", args="")
+    plt.figure(figsize=(8, 8))
+    nx.draw(G, pos, node_size=20, alpha=0.5, node_color="blue", with_labels=False)
+    plt.axis("equal")
+    st.pyplot(plt)
+
+# Create your own graph option
+def create_own_graph_circular_tree():
+    # Input fields to customize the graph
+    branching_factor = st.number_input("Branching Factor", min_value=2, max_value=10, value=3)
+    depth = st.number_input("Depth", min_value=1, max_value=10, value=5)
+    
+    # Add a button to generate the graph
+    generate_button = st.button("Generate Graph")
+
+    if generate_button:
+        # Generate graph and layout
+        G = nx.balanced_tree(branching_factor, depth)
+        pos = nx.nx_agraph.graphviz_layout(G, prog="twopi", args="")
+        
+        # Draw the graph
+        plt.figure(figsize=(8, 8))
+        nx.draw(G, pos, node_size=20, alpha=0.5, node_color="blue", with_labels=False)
+        plt.axis("equal")
+        st.pyplot(plt)
+
+if sidebar_option == "Graphviz Layout: Circular Tree":
+    st.title("Graphviz Layout: Circular Tree")
+
+    # Provide options for Default Example or Custom Graph
+    graph_mode = st.radio(
+        "Choose a Mode:",
+        ("Default Example", "Create Your Own"),
+        help="Default example shows a balanced tree graph, or you can create your own custom graph."
+    )
+
+    # Display the chosen option
+    if graph_mode == "Default Example":
+        default_example_circular_tree()
+    elif graph_mode == "Create Your Own":
+        create_own_graph_circular_tree()
+
 if sidebar_option == "3D Drawing: Animations of 3D Rotation":
     st.title("3D Drawing: Animations of 3D Rotation")