From 166dd4886cb0b4c005eb467999832c6a0f1aa9c8 Mon Sep 17 00:00:00 2001 From: s495728 Date: Mon, 10 Jun 2024 02:39:46 +0200 Subject: [PATCH] Add dialog policy --- src/main.py | 2 +- src/service/dialog_policy.py | 35 ++++++++++++++++++++++++++++++----- 2 files changed, 31 insertions(+), 6 deletions(-) diff --git a/src/main.py b/src/main.py index 0d6d7cd..3a505a5 100644 --- a/src/main.py +++ b/src/main.py @@ -32,7 +32,7 @@ while True: monitor.update(frame) # DP - # print(dialog_policy.next_dialogue_act(monitor.read()).act) + system_action = dialog_policy.predict(monitor) # NLG act, slots = parse_frame(frame) diff --git a/src/service/dialog_policy.py b/src/service/dialog_policy.py index ee9d29c..a40bb16 100644 --- a/src/service/dialog_policy.py +++ b/src/service/dialog_policy.py @@ -1,8 +1,33 @@ +from collections import defaultdict from model.frame import Frame +from src.model.slot import Slot class DialogPolicy: - def next_dialogue_act(self, frames: list[Frame]) -> Frame: - if frames[-1].act == "welcomemsg": - return Frame("system", "welcomemsg", []) - else: - return Frame("system", "canthelp", []) + + def predict(self, dsm): + system_action = defaultdict(list) + last_frame = dsm.state['history'][-1] + if(dsm.was_previous_order_invalid==False): + match(last_frame.act.strip('/')): + case "inform" | "affirm": + current_active_status = dsm.get_current_active_stage() + match(current_active_status): + case "collect_food": + system_action["inform"].append([Slot("menu", dsm.state['constants']['menu'])]) + case "collect_drinks": + system_action["inform"].append([Slot("drink"), dsm.state['constants']['drink']]) + case "collect_address": + system_action["request"].append([Slot("address")]) + if(current_active_status == None): + system_action["inform"].append([Slot("order"), dsm.belief_state]) + case "request": + for slot in last_frame.slots: + system_action["inform"].append([Slot(slot), dsm.state['constants'][slot]]) + case "welcomemsg": + system_action["inform"].append([Slot("menu"), dsm.state['constants']['menu']]) + case "bye": + system_action["bye"].append([], None) + case "negate": + system_action["affirm"].append([], None) + + return system_action \ No newline at end of file