app/api/register.py

import cv2
import numpy as np

from fastapi import APIRouter, Depends, HTTPException, UploadFile, File
from sqlalchemy.orm import Session

from app.database.database import get_db
from app.database.models import Person
from app.services.face_detector import detect_faces
from app.services.embedding import get_embedding
from app.services.face_quality import check_face_quality

router = APIRouter(prefix="/register", tags=["Register"])


@router.post("/")
async def register_person(
    name: str,
    file: UploadFile = File(...),
    db: Session = Depends(get_db),
):
    """
    Register a new person from an uploaded image.
    """

    # قراءة الصورة
    contents = await file.read()

    image = cv2.imdecode(
        np.frombuffer(contents, np.uint8),
        cv2.IMREAD_COLOR,
    )

    if image is None:
        raise HTTPException(
            status_code=400,
            detail="الصورة غير صالحة"
        )

    # كشف الوجه
    boxes = detect_faces(image)

    if len(boxes) == 0:
        raise HTTPException(
            status_code=400,
            detail="لم يتم العثور على أي وجه"
        )

    # أول وجه فقط
    x1, y1, x2, y2 = boxes[0]

    face_crop = image[y1:y2, x1:x2]

    if face_crop.size == 0:
        raise HTTPException(
            status_code=400,
            detail="فشل استخراج الوجه"
        )

    # فحص الجودة
    quality = check_face_quality(face_crop)

    if not quality["ok"]:
        raise HTTPException(
            status_code=400,
            detail=quality["message"]
        )

    # استخراج الـ Embedding
    embedding = get_embedding(face_crop)

    # حفظه في قاعدة البيانات
    person = Person(
        name=name,
        embedding=embedding.tolist(),
    )

    db.add(person)
    db.commit()
    db.refresh(person)

    return {
        "id": person.id,
        "name": person.name,
        "quality": quality,
        "message": "تم التسجيل بنجاح"
    }