Update app.py

app.py

@@ -192,29 +192,29 @@ def delete_record(container, record, partition_key_field=None):
 
         # Determine partition key value
         if partition_key_field:
-            # Use specified partition key field from the record
             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:
-            # If no partition key field specified, try to use the ID as the partition key
-            # (common when partition key is same as ID)
+            # Default to ID if no partition key field is specified
             partition_key_value = doc_id
 
+        # Debug info
+        st.write(f"Attempting to delete: ID={doc_id}, Partition Key={partition_key_value}")
+
         # Perform the deletion
         container.delete_item(item=doc_id, partition_key=partition_key_value)
-        return True, f"Record {doc_id} successfully deleted. 🗑️"
+        return True, f"Record {doc_id} successfully deleted from Cosmos DB. 🗑️"
 
     except exceptions.CosmosResourceNotFoundError:
-        # Document doesn't exist, treat as success since goal is deletion
-        return True, f"Record {doc_id} not found (already deleted?). 🗑️"
+        # 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:
-        # Specific HTTP errors (e.g., wrong partition key)
+        # 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())}. 😱"
-
         
 
 
@@ -637,13 +637,14 @@ def main():
                                 if st.button("🗑️ Delete", key=f'delete_{st.session_state.current_index}'):
                                     try:
                                         current_doc = json.loads(doc_str)
-                                        doc_id = current_doc.get("id")
-                                        if not doc_id:
-                                            st.error("Missing ID")
-                                            return
-                                        container.delete_item(item=doc_id, partition_key=doc_id)
-                                        st.success(f"Deleted {doc_id}")
-                                        st.rerun()
+                                        # Adjust this to your actual partition key field (e.g., "testPART", "category", or None if ID is the partition key)
+                                        partition_key_field = "testPART"  # Confirm this matches your container's partition key
+                                        success, message = delete_record(container, current_doc, partition_key_field)
+                                        if success:
+                                            st.success(message)  # Display as success even if not found
+                                            st.rerun()
+                                        else:
+                                            st.error(message)
                                     except Exception as e:
                                         st.error(f"Delete err: {str(e)}")
                     elif selected_view == 'Run AI':