CyberAI-VM-Project/vm_visual_graph.py at main · machphy/CyberAI-VM-Project · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
# vm_visual_graph.py

import matplotlib.pyplot as plt
import networkx as nx

# Create a directed graph
G = nx.DiGraph()

# Add nodes
G.add_node("User")
G.add_node("Web Dashboard")
G.add_node("VM1: Linux (AI)")
G.add_node("VM2: Windows")
G.add_node("Security Scan")
G.add_node("AI Model")

# Add connections
G.add_edges_from([
    ("User", "Web Dashboard"),
    ("Web Dashboard", "VM1: Linux (AI)"),
    ("Web Dashboard", "VM2: Windows"),
    ("VM1: Linux (AI)", "Security Scan"),
    ("VM1: Linux (AI)", "AI Model"),
    ("Security Scan", "VM2: Windows"),
    ("AI Model", "Web Dashboard")
])

# Position nodes
pos = nx.spring_layout(G, seed=42)

# Draw nodes and edges
plt.figure(figsize=(10, 6))
nx.draw_networkx_nodes(G, pos, node_size=2000, node_color="skyblue")
nx.draw_networkx_edges(G, pos, arrowstyle='->', arrowsize=20, edge_color='gray')
nx.draw_networkx_labels(G, pos, font_size=10, font_weight="bold")

plt.title("🚀 VM Activity Visual: CyberSentinel")
plt.axis('off')
plt.tight_layout()
plt.show()