state.py

class State:
    def __init__(self, story_graph, story_told, utterance, intent, bert_model, previous_intent):
        self._story_graph = story_graph
        self._story_told = [story_told]
        self._intent = intent
        self._utterance = utterance
        self._previous_intent = previous_intent
        self._bert_model = bert_model
        self.is_story_ended = False
        self.nodes_to_visit = []
        self.current_node = "sentence1"

    @property
    def story_graph(self):
        return self._story_graph

    @property
    def story_told(self):
        return self._story_told

    @property
    def intent(self):
        return self._intent

    @property
    def utterance(self):
        return self._utterance

    @property
    def previous_intent(self):
        return self._previous_intent

    @property
    def bert_model(self):
        return self._bert_model

    @utterance.setter
    def utterance(self, new_utterance):
        self._utterance = new_utterance

    @intent.setter
    def intent(self, new_intent):
        self._intent = new_intent

    @previous_intent.setter
    def previous_intent(self, new_previous_intent):
        self._previous_intent = new_previous_intent

    def add_increment_told(self, new_story_increment):
        self._story_told.append(new_story_increment)

    def set_new_state(self, fsm):
        """
        Given the current state and the intent, the aim is to throw an action to change
        the current state. It returns the new state in which the bot is in.
        :param fsm: object from class ConversationFMS
        :return: void - set new current fsm state
        """
        state = fsm.state
        if state == "introduction":
            if self.previous_intent == "":
                fsm.no_change()  # new state: introduction
            else:
                if self.intent == "ynq" or self.intent == "whq":
                    fsm.question()  # new state: answering
                elif self.intent == "affirm" or self.intent == "request_increment" or self.intent == "exclaim_pos"\
                        or self.intent == "ask_for_story" or self.intent == "thanks" or self.intent == "comment":
                    fsm.acceptance()  # new state: storytelling
                else:
                    fsm.no_change()  # new state: introduction

        elif state == "storytelling":
            if self.current_node == "sentence10":
                fsm.story_ends()  # new state: closing
            elif self.intent == "ynq" or self.intent == "whq":
                fsm.question()  # new state: bert
            else:
                fsm.non_question()  # new state: storytelling

        elif state == "closing":
            if self.intent == "ynq" or self.intent == "whq":
                fsm.question()  # new state: answering_f
            else:
                fsm.acceptance()  # new state: link_to_survey

        elif state == "answering":
            if self.intent == "ynq" or self.intent == "whq":
                fsm.question()  # new state: answering
            else:
                fsm.acceptance()  # new state: storytelling

        elif state == "bert":
            if self.intent == "ynq" or self.intent == "whq":
                fsm.question()  # new state: bert
            elif self.current_node == "sentence10":
                fsm.story_ends()  # new state: closing
            else:
                fsm.acceptance()  # new state: storytelling

        elif state == "answering_f":
            if self.intent == "ynq" or self.intent == "whq":
                fsm.question()  # new state: answering_f
            else:
                fsm.acceptance()  # new state: link_to_survey