Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import streamlit as st
-from azure.cosmos import CosmosClient, PartitionKey
+from azure.cosmos import CosmosClient
 from azure.cosmos.exceptions import CosmosResourceNotFoundError
 import glob
 from datetime import datetime
@@ -10,11 +10,11 @@ COSMOS_CONNECTION_STRING = os.getenv('COSMOS_CONNECTION_STRING')
 cosmos_client = CosmosClient.from_connection_string(COSMOS_CONNECTION_STRING)
 
 # Function to Delete Item by ID in Cosmos DB
-def delete_item_by_id(db_name, container_name, item_id):
+def delete_item_by_id(db_name, container_name, item_id, partition_key_value):
     database_client = cosmos_client.get_database_client(db_name)
     container_client = database_client.get_container_client(container_name)
     try:
-        container_client.delete_item(item=item_id, partition_key=item_id)
+        container_client.delete_item(item=item_id, partition_key=partition_key_value)
         st.success(f"Deleted Item: {item_id}")
     except CosmosResourceNotFoundError:
         st.error(f"Item not found: {item_id}")
@@ -29,9 +29,11 @@ def display_and_manage_cosmos_db():
             container_name = container_properties['id']
             container_client = database_client.get_container_client(container_name)
             for item in container_client.read_all_items():
-                st.markdown(f"- **Item ID**: `{item['id']}`")
+                # Assuming 'partitionKey' is the property name of the partition key in the items
+                partition_key_value = item.get('partitionKey', item['id'])
+                st.markdown(f"- **Item ID**: `{item['id']}` - **Partition Key**: `{partition_key_value}`")
                 if st.button(f"🗑️ Delete {item['id']}", key=f"delete_{item['id']}"):
-                    delete_item_by_id(db_name, container_name, item['id'])
+                    delete_item_by_id(db_name, container_name, item['id'], partition_key_value)
 
 # Insert PNG Images with Unique Identifiers
 def insert_png_images_with_unique_ids():