Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from diagrams import Diagram, Cluster
-from diagrams.azure.network import VirtualNetwork, LoadBalancer, TrafficManager
+from diagrams.azure.compute import VirtualNetworks
+from diagrams.azure.network import LoadBalancers, TrafficManagers
 
 def main():
     st.set_page_config(page_title="Azure Network Diagram Demo")
@@ -13,9 +14,9 @@ def main():
 
         with Diagram("Azure Network Diagram", show=False):
             with Cluster("Azure"):
-                vn = VirtualNetwork("Virtual Network")
-                lb = LoadBalancer("Load Balancer")
-                tm = TrafficManager("Traffic Manager")
+                vn = VirtualNetworks("Virtual Network")
+                lb = LoadBalancers("Load Balancer")
+                tm = TrafficManagers("Traffic Manager")
 
             vn >> lb >> tm