Update app.py

app.py

@@ -14,44 +14,25 @@ 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 in Predefined Databases
-st.subheader('List Containers and Items in Databases')
-
-databases_to_list = ["ToDoList", "CopilotSampleDb"]
-for db_name in databases_to_list:
-    st.markdown(f"### Database: {db_name}")
-    containers_info = get_containers_and_items(db_name)
-    for container_name, items in containers_info:
-        st.subheader(f'Container: {container_name}')
-        for item in items:
-            st.json(item)
-
-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.')
-
-
+# Function to Get Databases, Containers, and Items
+def get_databases_containers_and_items():
+    all_data = []
+    for db_properties in cosmos_client.list_databases():
+        db_name = db_properties['id']
+        database_client = cosmos_client.get_database_client(db_name)
+        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())
+            all_data.append({'database': db_name, 'container': container_name, 'items': items})
+    return all_data
+
+# 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)
         
 # Streamlit UI
 st.title('Azure Services Integration with Streamlit')