Update app.py

app.py

@@ -170,15 +170,14 @@ def update_record(container, updated_record):
     except Exception as e:
         return False, f"Error: {traceback.format_exc()} 😱"
 
-# 🗑️ Delete a record from Cosmos DB
-def delete_record(container, record, partition_key_field=None):
+# 🗑️ Delete a record from Cosmos DB with auto-detected partition key
+def delete_record(container, record):
     """
-    Delete a record from Cosmos DB using its ID and partition key.
+    Delete a record from Cosmos DB using its ID and an auto-detected partition key.
     
     Args:
         container: Cosmos DB container client
-        record: Dictionary containing at least 'id' and optionally the partition key field
-        partition_key_field: Name of the field containing the partition key value (optional)
+        record: Dictionary containing at least 'id' and potentially a partition key field
     
     Returns:
         tuple: (success: bool, message: str)
@@ -189,33 +188,38 @@ def delete_record(container, record, partition_key_field=None):
             return False, "Record must contain an 'id' field. 🛑"
 
         doc_id = record["id"]
+        st.write(f"Document to delete: {json.dumps(record, indent=2)}")  # Debug output
 
-        # Determine partition key value
-        if partition_key_field:
-            partition_key_value = record.get(partition_key_field)
-            if partition_key_value is None:
-                return False, f"Partition key field '{partition_key_field}' not found in record {doc_id}. 🛑"
-        else:
-            # Default to ID if no partition key field is specified
+        # Auto-detect partition key from common fields
+        # List of potential partition key fields (customize if needed)
+        potential_partition_keys = ["partitionKey", "category", "type", "name", "testPART"]
+        
+        # Check if any potential partition key field exists in the document
+        partition_key_value = None
+        partition_key_field = None
+        for field in potential_partition_keys:
+            if field in record:
+                partition_key_value = record[field]
+                partition_key_field = field
+                break
+        
+        # If no specific partition key field is found, default to 'id' (matches your /id setup)
+        if partition_key_value is None:
             partition_key_value = doc_id
+            partition_key_field = "id (default)"
 
-        # Debug info
-        st.write(f"Attempting to delete: ID={doc_id}, Partition Key={partition_key_value}")
+        st.write(f"Detected Partition Key Field: {partition_key_field}, Value: {partition_key_value}")  # Debug
 
         # Perform the deletion
         container.delete_item(item=doc_id, partition_key=partition_key_value)
         return True, f"Record {doc_id} successfully deleted from Cosmos DB. 🗑️"
 
     except exceptions.CosmosResourceNotFoundError:
-        # Document doesn't exist, which is fine for a delete operation
         return True, f"Record {doc_id} not found in Cosmos DB (already deleted or never existed). 🗑️"
     except exceptions.CosmosHttpResponseError as e:
-        # Other HTTP errors (e.g., wrong partition key, permissions)
         return False, f"HTTP error deleting {doc_id}: {str(e)}. 🚨"
     except Exception as e:
-        # Unexpected errors with full traceback
         return False, f"Unexpected error deleting {doc_id}: {str(traceback.format_exc())}. 😱"
-        
 
 
 # 💾 Save a new document to Cosmos DB with extra fields