app.py

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

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

        with Cluster("Docker"):
            docker = Docker("Docker")
            slimbuster = Custom("Slim Buster", "./slimbuster.png")

        with Cluster("Python"):
            python = FastAPI("Python / Uvicorn")

        with Cluster("Snowflake"):
            snowflake = Snowflake("Snowflake")

        container_registry >> docker >> slimbuster >> python
        container_registry_api >> docker
        python >> snowflake

    return "architecture_diagram.png"

# Create and show the diagram in Streamlit
diagram = create_diagram()
st.image(diagram, caption="Architecture Diagram", use_column_width=True)