Update app.py

app.py

@@ -14,25 +14,26 @@ cosmos_client = CosmosClient.from_connection_string(COSMOS_CONNECTION_STRING)
 # Initialize Azure Blob Storage Client
 blob_service = BlobServiceClient.from_connection_string(BLOB_STORAGE_CONNECTION_STRING)
 
-# Function to Get Databases, Containers, and Items
-def get_databases_containers_and_items():
-    all_data = []
+# Function to Retrieve and Display Cosmos DB Structure
+def display_cosmos_db_structure():
+    st.subheader('Azure Cosmos DB Structure')
     for db_properties in cosmos_client.list_databases():
         db_name = db_properties['id']
+        st.markdown(f"#### Database: {db_name}")
         database_client = cosmos_client.get_database_client(db_name)
+        
         for container_properties in database_client.list_containers():
             container_name = container_properties['id']
+            st.markdown(f"- **Container**: {container_name}")
             container_client = database_client.get_container_client(container_name)
+            
             items = list(container_client.read_all_items())
-            all_data.append({'database': db_name, 'container': container_name, 'items': items})
-    return all_data
+            for item in items:
+                st.markdown(f"  - Item: `{item['id']}`")  # Replace 'id' with the appropriate key if different
 
-# Displaying the Data
-all_data = get_databases_containers_and_items()
-for data in all_data:
-    st.subheader(f"Database: {data['database']}, Container: {data['container']}")
-    for item in data['items']:
-        st.json(item)
+# Button to Trigger Display of Cosmos DB Structure
+if st.button('Show Cosmos DB Structure'):
+    display_cosmos_db_structure()
         
 # Streamlit UI
 st.title('Azure Services Integration with Streamlit')