updated commit 006

utils/visualization.py

@@ -1,13 +1,62 @@
+# utils/visualization.py
 
-import matplotlib.pyplot as plt
+import plotly.graph_objs as go
 import networkx as nx
+import random
 
 def plot_network(nodes, path):
-    G = nx.DiGraph()
-    for i in range(len(nodes)-1):
-        G.add_edge(nodes[i], nodes[i+1])
-    color_map = ['red' if node in path else 'lightblue' for node in nodes]
-    pos = nx.spring_layout(G)
-    plt.figure(figsize=(8,5))
-    nx.draw(G, pos, with_labels=True, node_color=color_map, node_size=1000, font_size=10, arrows=True)
-    return plt.gcf()
+    G = nx.Graph()
+
+    for node in nodes:
+        G.add_node(node)
+
+    for i in range(len(path) - 1):
+        G.add_edge(path[i], path[i + 1])
+
+    # Random 3D coordinates for visual effect
+    pos = {node: (random.uniform(-10, 10), random.uniform(-10, 10), random.uniform(-10, 10)) for node in G.nodes}
+
+    edge_trace = go.Scatter3d(
+        x=[],
+        y=[],
+        z=[],
+        line=dict(width=4, color='gray'),
+        hoverinfo='none',
+        mode='lines'
+    )
+
+    for edge in G.edges():
+        x0, y0, z0 = pos[edge[0]]
+        x1, y1, z1 = pos[edge[1]]
+        edge_trace['x'] += (x0, x1, None)
+        edge_trace['y'] += (y0, y1, None)
+        edge_trace['z'] += (z0, z1, None)
+
+    node_trace = go.Scatter3d(
+        x=[pos[node][0] for node in G.nodes],
+        y=[pos[node][1] for node in G.nodes],
+        z=[pos[node][2] for node in G.nodes],
+        mode='markers+text',
+        text=list(G.nodes),
+        textposition="top center",
+        marker=dict(
+            size=10,
+            color='skyblue',
+            line=dict(width=1)
+        ),
+        hoverinfo='text'
+    )
+
+    fig = go.Figure(data=[edge_trace, node_trace])
+    fig.update_layout(
+        title="3D Network Path Visualization",
+        showlegend=False,
+        margin=dict(l=0, r=0, b=0, t=40),
+        scene=dict(
+            xaxis=dict(showbackground=False),
+            yaxis=dict(showbackground=False),
+            zaxis=dict(showbackground=False)
+        )
+    )
+
+    return fig