Update app.py

app.py

@@ -60,7 +60,44 @@ def delete_item(db_name, container_name, item_id):
         st.error("Item not found for deletion")
 
 def display_cosmos_db_structure():
-    # Existing function logic here
+    st.subheader('Azure Cosmos DB Structure')
+
+    # Fetching databases
+    databases = list_databases()
+    if not databases:
+        st.error("No databases found.")
+        return
+
+    # Iterating through databases
+    for db_properties in databases:
+        db_name = db_properties['id']
+        st.markdown(f"#### Database: {db_name}")
+
+        # Fetching containers in the database
+        containers = list_containers(db_name)
+        if not containers:
+            st.markdown("No containers found in this database.")
+            continue
+
+        # Iterating through containers
+        for container_properties in containers:
+            container_name = container_properties['id']
+            st.markdown(f"- **Container**: {container_name}")
+
+            # Fetching and displaying items in the container
+            items = list_items(db_name, container_name)
+            if not items:
+                st.markdown("  - No items in this container.")
+                continue
+
+            for item in items:
+                item_desc = f"  - Item: `{item['id']}`"
+                if 'file_name' in item and item['file_name'].endswith('.png'):
+                    st.markdown(item_desc)
+                    st.image(item['file_name'])
+                else:
+                    st.markdown(item_desc)
+
 
 def manage_png_images():
     # Get database and container names