lint

Author: DecodeX15

backend/app/database/face_clusters.py

@@ -195,9 +195,9 @@ def db_delete_all_clusters() -> int:
     return deleted_count
 
 
-def db_get_all_clusters_with_face_counts() -> List[
-    Dict[str, Union[str, Optional[str], int]]
-]:
+def db_get_all_clusters_with_face_counts() -> (
+    List[Dict[str, Union[str, Optional[str], int]]]
+):
     """
     Retrieve all clusters with their face counts and stored face images.
 

backend/app/database/faces.py

@@ -234,9 +234,9 @@ def db_get_faces_unassigned_clusters() -> List[Dict[str, Union[FaceId, FaceEmbed
     return faces
 
 
-def db_get_all_faces_with_cluster_names() -> List[
-    Dict[str, Union[FaceId, FaceEmbedding, Optional[str]]]
-]:
+def db_get_all_faces_with_cluster_names() -> (
+    List[Dict[str, Union[FaceId, FaceEmbedding, Optional[str]]]]
+):
     """
     Get all faces with their corresponding cluster names.
 
@@ -271,7 +271,7 @@ def db_get_all_faces_with_cluster_names() -> List[
 
 
 def db_update_face_cluster_ids_batch(
-    face_cluster_mapping: List[Dict[str, Union[FaceId, ClusterId]]]
+    face_cluster_mapping: List[Dict[str, Union[FaceId, ClusterId]]],
 ) -> None:
     """
     Update cluster IDs for multiple faces in batch.

backend/app/database/folders.py

@@ -382,9 +382,9 @@ def db_get_folder_ids_by_paths(
         conn.close()
 
 
-def db_get_all_folder_details() -> List[
-    Tuple[str, str, Optional[str], int, bool, Optional[bool]]
-]:
+def db_get_all_folder_details() -> (
+    List[Tuple[str, str, Optional[str], int, bool, Optional[bool]]]
+):
     """
     Get all folder details including folder_id, folder_path, parent_folder_id,
     last_modified_time, AI_Tagging, and taggingCompleted.

backend/app/database/images.py

@@ -391,9 +391,11 @@ def db_toggle_image_favourite_status(image_id: str) -> bool:
             return False
         cursor.execute("PRAGMA table_info(images)")
         columns = [column[1] for column in cursor.fetchall()]
-        if 'isFavourite' not in columns:
+        if "isFavourite" not in columns:
             print("isFavourite column doesn't exist, adding it now...")
-            cursor.execute("ALTER TABLE images ADD COLUMN isFavourite BOOLEAN DEFAULT 0")
+            cursor.execute(
+                "ALTER TABLE images ADD COLUMN isFavourite BOOLEAN DEFAULT 0"
+            )
             conn.commit()
         cursor.execute(
             """
@@ -412,4 +414,4 @@ def db_toggle_image_favourite_status(image_id: str) -> bool:
         conn.rollback()
         return False
     finally:
-        conn.close()
+        conn.close()

backend/app/routes/albums.py

@@ -29,6 +29,7 @@
 
 router = APIRouter()
 
+
 # GET /albums/ - Get all albums
 @router.get("/", response_model=GetAlbumsResponse)
 def get_albums(show_hidden: bool = Query(False)):

backend/app/routes/images.py

@@ -5,6 +5,7 @@
 from app.utils.images import image_util_parse_metadata
 from pydantic import BaseModel
 from app.database.images import db_toggle_image_favourite_status
+
 router = APIRouter()
 
 
@@ -52,15 +53,15 @@ def get_all_images():
 
         # Convert to response format
         image_data = [
-                ImageData(
+            ImageData(
                 id=image["id"],
                 path=image["path"],
                 folder_id=image["folder_id"],
                 thumbnailPath=image["thumbnailPath"],
                 metadata=image_util_parse_metadata(image["metadata"]),
-                    isTagged=image["isTagged"],
-                    isFavourite=image.get("isFavourite", False),
-                    tags=image["tags"],
+                isTagged=image["isTagged"],
+                isFavourite=image.get("isFavourite", False),
+                tags=image["tags"],
             )
             for image in images
         ]
@@ -84,23 +85,35 @@ def get_all_images():
 
 # adding add to favourite and remove from favourite routes
 
+
 class ToggleFavouriteRequest(BaseModel):
     image_id: str
+
+
 @router.post("/toggle-favourite")
 def toggle_favourite(req: ToggleFavouriteRequest):
     image_id = req.image_id
     try:
         success = db_toggle_image_favourite_status(image_id)
         if not success:
-            raise HTTPException(status_code=404, detail="Image not found or failed to toggle")
+            raise HTTPException(
+                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)
-        return {"success": True, "image_id": image_id, "isFavourite": image.get("isFavourite", False)}
+        image = next(
+            (img for img in db_get_all_images() if img["id"] == image_id), None
+        )
+        return {
+            "success": True,
+            "image_id": image_id,
+            "isFavourite": image.get("isFavourite", False),
+        }
 
     except Exception as e:
         print(f"Toggle favourite error: {e}")
         raise HTTPException(status_code=500, detail=f"Internal server error: {e}")
 
+
 class ImageInfoResponse(BaseModel):
     id: str
     path: str
@@ -111,6 +124,7 @@ class ImageInfoResponse(BaseModel):
     isFavourite: bool
     tags: Optional[List[str]] = None
 
+
 @router.get("/info/{image_id}", response_model=ImageInfoResponse)
 def get_image_info(image_id: str):
     # 1. Fetch all images
@@ -128,5 +142,5 @@ def get_image_info(image_id: str):
         metadata=image["metadata"],
         isTagged=image["isTagged"],
         isFavourite=image.get("isFavourite", False),
-        tags=image.get("tags", [])
-    )
+        tags=image.get("tags", []),
+    )

backend/app/utils/images.py

@@ -99,7 +99,7 @@ def image_util_process_untagged_images() -> bool:
 
 
 def image_util_classify_and_face_detect_images(
-    untagged_images: List[Dict[str, str]]
+    untagged_images: List[Dict[str, str]],
 ) -> None:
     """Classify untagged images and detect faces if applicable."""
     object_classifier = ObjectClassifier()
@@ -262,7 +262,7 @@ def image_util_remove_obsolete_images(folder_id_list: List[int]) -> int:
 
 
 def image_util_create_folder_path_mapping(
-    folder_ids: List[Tuple[int, str]]
+    folder_ids: List[Tuple[int, str]],
 ) -> Dict[str, int]:
     """
     Create a dictionary mapping folder paths to their IDs.