Update app.py

app.py

@@ -3,88 +3,65 @@ import streamlit as st
 from azure.cosmos import CosmosClient, PartitionKey
 from azure.storage.blob import BlobServiceClient
 from azure.cosmos.exceptions import CosmosResourceNotFoundError
-
 import requests
 import glob
+from datetime import datetime
 
-# Environment Variables
+# Initialize Azure Clients
 COSMOS_CONNECTION_STRING = os.getenv('COSMOS_CONNECTION_STRING')
 BLOB_STORAGE_CONNECTION_STRING = os.getenv('BLOB_STORAGE_CONNECTION_STRING')
-
-# Initialize Azure Cosmos DB Client
 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 Retrieve and Display Cosmos DB Structure
-def display_cosmos_db_structure():
+# Function to Delete All Items in a Container
+def delete_all_items_in_container(db_name, container_name):
+    database_client = cosmos_client.get_database_client(db_name)
+    container_client = database_client.get_container_client(container_name)
+    
+    for item in container_client.read_all_items():
+        try:
+            container_client.delete_item(item=item['id'], partition_key=item['id'])
+            st.write(f"Deleted Item: {item['id']}")
+        except CosmosResourceNotFoundError:
+            st.error(f"Item not found: {item['id']}")
+
+
+# Display and Manage Cosmos DB Structure
+def display_and_manage_cosmos_db():
     st.subheader('Azure Cosmos DB Structure')
-    db_properties = next(cosmos_client.list_databases(), None)
-    if db_properties:
+    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)
-        container_properties = next(database_client.list_containers(), None)
-        if container_properties:
+
+        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())
-            for item in items:
+
+            for item in container_client.read_all_items():
                 item_desc = f"  - Item: `{item['id']}`"
-                if 'file_name' in item and item['file_name'].endswith('.png'):
-                    st.markdown(item_desc)
+                st.markdown(item_desc)
+                if 'file_name' in item:
                     st.image(item['file_name'])
-                else:
-                    st.markdown(item_desc)
+                
+                # Update and Delete buttons for each item
+                if st.button(f"🗑️ Delete {item['id']}", key=f"delete_{item['id']}"):
+                    container_client.delete_item(item=item['id'], partition_key=item['id'])
+                    st.success(f"Deleted Item: {item['id']}")
 
-# Button to Trigger Display of Cosmos DB Structure
-if st.button('Show Cosmos DB Structure'):
-    display_cosmos_db_structure()
+# Insert PNG Images with Unique Identifiers
+def insert_png_images_with_unique_ids():
+    db_name = st.selectbox("Select Database", [db['id'] for db in cosmos_client.list_databases()])
+    container_name = st.selectbox("Select Container", [container['id'] for container in cosmos_client.get_database_client(db_name).list_containers()])
+    container_client = cosmos_client.get_database_client(db_name).get_container_client(container_name)
 
-# Function to Add or Update PNG Images
-def add_or_update_png_images():
-    db_properties = next(cosmos_client.list_databases(), None)
-    if db_properties:
-        db_name = db_properties['id']
-        database_client = cosmos_client.get_database_client(db_name)
-        container_properties = next(database_client.list_containers(), None)
-        if container_properties:
-            container_name = container_properties['id']
-            container_client = database_client.get_container_client(container_name)
-            existing_items = list(container_client.read_all_items())
-            existing_ids = {item['id'] for item in existing_items}
-
-            # Add or update PNG files from directory
-            png_files = glob.glob('*.png')
-            for file_name in png_files:
-                item_id = os.path.splitext(file_name)[0]
-                item_data = {"id": item_id, "file_name": file_name}
-            
-                if item_id not in existing_ids:
-                    container_client.create_item(body=item_data)
-                    st.write(f"Added Item: {item_id}")
-                else:
-                    delete_button = st.button(f"🗑️ Delete {item_id}", key=f"delete_{item_id}")
-                    if delete_button:
-                        # Attempt to delete the item
-                        try:
-                            container_client.delete_item(item=item_id, partition_key=item_id)
-                            # Verify deletion by trying to read the deleted item
-                            try:
-                                container_client.read_item(item=item_id, partition_key=item_id)
-                                st.error(f"Failed to delete Item: {item_id}")
-                            except CosmosResourceNotFoundError:
-                                st.success(f"Successfully deleted Item: {item_id}")
-                        except Exception as e:
-                            st.error(f"Error deleting Item: {item_id}. Error: {str(e)}")
-            
-                    st.write(f"Item already exists: {item_id}")
-
-# UI to Add or Update PNG Images
-if st.button('Manage PNG Images'):
-    add_or_update_png_images()
+    png_files = glob.glob('*.png')
+    for file_name in png_files:
+        unique_id = f"{os.path.splitext(file_name)[0]}_{datetime.now().strftime('%Y%m%d%H%M%S')}"
+        item_data = {"id": unique_id, "file_name": file_name}
+        container_client.create_item(body=item_data)
+        st.write(f"Inserted Item: {unique_id}")
 
 # Azure Blob Storage - Upload/Download
 st.subheader('Azure Blob Storage - Upload/Download')
@@ -94,6 +71,7 @@ blob_file = st.file_uploader('Upload file to Blob')
 if blob_file is not None and st.button('Upload to Blob'):
     blob_client = blob_service.get_blob_client(container=blob_container, blob=blob_file.name)
     blob_client.upload_blob(blob_file.getvalue())
+    st.success('File uploaded successfully.')
 
 # Azure Functions - Trigger
 st.subheader('Azure Functions - Trigger')
@@ -101,4 +79,20 @@ function_url = st.text_input('Function URL')
 
 if st.button('Call Azure Function'):
     response = requests.get(function_url)
-    st.write('Function Response:', response.text)
+    st.write('Function Response:', response.text)
+
+
+# Display Cosmos DB Structure and Manage Items
+display_and_manage_cosmos_db()
+
+# Button to Insert PNG Images with Unique Identifiers
+if st.button('Insert PNG Images with Unique IDs'):
+    insert_png_images_with_unique_ids()
+
+# Button to Delete All Items in a Container
+db_name_to_delete = st.selectbox("Select Database to Delete From", [db['id'] for db in cosmos_client.list_databases()])
+container_name_to_delete = st.selectbox("Select Container to Delete From", [container['id'] for container in cosmos_client.get_database_client(db_name_to_delete).list_containers()])
+
+if st.button('Delete All Items in Container'):
+    delete_all_items_in_container(db_name_to_delete, container_name_to_delete)
+    st.success("All items deleted successfully.")