feat: add profile avatar update functionality

backend/app/routes/albums.py

@@ -8,11 +8,10 @@
     GetAlbumImagesRequest,
     GetAlbumImagesResponse,
     UpdateAlbumRequest,
-    SuccessResponse,
-    ErrorResponse,
     ImageIdsRequest,
     Album,
 )
+from app.schemas.common import ErrorResponse, SuccessResponse
 from app.database.albums import (
     db_get_all_albums,
     db_get_album_by_name,

backend/app/routes/avatar.py

@@ -0,0 +1,117 @@
+from fastapi import APIRouter, HTTPException, UploadFile, File, status
+from fastapi.responses import FileResponse
+import os
+import uuid
+import io
+from PIL import Image
+from app.database.metadata import db_get_metadata, db_update_metadata
+from app.schemas.common import ErrorResponse
+
+router = APIRouter()
+
+UPLOAD_DIR = "uploads/avatars"
+ALLOWED_EXTENSIONS = {".png", ".jpg", ".jpeg"}
+MAX_FILE_SIZE = 5 * 1024 * 1024  # 5MB
+
+os.makedirs(UPLOAD_DIR, exist_ok=True)
+
+@router.post("/upload")
+async def upload_avatar(file: UploadFile = File(...)):
+    """Upload and set user avatar"""
+    try:
+        # Validate file extension
+        file_ext = os.path.splitext(file.filename)[1].lower()
+        if file_ext not in ALLOWED_EXTENSIONS:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail=ErrorResponse(
+                    success=False,
+                    error="Invalid file type",
+                    message="Only PNG, JPG, and JPEG files are allowed"
+                ).model_dump()
+            )
+
+        # Read and validate file size
+        content = await file.read()
+        if len(content) > MAX_FILE_SIZE:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail=ErrorResponse(
+                    success=False,
+                    error="File too large",
+                    message="File size must be less than 5MB"
+                ).model_dump()
+            )
+
+        # Generate unique filename
+        filename = f"{uuid.uuid4()}{file_ext}"
+        filepath = os.path.join(UPLOAD_DIR, filename)
+
+        # Process and save image
+        try:
+            image = Image.open(io.BytesIO(content))
+            # Resize to 200x200 for consistency
+            image = image.resize((200, 200), Image.Resampling.LANCZOS)
+            image.save(filepath, optimize=True, quality=85)
+        except Exception as e:
+            raise HTTPException(
+                status_code=status.HTTP_400_BAD_REQUEST,
+                detail=ErrorResponse(
+                    success=False,
+                    error="Invalid image",
+                    message="Unable to process image file"
+                ).model_dump()
+            )
+
+        # Update user preferences with new avatar path
+        metadata = db_get_metadata() or {}
+        user_prefs = metadata.get("user_preferences", {})
+
+        # Remove old avatar file if exists
+        old_avatar = user_prefs.get("avatar")
+        if old_avatar and old_avatar.startswith("/avatars/uploads/"):
+            old_path = old_avatar.replace("/avatars/uploads/", f"{UPLOAD_DIR}/")
+            if os.path.exists(old_path):
+                os.remove(old_path)
+
+        user_prefs["avatar"] = f"/avatars/uploads/{filename}"
+        metadata["user_preferences"] = user_prefs
+
+        if not db_update_metadata(metadata):
+            # Clean up uploaded file on database error
+            if os.path.exists(filepath):
+                os.remove(filepath)
+            raise HTTPException(
+                status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+                detail=ErrorResponse(
+                    success=False,
+                    error="Database error",
+                    message="Failed to save avatar preference"
+                ).model_dump()
+            )
+
+        return {
+            "success": True,
+            "message": "Avatar uploaded successfully",
+            "avatar_url": f"/avatars/uploads/{filename}"
+        }
+
+    except HTTPException:
+        raise
+    except Exception as e:
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail=ErrorResponse(
+                success=False,
+                error="Internal server error",
+                message=f"Failed to upload avatar: {str(e)}"
+            ).model_dump()
+        )
+
+@router.get("/uploads/{filename}")
+async def get_avatar(filename: str):
+    """Serve uploaded avatar files"""
+    filepath = os.path.join(UPLOAD_DIR, filename)
+    if not os.path.exists(filepath):
+        raise HTTPException(status_code=404, detail="Avatar not found")
+    return FileResponse(filepath)

backend/app/routes/face_clusters.py

@@ -11,7 +11,7 @@
     db_get_all_clusters_with_face_counts,
     db_get_images_by_cluster_id,  # Add this import
 )
-from app.schemas.face_clusters import (
+from app.schemas.faces import (
     RenameClusterRequest,
     RenameClusterResponse,
     RenameClusterData,

backend/app/routes/folders.py

@@ -16,7 +16,6 @@
     AddFolderRequest,
     AddFolderResponse,
     AddFolderData,
-    ErrorResponse,
     UpdateAITaggingRequest,
     UpdateAITaggingResponse,
     UpdateAITaggingData,
@@ -30,6 +29,7 @@
     GetAllFoldersData,
     FolderDetails,
 )
+from app.schemas.common import ErrorResponse
 import os
 from app.utils.folders import (
     folder_util_add_folder_tree,

backend/app/routes/images.py

@@ -1,7 +1,7 @@
 from fastapi import APIRouter, HTTPException, Query, status
 from typing import List, Optional
 from app.database.images import db_get_all_images
-from app.schemas.images import ErrorResponse
+from app.schemas.common 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

backend/app/routes/user_preferences.py

@@ -5,8 +5,8 @@
     UpdateUserPreferencesRequest,
     UpdateUserPreferencesResponse,
     UserPreferencesData,
-    ErrorResponse,
 )
+from app.schemas.common import ErrorResponse
 
 
 router = APIRouter()
@@ -31,6 +31,7 @@ def get_user_preferences():
         user_preferences = UserPreferencesData(
             YOLO_model_size=user_prefs_data.get("YOLO_model_size", "small"),
             GPU_Acceleration=user_prefs_data.get("GPU_Acceleration", True),
+            avatar=user_prefs_data.get("avatar"),
         )
 
         return GetUserPreferencesResponse(
@@ -59,7 +60,9 @@ def update_user_preferences(request: UpdateUserPreferencesRequest):
     """Update user preferences in metadata."""
     try:
         # Step 1: Validate that at least one field is provided
-        if request.YOLO_model_size is None and request.GPU_Acceleration is None:
+        if (request.YOLO_model_size is None and 
+            request.GPU_Acceleration is None and 
+            request.avatar is None):
             raise ValueError("At least one preference field must be provided")
 
         # Step 2: Get current metadata
@@ -74,6 +77,9 @@ def update_user_preferences(request: UpdateUserPreferencesRequest):
 
         if request.GPU_Acceleration is not None:
             current_user_prefs["GPU_Acceleration"] = request.GPU_Acceleration
+
+        if request.avatar is not None:
+            current_user_prefs["avatar"] = request.avatar
 
         # Step 5: Update metadata with new user preferences
         metadata["user_preferences"] = current_user_prefs
@@ -95,6 +101,7 @@ def update_user_preferences(request: UpdateUserPreferencesRequest):
         user_preferences = UserPreferencesData(
             YOLO_model_size=current_user_prefs.get("YOLO_model_size", "small"),
             GPU_Acceleration=current_user_prefs.get("GPU_Acceleration", True),
+            avatar=current_user_prefs.get("avatar"),
         )
 
         return UpdateUserPreferencesResponse(

backend/app/schemas/album.py

@@ -1,6 +1,7 @@
 from pydantic import BaseModel, Field, field_validator
 from typing import Optional, List
 from pydantic_core.core_schema import ValidationInfo
+from .common import ErrorResponse, SuccessResponse
 
 
 class Album(BaseModel):
@@ -75,12 +76,4 @@ class GetAlbumImagesResponse(BaseModel):
     image_ids: List[str]
 
 
-class SuccessResponse(BaseModel):
-    success: bool
-    msg: str
-
 
-class ErrorResponse(BaseModel):
-    success: bool = False
-    message: str
-    error: str

backend/app/schemas/common.py

@@ -0,0 +1,15 @@
+from pydantic import BaseModel
+from typing import Optional
+
+
+class ErrorResponse(BaseModel):
+    """Common error response model used across all endpoints"""
+    success: bool = False
+    message: str
+    error: str
+
+
+class SuccessResponse(BaseModel):
+    """Common success response model"""
+    success: bool
+    message: str

backend/app/schemas/face_clusters.py → backend/app/schemas/faces.py

@@ -1,13 +1,38 @@
 from pydantic import BaseModel
 from typing import List, Optional, Dict, Union, Any
+from .common import ErrorResponse
 
 
-# Request Models
+# Face Tagging Models
+class SimilarPair(BaseModel):
+    image1: str
+    image2: str
+    similarity: float
+
+
+class FaceMatchingResponse(BaseModel):
+    success: bool
+    message: str
+    similar_pairs: List[SimilarPair]
+
+
+class FaceClustersResponse(BaseModel):
+    success: bool
+    message: str
+    clusters: Dict[int, List[str]]
+
+
+class GetRelatedImagesResponse(BaseModel):
+    success: bool
+    message: str
+    data: Dict[str, List[str]]
+
+
+# Face Clusters Models
 class RenameClusterRequest(BaseModel):
     cluster_name: str
 
 
-# Response Models
 class RenameClusterData(BaseModel):
     cluster_id: str
     cluster_name: str
@@ -20,12 +45,6 @@ class RenameClusterResponse(BaseModel):
     data: Optional[RenameClusterData] = None
 
 
-class ErrorResponse(BaseModel):
-    success: bool = False
-    message: Optional[str] = None
-    error: Optional[str] = None
-
-
 class ClusterMetadata(BaseModel):
     cluster_id: str
     cluster_name: Optional[str]
@@ -46,7 +65,6 @@ class GetClustersResponse(BaseModel):
 
 class ImageInCluster(BaseModel):
     """Represents an image that contains faces from a specific cluster."""
-
     id: str
     path: str
     thumbnailPath: Optional[str] = None
@@ -58,7 +76,6 @@ class ImageInCluster(BaseModel):
 
 class GetClusterImagesData(BaseModel):
     """Data model for cluster images response."""
-
     cluster_id: str
     cluster_name: Optional[str] = None
     images: List[ImageInCluster]
@@ -67,7 +84,6 @@ class GetClusterImagesData(BaseModel):
 
 class GetClusterImagesResponse(BaseModel):
     """Response model for getting images in a cluster."""
-
     success: bool
     message: Optional[str] = None
     error: Optional[str] = None
@@ -82,4 +98,4 @@ class GlobalReclusterResponse(BaseModel):
     success: bool
     message: Optional[str] = None
     error: Optional[str] = None
-    data: Optional[GlobalReclusterData] = None
+    data: Optional[GlobalReclusterData] = None

backend/app/schemas/folders.py

@@ -1,5 +1,6 @@
 from pydantic import BaseModel
 from typing import Optional, List
+from .common import ErrorResponse
 
 
 # Request Models
@@ -97,7 +98,4 @@ class SyncFolderResponse(BaseModel):
     data: Optional[SyncFolderData] = None
 
 
-class ErrorResponse(BaseModel):
-    success: bool = False
-    message: Optional[str] = None
-    error: Optional[str] = None
+