Update database.py

database.py

@@ -281,7 +281,7 @@ class DogDatabase:
         return self.cursor.lastrowid  # THIS LINE - make sure it returns the image_id
     
     def get_dog_images(self, dog_id: int, validated_only: bool = False,
-                      include_discarded: bool = False) -> List[Dict]:
+                  include_discarded: bool = False) -> List[Dict]:
         """Get all images for a dog"""
         query = "SELECT * FROM dog_images WHERE dog_id = ?"
         params = [dog_id]
@@ -294,9 +294,10 @@ class DogDatabase:
         query += " ORDER BY timestamp DESC"
         
         self.cursor.execute(query, params)
+        rows = self.cursor.fetchall()
         
         images = []
-        for row in self.cursor.fetchall():
+        for row in rows:
             # Decode image
             image_bytes = base64.b64decode(row['image_data'])
             nparr = np.frombuffer(image_bytes, np.uint8)
@@ -316,6 +317,7 @@ class DogDatabase:
             })
         
         return images
+
     
     def validate_image(self, image_id: int, is_valid: bool = True):
         """Mark image as validated or discarded"""