from speechbrain.pretrained import EncoderClassifier
from typing import Dict, List, Any


class EndpointHandler:
    def __init__(self, path=""):
        self.model = EncoderClassifier.from_hparams("speechbrain/lang-id-voxlingua107-ecapa")
        print('model loaded')

    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
        """
         data args:
              inputs (:obj: `str`)
              date (:obj: `str`)
        Return:
              A :obj:`list` | `dict`: will be serialized and returned
        """
        # get inputs
        audio_url = data.pop("audio_url", data)

        print('audio_url', audio_url)

        # check if date exists and if it is a holiday
        # run normal prediction
        output = self.model.classify_file(audio_url)
        return {"prediction": output[1].exp()}