Update app.py

app.py

@@ -61,12 +61,25 @@ def add_or_update_png_images():
             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:
-                    if st.button(f"🗑️ Delete {item_id}", key=f"delete_{item_id}"):
-                        container_client.delete_item(item=item_data, partition_key=item_id)
+                    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