updated commit 009

app.py

@@ -14,11 +14,11 @@ def process_alarm(alarm_text, source_node):
 
 # Define example inputs
 examples = [
-    ["Power failure at base station", "Node1"],
-    ["Signal lost between node and switch", "Node3"],
-    ["Configuration mismatch in transmission settings", "Node5"],
-    ["Hardware fault detected in the antenna module", "Node6"],
-    ["Abnormal reboot of base transceiver station", "Node2"]
+    ["Power failure detected at BTS1", "BTS1"],
+    ["Signal loss between BTS2 and BSC1", "BTS2"],
+    ["Link instability reported at BSC2", "BSC2"],
+    ["Transmission error from MSC1 to CoreNetwork", "MSC1"],
+    ["Device reboot alert from BTS3", "BTS3"]
 ]
 
 iface = gr.Interface(

data/network_topology.csv

@@ -1,6 +1,7 @@
-Node
-Node1
-Node2
-Node3
-Node4
-CoreNetwork
+Source,Target
+BTS1,BSC1
+BTS2,BSC1
+BTS3,BSC2
+BSC1,MSC1
+BSC2,MSC1
+MSC1,CoreNetwork

model/trace_analyzer.py

@@ -1,16 +1,24 @@
-
 import pandas as pd
+import networkx as nx
 
 def analyze_trace(source_node):
     df = pd.read_csv("data/network_topology.csv")
-    nodes = df['Node'].tolist()
-    path = []
-    found = False
-    for i, node in enumerate(nodes):
-        if node == source_node:
-            path = nodes[i:]
-            found = True
-            break
-    if not found:
-        path = ["Unknown"]
+    
+    # Build directed graph from the CSV
+    G = nx.from_pandas_edgelist(df, source='Source', target='Target', create_using=nx.DiGraph())
+
+    # Get all nodes
+    nodes = list(G.nodes())
+
+    # Default: no path found
+    path = ["Unknown"]
+
+    # Attempt to find path to CoreNetwork
+    target_node = "CoreNetwork"
+    if source_node in G and target_node in G:
+        try:
+            path = nx.shortest_path(G, source=source_node, target=target_node)
+        except nx.NetworkXNoPath:
+            path = ["No path to CoreNetwork"]
+
     return path, nodes

utils/visualization.py

@@ -1,13 +1,52 @@
-
-import matplotlib.pyplot as plt
+import plotly.graph_objects as go
 import networkx as nx
+import random
 
 def plot_network(nodes, path):
+    # Create a graph
     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()
+    for i in range(len(nodes) - 1):
+        G.add_edge(nodes[i], nodes[i + 1])
+
+    # Assign 3D positions
+    pos = {node: (random.uniform(-1, 1), random.uniform(-1, 1), random.uniform(-1, 1)) for node in G.nodes()}
+
+    edge_trace = []
+    for edge in G.edges():
+        x0, y0, z0 = pos[edge[0]]
+        x1, y1, z1 = pos[edge[1]]
+        color = "red" if edge[0] in path and edge[1] in path and path.index(edge[1]) - path.index(edge[0]) == 1 else "gray"
+        edge_trace.append(go.Scatter3d(
+            x=[x0, x1, None], y=[y0, y1, None], z=[z0, z1, None],
+            mode='lines',
+            line=dict(color=color, width=6 if color == "red" else 2),
+            hoverinfo='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',
+        marker=dict(
+            size=10,
+            color=['red' if node in path else 'lightblue' for node in G.nodes()],
+            opacity=0.8
+        ),
+        text=list(G.nodes()),
+        textposition="top center",
+        hoverinfo='text'
+    )
+
+    fig = go.Figure(data=edge_trace + [node_trace])
+    fig.update_layout(
+        title="Telecom Network Topology - 3D View",
+        showlegend=False,
+        scene=dict(
+            xaxis=dict(showbackground=False),
+            yaxis=dict(showbackground=False),
+            zaxis=dict(showbackground=False)
+        ),
+        margin=dict(l=0, r=0, b=0, t=40)
+    )
+    return fig