message.py

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)