Update app.py

app.py

@@ -176,31 +176,28 @@ 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:
-        # Query to find the document
-        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. 🕵️‍♂️"
+        # Direct read attempt first
+        try:
+            doc = container.read_item(item=id, partition_key=id)
+            container.delete_item(item=id, partition_key=id)
+            return True, f"Successfully deleted record with id: {id} 🗑️"
+        except exceptions.CosmosResourceNotFoundError:
+            # Fall back to cross-partition query
+            query = "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]
+            container.delete_item(item=id, partition_key=doc['id'])
+            return True, f"Successfully deleted record with id: {id} 🗑️"
             
-        doc = items[0]
-        partition_key = doc.get('id')  # Using id as partition key
-        
-        # Delete with partition key
-        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: