Update app.py

app.py

@@ -176,25 +176,32 @@ def update_record(container, updated_record):
 # 🗑️ Delete record - Saying goodbye to data (it's not you, it's me)
 def delete_record(container, id):
     try:
-        # 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]
+        # First try to find the document using a query
+        query = f"SELECT * FROM c WHERE c.id = @id"
+        params = [{"name": "@id", "value": id}]
+        items = list(container.query_items(
+            query=query,
+            parameters=params,
+            enable_cross_partition_query=True
+        ))
+        
+        if not items:
+            return False, f"Record with id {id} not found. 🕵️‍♂️"
+            
+        doc = items[0]
+        partition_key = doc.get('id')  # Using id as partition key
+        
+        # Delete with explicit partition key
+        container.delete_item(
+            item=id,
+            partition_key=partition_key,
+            enable_cross_partition_query=True
+        )
         
-        # 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.CosmosResourceNotFoundError:
+        return False, f"Record with id {id} not found or already deleted. 🕵️‍♂️"
     except exceptions.CosmosHttpResponseError as e:
         return False, f"HTTP error occurred: {str(e)} 🚨"
     except Exception as e: