import streamlit as st
from diagrams import Diagram, Cluster
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")

    with st.sidebar:
        st.title("Azure Network Diagram Demo")

    with st.container():
        st.header("Azure Network Diagram")

        with Diagram("Azure Network Diagram", show=False):
            with Cluster("Azure"):
                vn = VirtualNetworks("Virtual Network")
                lb = LoadBalancers("Load Balancer")
                tm = TrafficManagers("Traffic Manager")

            vn >> lb >> tm

        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.")

if __name__ == '__main__':
    main()