class Message: def __init__(self, msg=None): self._messages = msg if msg else [] self._images = [] self.skip_next = False def add_message(self, question, answer=None): quension_msg_dict = {'from': 'human'} quension_msg_dict['value'] = question answer_msg_dict = {'from': 'gpt'} answer_msg_dict['value'] = answer self._messages.append(quension_msg_dict) self._messages.append(answer_msg_dict) def add_image(self, image, index=0): self._images.append((image, index)) @property def images(self): return self._images @property def messages(self): return self._messages def copy(self): return Message(self._messages)