Update app.py

app.py

@@ -43,24 +43,24 @@ def display_cosmos_db_structure():
 # Function to Add or Update an Item
 def add_or_update_item(database_name, container_name, item_id, item_data):
     container = cosmos_client.get_database_client(database_name).get_container_client(container_name)
+
+    # Check if item exists
     query = "SELECT * FROM c WHERE c.id = @id"
     parameters = [{"name": "@id", "value": item_id}]
     items = list(container.query_items(query=query, parameters=parameters, enable_cross_partition_query=True))
 
     if items:
-        # Item exists, update it
-        existing_item = items[0]
-        existing_item.update(item_data)
-        container.replace_item(item=existing_item, body=existing_item)
-        st.write(f"Updated Item: {item_id}")
-    else:
-        # Item does not exist, create new
-        container.create_item(body=item_data)
-        st.write(f"Created New Item: {item_id}")
+        # Item exists, delete it
+        container.delete_item(item=items[0], partition_key=items[0]['partitionKeyField']) # Replace 'partitionKeyField' with your actual partition key field
+
+    # Add new item
+    container.create_item(body=item_data)
+    st.write(f"Added Item: {item_id}")
 
     # Display item ID and image if available
     display_item(database_name, container_name, item_id)
 
+
 # Function to Display an Item
 def display_item(database_name, container_name, item_id):
     container = cosmos_client.get_database_client(database_name).get_container_client(container_name)