Update app.py

app.py

@@ -1,35 +1,28 @@
 import streamlit as st
-from diagrams import Cluster, Diagram
-from diagrams.azure.compute import ContainerInstances
-from diagrams.azure.integration import APIManagement
-from diagrams.onprem.container import Docker
-#from diagrams.onprem.database import Snowflake
-from diagrams.programming.framework import FastAPI
-from diagrams.custom import Custom
+from diagrams import Diagram, Cluster
+from diagrams.azure.network import VirtualNetwork, LoadBalancer, TrafficManager
 
-# Define the diagram
-def create_diagram():
-    with Diagram("Architecture Diagram", show=False):
-        with Cluster("Azure"):
-            container_registry = ContainerInstances("Azure Container Registry")
-            container_registry_api = APIManagement("ACR API")
+def main():
+    st.set_page_config(page_title="Azure Network Diagram Demo")
 
-        with Cluster("Docker"):
-            docker = Docker("Docker")
-            slimbuster = Custom("Slim Buster", "./slimbuster.png")
+    with st.sidebar:
+        st.title("Azure Network Diagram Demo")
 
-        with Cluster("Python"):
-            python = FastAPI("Python / Uvicorn")
+    with st.container():
+        st.header("Azure Network Diagram")
 
-        #with Cluster("Snowflake"):
-        #    snowflake = Snowflake("Snowflake")
+        with Diagram("Azure Network Diagram", show=False):
+            with Cluster("Azure"):
+                vn = VirtualNetwork("Virtual Network")
+                lb = LoadBalancer("Load Balancer")
+                tm = TrafficManager("Traffic Manager")
 
-        container_registry >> docker >> slimbuster >> python
-        container_registry_api >> docker
-        python >> snowflake
+            vn >> lb >> tm
 
-    return "architecture_diagram.png"
+        st.markdown("#### Legend")
+        st.markdown("- Virtual Network: A virtual network to isolate your Azure resources.")
+        st.markdown("- Load Balancer: A traffic distribution tool for Azure resources.")
+        st.markdown("- Traffic Manager: A DNS-based traffic routing service.")
 
-# Create and show the diagram in Streamlit
-diagram = create_diagram()
-st.image(diagram, caption="Architecture Diagram", use_column_width=True)
+if __name__ == '__main__':
+    main()