import logging
import time

from fastapi import APIRouter

from app.schemas.similar_match import SimilarMatchMItem
from app.service.similar_match.service import SimilarMatch
from app.service.utils.decorator import RunTime

logger = logging.getLogger()
router = APIRouter()


@RunTime
@router.post("similar_match")
def similar_match(request_item: SimilarMatchMItem):
    try:
        if request_item.result_number <= 0:
            raise KeyError("result number can't be less than 0")
        service = SimilarMatch(request_item)
        search_response = service.match_features()
        response_data = []
        for response in search_response[0]:
            response_data.append(response['entity'])
        return {"message": "ok", "data": response_data}
    except KeyError as e:
        logger.warning(str(e))
        return {"message": "result number can't be less than 0", "data": []}