fix: optimize toggle_favourite endpoint performance

backend/app/database/images.py

@@ -1,6 +1,8 @@
 # Standard library imports
 import sqlite3
-from typing import Any, List, Mapping, Tuple, TypedDict, Union
+from typing import Any, List, Mapping, Tuple, TypedDict, Union, Optional
+import json
+
 
 # App-specific imports
 from app.config.settings import (
@@ -419,3 +421,28 @@ def db_toggle_image_favourite_status(image_id: str) -> bool:
         return False
     finally:
         conn.close()
+
+def db_get_image_by_id(image_id: str) -> Optional[dict]:
+    """Get a single image by ID with its favorite status."""
+    conn = _connect()
+    cursor = conn.cursor()
+    try:
+        cursor.execute("""
+            SELECT id, path, folder_id, thumbnailPath, metadata, isTagged, isFavourite
+            FROM images
+            WHERE id = ?
+        """, (image_id,))
+        row = cursor.fetchone()
+        if not row:
+            return None
+        return {
+            "id": row[0],
+            "path": row[1],
+            "folder_id": row[2],
+            "thumbnailPath": row[3],
+            "metadata": json.loads(row[4]) if row[4] else {},
+            "isTagged": bool(row[5]),
+            "isFavourite": bool(row[6]),
+        }
+    finally:
+        conn.close()

backend/app/routes/images.py

@@ -4,9 +4,10 @@
 from app.schemas.images import ErrorResponse
 from app.utils.images import image_util_parse_metadata
 from pydantic import BaseModel
-from app.database.images import db_toggle_image_favourite_status
+from app.database.images import db_toggle_image_favourite_status, db_get_image_by_id
 from app.logging.setup_logging import get_logger
 
+
 # Initialize logger
 logger = get_logger(__name__)
 router = APIRouter()
@@ -105,9 +106,14 @@ def toggle_favourite(req: ToggleFavouriteRequest):
                 status_code=404, detail="Image not found or failed to toggle"
             )
         # Fetch updated status to return
-        image = next(
-            (img for img in db_get_all_images() if img["id"] == image_id), None
-        )
+        #image = next(
+        #    (img for img in db_get_all_images() if img["id"] == image_id), None
+        #)
+        image = db_get_image_by_id(image_id)
+        if not image:
+            raise HTTPException(
+                status_code=404, detail="Image not found after toggle"
+            )
         return {
             "success": True,
             "image_id": image_id,