Update app.py

app.py

@@ -14,6 +14,35 @@ 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 Containers and Items
+def get_containers_and_items(database_name):
+    database_client = cosmos_client.get_database_client(database_name)
+    containers_info = []
+    for container_properties in database_client.list_containers():
+        container_name = container_properties['id']
+        container_client = database_client.get_container_client(container_name)
+        items = list(container_client.read_all_items())
+        containers_info.append((container_name, items))
+    return containers_info
+
+# UI to List Containers and Items
+st.subheader('List Database Containers and Items')
+selected_db_for_listing = st.text_input('Enter Database Name to List Containers and Items')
+emoji_button = st.button('🔍 Retrieve Data')
+
+if emoji_button:
+    if selected_db_for_listing:
+        containers_info = get_containers_and_items(selected_db_for_listing)
+        for container_name, items in containers_info:
+            st.subheader(f'Container: {container_name}')
+            for item in items:
+                st.json(item)
+    else:
+        st.error('Please enter a database name.')
+
+
+        
 # Streamlit UI
 st.title('Azure Services Integration with Streamlit')