feat: Implement face clustering and media retrieval for persons
This commit is contained in:
@@ -58,4 +58,23 @@ impl FaceEmbeddingRepository for PostgresFaceEmbeddingRepository {
|
||||
|
||||
Ok(pg_embedding.map(FaceEmbedding::from))
|
||||
}
|
||||
|
||||
async fn list_unassigned_by_user(&self, user_id: Uuid) -> CoreResult<Vec<FaceEmbedding>> {
|
||||
let pg_embeddings = sqlx::query_as!(
|
||||
PostgresFaceEmbedding,
|
||||
r#"
|
||||
SELECT fe.id, fe.face_region_id, fe.model_id, fe.embedding
|
||||
FROM face_embeddings fe
|
||||
JOIN face_regions fr ON fe.face_region_id = fr.id
|
||||
JOIN media m ON fr.media_id = m.id
|
||||
WHERE fr.person_id IS NULL AND m.owner_id = $1
|
||||
"#,
|
||||
user_id
|
||||
)
|
||||
.fetch_all(&self.pool)
|
||||
.await
|
||||
.map_err(|e| CoreError::Database(e.to_string()))?;
|
||||
|
||||
Ok(pg_embeddings.into_iter().map(FaceEmbedding::from).collect())
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user