Make better, upgraded class Purtek for holding question answer data for training Keras Transformer model:

class Purtek:
    """ Single data representation for PurtkGPT """

    def __init__(self, question: str, answer: str, contex = '', kategoria = Kategoria.BRAK):
        # DATA
        self.question = question
        self.answer = answer
        self.contex = contex

        # INFO
        self.kategoria = kategoria
		
    def save(self, file_name):
        with open(file_name, 'w') as f:
            json.dump(self.__dict__, f)

    def load(self, file_name):
        with open(file_name, 'r') as f:
            self.__dict__.update(json.load(f))
			
    def __dict__(self):
        return {
            'question': self.question,
            'answer': self.answer,
            'contex': self.contex,
            'kategoria': self.kategoria
        }