services/answers_service.py

from models.prediction import Prediction
from core.settings import Config
from repositories.answers_repository import AnswersRepository


class AnswersService:

    @staticmethod
    def get_answer_by_id(prediction: Prediction):
        if prediction.score > Config.threshold:

            return AnswersRepository.get_elem_by_id(prediction.label)

        return AnswersRepository.default_answer