Update app.py

app.py

@@ -170,10 +170,10 @@ def update_record(container, updated_record):
     except Exception as e:
         return False, f"Error: {traceback.format_exc()} 😱"
 
-# 🗑️ Delete record – query first, then delete using the actual partition key
-def delete_record(container, record):
+# 🗑️ Delete record – query first to get the proper partition key value from "testPART"
+def delete_record(container, record, partition_key_field="testPART"):
     try:
-        # Query for the document by id to get the actual partition key value.
+        # Query for the document using its id
         query = "SELECT * FROM c WHERE c.id=@id"
         parameters = [{"name": "@id", "value": record["id"]}]
         docs = list(container.query_items(
@@ -182,13 +182,17 @@ def delete_record(container, record):
             enable_cross_partition_query=True
         ))
         if not docs:
-            # Document doesn't exist; treat as already deleted.
+            # Document not found; treat as already deleted
             return True, f"Record {record['id']} not found (already deleted). 🗑️"
         
-        # Use the partition key value from the retrieved document.
         doc = docs[0]
-        partition_key = doc.get("name")  # assuming 'name' is your partition key
-        container.delete_item(item=doc["id"], partition_key=partition_key)
+        # Retrieve the partition key value from the document (adjust field name as needed)
+        partition_key_value = doc.get(partition_key_field)
+        if partition_key_value is None:
+            return False, f"Partition key field '{partition_key_field}' not found in document {doc['id']}."
+        
+        # Delete the document using its id and the actual partition key value
+        container.delete_item(item=doc["id"], partition_key=partition_key_value)
         return True, f"Record {doc['id']} successfully deleted. 🗑️"
     except exceptions.CosmosHttpResponseError as e:
         return False, f"HTTP error: {str(e)} 🚨"
@@ -196,7 +200,6 @@ def delete_record(container, record):
         return False, f"Unexpected error: {traceback.format_exc()} 😱"
 
 
-
 # 💾 Save a new document to Cosmos DB with extra fields
 def save_to_cosmos_db(container, query, response1, response2):
     try: