Update app.py

app.py

@@ -174,17 +174,33 @@ def update_record(container, updated_record):
         return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
 
 # 🗑️ Delete record - Saying goodbye to data (it's not you, it's me)
-def delete_record(container, name, id):
+def delete_record(container, id):
     try:
-        container.delete_item(item=id, partition_key=id)
-        return True, f"Successfully deleted record with name: {name} and id: {id} 🗑️"
-    except exceptions.CosmosResourceNotFoundError:
-        return False, f"Record with id {id} not found. It may have been already deleted. 🕵️‍♂️"
+        # First try to read the document with id as both item id and partition key
+        try:
+            doc = container.read_item(item=id, partition_key=id)
+        except exceptions.CosmosResourceNotFoundError:
+            # If that fails, query to find the document and get its partition key
+            query = f"SELECT * FROM c WHERE c.id = '{id}'"
+            items = list(container.query_items(
+                query=query,
+                enable_cross_partition_query=True
+            ))
+            if not items:
+                return False, f"Record with id {id} not found. 🕵️‍♂️"
+            doc = items[0]
+        
+        # Now delete with the correct partition key
+        partition_key = doc.get('id', id)  # Default to id if no partition key found
+        container.delete_item(item=id, partition_key=partition_key)
+        return True, f"Successfully deleted record with id: {id} 🗑️"
+        
     except exceptions.CosmosHttpResponseError as e:
         return False, f"HTTP error occurred: {str(e)} 🚨"
     except Exception as e:
-        return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
+        return False, f"An unexpected error occurred: {str(e)} 😱"
 
+        
 # 💾 Save to Cosmos DB - Preserving data for future generations (or just until the next update)
 def save_to_cosmos_db(container, query, response1, response2):
     try:
@@ -517,27 +533,6 @@ def preprocess_text(text):
     text = text.strip()
     return text
 
-
-
-# 🗑️ Delete record - Handling partition key correctly
-def delete_record(container, id):
-    try:
-        # Retrieve the document to ensure it exists and get the partition key
-        doc = container.read_item(item=id, partition_key=id)  # Assuming 'id' is the partition key
-        # If your partition key is something else, you need to pass it explicitly here:
-        # doc = container.read_item(item=id, partition_key=doc['partition_key_field'])
-
-        # Now proceed with the deletion
-        container.delete_item(item=id, partition_key=doc['id'])  # If 'id' is partition key
-        return True, f"Successfully deleted record with id: {id} 🗑️"
-    except exceptions.CosmosResourceNotFoundError:
-        return False, f"Record with id {id} not found. It may have been already deleted. 🕵️‍♂️"
-    except exceptions.CosmosHttpResponseError as e:
-        return False, f"HTTP error occurred: {str(e)} 🚨"
-    except Exception as e:
-        return False, f"An unexpected error occurred: {traceback.format_exc()} 😱"
-
-
                 
 # 🎭 Main function - "All the world's a stage, and all the code merely players" -Shakespeare, probably
 def main():