def reset_new_world(self):
self.frequency_count = np.zeros(len(DIALOGUE_LIST))
self.is_usable = [False] * len(DIALOGUE_LIST)
self.move_index = -1
self.dialogue_history = []
self.usable_templates = []
self.curr_event = None
self.dialogue_template = DBODialogueTemplate('templates')
self.chosen_move_index = -1
self.chosen_dialogue_move = None
self.chosen_dialogue_template = []
self.seed_time = time.time()
self.num_action_events = 0
# TODO seed(Handle triggered
np.random.seed(DEFAULT_SEED)
self.response = ""
def reset_new_world(self):
self.chosen_dialogue_move = None
self.chosen_dialogue_template = []
self.chosen_move_index = -1
self.curr_event = None
self.dialogue_history = []
self.dialogue_template = DBODialogueTemplate('templates')
self.frequency_count = np.zeros(len(DIALOGUE_LIST))
self.is_usable = [False] * len(DIALOGUE_LIST)
self.move_index = -1
self.num_action_events = 0
#occ manager
self.ongoing_c_pumping = False
self.response = ""
self.seed_time = time.time()
self.usable_templates = []
np.random.seed(DEFAULT_SEED)
self.occ_manager.reset_occ()
class DialoguePlanner:
def __init__(self):
super().__init__()
def reset_new_world(self):
print("--==--Dialogue Planner reset new world--==--")
self.frequency_count = np.zeros(len(DIALOGUE_LIST))
self.is_usable = [False] * len(DIALOGUE_LIST)
self.move_index = -1
self.dialogue_history = []
self.usable_templates = []
self.curr_event = None
self.dialogue_template = DBODialogueTemplate('templates')
self.chosen_move_index = -1
self.chosen_dialogue_move = None
self.chosen_dialogue_template = []
self.seed_time = time.time()
self.num_action_events = 0
# TODO seed(Handle triggered
np.random.seed(DEFAULT_SEED)
self.response = ""
def set_state(self, curr_event, num_action_events):
print("--==--Dialogue Planner set state--==--")
self.curr_event = curr_event
self.num_action_events = num_action_events
def reset_state(self):
print("--==--Dialogue Planner reset state--==--")
self.chosen_dialogue_move = None
self.chosen_dialogue_template = []
self.chosen_move_index = -1
self.move_index = -1
# self.curr_event = None
self.num_action_events = 0
self.is_usable = []
self.is_usable = [False] * len(DIALOGUE_LIST)
self.usable_templates = []
def perform_dialogue_planner(self, dialogue_move=""):
print("--==--Dialogue Planner perform dialogue planner--==--")
Logger.log_dialogue_model("Entering perform_dialogue_planner")
if dialogue_move == "":
self.setup_templates_is_usable()
Logger.log_dialogue_model_basic("Breakdown of values used:")
for i in range(len(DIALOGUE_LIST)):
Logger.log_dialogue_model_basic_example(DIALOGUE_LIST[i])
Logger.log_dialogue_model_basic_example(self.is_usable)
Logger.log_dialogue_model_basic_example(self.frequency_count)
self.chosen_move_index = self.choose_dialogue()
Logger.log_dialogue_model_basic("Chosen dialogue index: " +
str(self.chosen_move_index))
self.chosen_dialogue_move = DIALOGUE_LIST[
self.chosen_move_index].get_type()
self.chosen_dialogue_template = self.usable_templates[
self.chosen_move_index]
self.dialogue_history.append(
DialogueHistoryTemplate(
dialogue_type=self.chosen_dialogue_move))
self.print_dialogue_list()
else:
self.chosen_dialogue_move = dialogue_move
self.chosen_dialogue_template = self.get_usable_templates(
dialogue_move)
print("DP Line 87")
print(self.chosen_dialogue_template)
self.dialogue_history.append(
DialogueHistoryTemplate(
dialogue_type=self.chosen_dialogue_move))
Logger.log_dialogue_model_basic("PRINT ALL DIALOGUE TURNS: ")
for x in range(len(self.dialogue_history)):
Logger.log_dialogue_model_basic_example(
"ORSEN turn " + str(x) + ": " +
str(self.dialogue_history[x].dialogue_type))
Logger.log_dialogue_model_basic("FINAL CHOSEN DIALOGUE MOVE: " +
str(self.chosen_dialogue_move))
return self.chosen_dialogue_move
def setup_templates_is_usable(self):
print("--==--Dialogue Planner setup templates is usable--==--")
self.init_set_dialogue_moves_usable()
# fetch all usable dialogue templates
# For Logging
Logger.log_dialogue_model_basic("Initial Valid Dialogue Moves:")
for i in range(len(DIALOGUE_LIST)):
Logger.log_dialogue_model_basic_example(DIALOGUE_LIST[i])
Logger.log_dialogue_model_basic_example(self.is_usable)
for i in range(len(DIALOGUE_LIST)):
#check if dialogue move is initially valid
to_check = DIALOGUE_LIST[i]
Logger.log_dialogue_model_basic(to_check)
if self.is_usable[i]:
# check if dialogue has templates
self.usable_templates.append(
self.get_usable_templates(DIALOGUE_LIST[i].get_type()))
else:
self.usable_templates.append([])
# gets number of occurences
self.frequency_count[i] = self.get_num_usage(
DIALOGUE_LIST[i].get_type())
# recheck dialogue moves given templates
for i in range(len(DIALOGUE_LIST)):
self.is_usable[i] = self.is_dialogue_usable(
DIALOGUE_LIST[i].get_type(), self.usable_templates[i])
def init_set_dialogue_moves_usable(self):
print("--==--Dialogue Planner init set dialogue moves usable--==--")
# check which dialogue moves are usable
set_to_true = []
# set_to_true.append(DIALOGUE_TYPE_HINTING)
# set_to_true.append(DIALOGUE_TYPE_SUGGESTING)
if self.num_action_events <= 3:
set_to_true.append(DIALOGUE_TYPE_FEEDBACK)
set_to_true.append(DIALOGUE_TYPE_PUMPING_GENERAL)
elif self.get_num_usage(DIALOGUE_TYPE_FEEDBACK) + self.get_num_usage(
DIALOGUE_TYPE_PUMPING_GENERAL) == 3:
set_to_true.append(DIALOGUE_TYPE_PUMPING_SPECIFIC)
set_to_true.append(DIALOGUE_TYPE_PUMPING_GENERAL)
else:
set_to_true = ['feedback', 'general', 'specific', 'hinting']
self.set_dialogue_list_true(set_to_true)
def set_dialogue_list_true(self, set_to_true):
print("--==--Dialogue Planner set dialogue list true--==--")
for i in range(len(set_to_true)):
for j in range(len(DIALOGUE_LIST)):
if DIALOGUE_LIST[j].get_type() == set_to_true[i]:
self.is_usable[j] = True
self.print_dialogue_list()
def is_dialogue_usable(self, dialogue_type, curr_usable_templates):
if len(curr_usable_templates) == 0:
return False
# can be repeated 3 times only
if len(self.dialogue_history) >= 3:
len_dialogue = len(self.dialogue_history)
if self.dialogue_history[len_dialogue - 3] == dialogue_type and \
self.dialogue_history[len_dialogue - 2] == dialogue_type and \
self.dialogue_history[len_dialogue - 1] == dialogue_type:
return False
return True
def get_usable_templates(self, move_to_execute):
print("--==--Dialogue Planner get usable templates--==--")
usable_template_list = []
Logger.log_dialogue_model_basic("Fetching all templates of: " +
str(move_to_execute))
template_list = self.dialogue_template.get_templates_of_type(
move_to_execute)
for x in template_list:
Logger.log_dialogue_model_basic_example(x)
# check which template is usable
for X in template_list:
print("Checking:", X)
Logger.log_dialogue_model("Checking template " + str(X))
print(self.curr_event)
if X.is_usable(self.curr_event, self.get_num_usage(X.get_type())):
usable_template_list.append(X)
return usable_template_list
def get_num_usage(self, dialogue_type):
print("--==--Dialogue Planner get num usage--==--")
# returns number of times it has been used
count = 0
for X in self.dialogue_history:
if X.dialogue_type == dialogue_type:
count = count + 1
return count
def select_dialogue_from_weights(self, weights_to_use):
print("--==--Dialogue Planner select dialogue from weights--==--")
# weights_to_use = self.frequency_count
probability = np.repeat(1 / len(weights_to_use), len(weights_to_use))
if np.count_nonzero(weights_to_use) > 0:
max_value = np.max(weights_to_use)
max_value_list = np.repeat(max_value, len(weights_to_use))
weights_to_use - np.asarray(weights_to_use)
numerator = max_value_list - weights_to_use
probability = numerator / max_value_list
candidates = np.argwhere(probability == np.amax(probability))
candidates = candidates.flatten().tolist()
# np.random.seed(int(self.seed_time))
choice = np.random.choice(candidates)
return choice
def choose_dialogue(self):
print("--==--Dialogue Planner choose dialogue--==--")
moves_to_eval = self.get_valid_moves_index()
weights_to_eval = self.get_weights_from_index(moves_to_eval)
dialogue_move_index = self.select_dialogue_from_weights(
weights_to_eval)
return moves_to_eval[dialogue_move_index]
def get_weights_from_index(self, indexes):
print("--==--Dialogue Planner get weights from index--==--")
weights = []
for i in indexes:
weights.append(self.frequency_count[i])
return weights
def get_valid_moves_index(self):
print("--==--Dialogue Planner get valid moves index--==--")
valid_moves = []
print(self.is_usable)
for i in range(len(self.is_usable)):
if self.is_usable[i]:
valid_moves.append(i)
return valid_moves
def check_trigger_phrases(self, response, event_chain):
print("--==--Dialogue Planner check trigger--==--")
response = response.lower()
if self.response in IS_END:
return DIALOGUE_TYPE_E_END
elif response in PUMPING_TRIGGER:
if len(event_chain) > 0:
return DIALOGUE_TYPE_PUMPING_SPECIFIC
return DIALOGUE_TYPE_PROMPT
elif response in PROMPT_TRIGGER:
return DIALOGUE_TYPE_PROMPT
elif response in HINTING_TRIGGER:
if len(event_chain) > 0:
return DIALOGUE_TYPE_HINTING
return DIALOGUE_TYPE_PUMPING_GENERAL
return None
def set_template_details_history(self, chosen_template):
print("--==--Dialogue Planner set template details hist--==--")
self.dialogue_history[len(self.dialogue_history) -
1].set_template_details(chosen_template)
def print_dialogue_list(self):
print("--==--Dialogue Planner print dial list--==--")
print("\n\nCHOSEN DIALOGUE MOVE: ", self.chosen_dialogue_move)
print("move", "\t", "is_usable")
for i in range(len(DIALOGUE_LIST)):
print(DIALOGUE_LIST[i], "\t", self.is_usable[i])
def get_last_dialogue_move(self):
if len(self.dialogue_history) == 0:
return None
return self.dialogue_history[len(self.dialogue_history) - 1]
def get_suggestion_word_rel(self):
if len(self.dialogue_history) == 0:
return None
elif self.dialogue_history[
len(self.dialogue_history) -
2].dialogue_type == DIALOGUE_TYPE_SUGGESTING:
print(self.dialogue_history[len(self.dialogue_history) -
2].dialogue_type)
return self.dialogue_history[len(self.dialogue_history) -
2].word_relation
def get_second_to_last_dialogue_move(self):
if len(self.dialogue_history) >= 2:
return self.dialogue_history[len(self.dialogue_history) - 2]
return None
def is_move_eden(self, type):
for X in EDEN_DIALOGUE_LIST:
if X.dialogue_type == type:
return True
return False
def get_welcome_message_type(self):
return DIALOGUE_TYPE_ORSEN_WELCOME