Update app.py
Browse files
app.py
CHANGED
|
@@ -20,8 +20,7 @@ sidebar_option = st.sidebar.radio("Select an option",
|
|
| 20 |
"Drawing: Multipartite Layout", "Drawing: Node Colormap",
|
| 21 |
"Drawing: Rainbow Coloring", "Drawing: Random Geometric Graph","Drawing: Self-loops",
|
| 22 |
"Drawing: Simple Path", "Drawing: Spectral Embedding", "Drawing: Traveling Salesman Problem",
|
| 23 |
-
"Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation", "3D Drawing: Basic Matplotlib"
|
| 24 |
-
"Graphviz Layout: Lanl Routes"])
|
| 25 |
|
| 26 |
# Helper function to draw and display graph
|
| 27 |
def draw_graph(G, pos=None, title="Graph Visualization"):
|
|
@@ -29,91 +28,6 @@ def draw_graph(G, pos=None, title="Graph Visualization"):
|
|
| 29 |
nx.draw(G, pos=pos, with_labels=True, node_color='lightblue', node_size=500, font_size=10, font_weight='bold')
|
| 30 |
st.pyplot(plt)
|
| 31 |
|
| 32 |
-
# Default example: Lanl Routes graph
|
| 33 |
-
def lanl_graph():
|
| 34 |
-
"""Return the LANL internet view graph from lanl.edges"""
|
| 35 |
-
try:
|
| 36 |
-
fh = open("lanl_routes.edgelist")
|
| 37 |
-
except OSError:
|
| 38 |
-
print("lanl.edges not found")
|
| 39 |
-
raise
|
| 40 |
-
|
| 41 |
-
G = nx.Graph()
|
| 42 |
-
|
| 43 |
-
time = {}
|
| 44 |
-
time[0] = 0 # assign 0 to center node
|
| 45 |
-
for line in fh.readlines():
|
| 46 |
-
(head, tail, rtt) = line.split()
|
| 47 |
-
G.add_edge(int(head), int(tail))
|
| 48 |
-
time[int(head)] = float(rtt)
|
| 49 |
-
|
| 50 |
-
# get largest component and assign ping times to G0time dictionary
|
| 51 |
-
Gcc = sorted(nx.connected_components(G), key=len, reverse=True)[0]
|
| 52 |
-
G0 = G.subgraph(Gcc)
|
| 53 |
-
G0.rtt = {}
|
| 54 |
-
for n in G0:
|
| 55 |
-
G0.rtt[n] = time[n]
|
| 56 |
-
|
| 57 |
-
return G0
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
# Streamlit Layout for Graphviz Layout: Lanl Routes
|
| 61 |
-
def graphviz_layout_lanl_routes():
|
| 62 |
-
st.title("Graphviz Layout: Lanl Routes")
|
| 63 |
-
|
| 64 |
-
# Sidebar selection for Default Example or Custom Graph
|
| 65 |
-
graph_mode = st.radio(
|
| 66 |
-
"Choose a Mode:",
|
| 67 |
-
("Default Example", "Create Your Own"),
|
| 68 |
-
help="Default example shows LANL routes, or you can create your own custom graph."
|
| 69 |
-
)
|
| 70 |
-
|
| 71 |
-
if graph_mode == "Default Example":
|
| 72 |
-
# Load LANL graph and visualize
|
| 73 |
-
G = lanl_graph()
|
| 74 |
-
|
| 75 |
-
st.write("Graph Summary:")
|
| 76 |
-
st.write(G)
|
| 77 |
-
st.write(f"{nx.number_connected_components(G)} connected components")
|
| 78 |
-
|
| 79 |
-
# Create and display the graph
|
| 80 |
-
plt.figure(figsize=(8, 8))
|
| 81 |
-
# Use graphviz to find radial layout
|
| 82 |
-
pos = nx.nx_agraph.graphviz_layout(G, prog="twopi", root=0)
|
| 83 |
-
|
| 84 |
-
# Draw nodes, coloring by rtt ping time
|
| 85 |
-
options = {"with_labels": False, "alpha": 0.5, "node_size": 15}
|
| 86 |
-
nx.draw(G, pos, node_color=[G.rtt[v] for v in G], **options)
|
| 87 |
-
|
| 88 |
-
# Adjust the plot limits
|
| 89 |
-
xmax = 1.02 * max(xx for xx, yy in pos.values())
|
| 90 |
-
ymax = 1.02 * max(yy for xx, yy in pos.values())
|
| 91 |
-
plt.xlim(0, xmax)
|
| 92 |
-
plt.ylim(0, ymax)
|
| 93 |
-
st.pyplot(plt)
|
| 94 |
-
|
| 95 |
-
elif graph_mode == "Create Your Own":
|
| 96 |
-
st.write("### Custom Graph Creation")
|
| 97 |
-
|
| 98 |
-
# Input form for creating custom graph
|
| 99 |
-
num_nodes = st.slider("Number of Nodes", min_value=2, max_value=50, value=10)
|
| 100 |
-
num_edges = st.slider("Number of Edges", min_value=1, max_value=num_nodes*(num_nodes-1)//2, value=10)
|
| 101 |
-
|
| 102 |
-
# Generate custom graph
|
| 103 |
-
G_custom = nx.gnm_random_graph(num_nodes, num_edges)
|
| 104 |
-
|
| 105 |
-
st.write("Generated Custom Graph:")
|
| 106 |
-
st.write(G_custom)
|
| 107 |
-
|
| 108 |
-
# Use graphviz layout
|
| 109 |
-
pos = nx.nx_agraph.graphviz_layout(G_custom, prog="twopi")
|
| 110 |
-
nx.draw(G_custom, pos, with_labels=True, node_size=500, node_color="lightblue", font_size=10)
|
| 111 |
-
st.pyplot(plt)
|
| 112 |
-
|
| 113 |
-
# Display the corresponding page based on sidebar option
|
| 114 |
-
if sidebar_option == "Graphviz Layout: Lanl Routes":
|
| 115 |
-
graphviz_layout_lanl_routes()
|
| 116 |
-
|
| 117 |
if sidebar_option == "3D Drawing: Animations of 3D Rotation":
|
| 118 |
st.title("3D Drawing: Animations of 3D Rotation")
|
| 119 |
|
|
|
|
| 20 |
"Drawing: Multipartite Layout", "Drawing: Node Colormap",
|
| 21 |
"Drawing: Rainbow Coloring", "Drawing: Random Geometric Graph","Drawing: Self-loops",
|
| 22 |
"Drawing: Simple Path", "Drawing: Spectral Embedding", "Drawing: Traveling Salesman Problem",
|
| 23 |
+
"Drawing: Weighted Graph", "3D Drawing: Animations of 3D Rotation", "3D Drawing: Basic Matplotlib"])
|
|
|
|
| 24 |
|
| 25 |
# Helper function to draw and display graph
|
| 26 |
def draw_graph(G, pos=None, title="Graph Visualization"):
|
|
|
|
| 28 |
nx.draw(G, pos=pos, with_labels=True, node_color='lightblue', node_size=500, font_size=10, font_weight='bold')
|
| 29 |
st.pyplot(plt)
|
| 30 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 31 |
if sidebar_option == "3D Drawing: Animations of 3D Rotation":
|
| 32 |
st.title("3D Drawing: Animations of 3D Rotation")
|
| 33 |
|