262 lines
14 KiB
Python
262 lines
14 KiB
Python
from SystemAct import SystemAct
|
|
from UserActType import UserActType
|
|
from SystemActType import SystemActType
|
|
from collections import defaultdict
|
|
|
|
|
|
class DP:
|
|
def __init__(self, dst):
|
|
self.DST = dst
|
|
self.meeting_to_update = False
|
|
|
|
def choose_tactic(self) -> SystemAct:
|
|
dialogue_state, last_user_act, last_system_act = self.DST.get_dialogue_state()
|
|
slots = self.DST.get_dialogue_slots()
|
|
# stan dodawania spotkania
|
|
if dialogue_state == UserActType.CREATE_MEETING:
|
|
if not last_system_act:
|
|
if 'date' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'time' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['time'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'place' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['place'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'participants' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['participants'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'description' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['description'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
system_act = SystemAct(SystemActType.CONFIRM_DOMAIN, slots)
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif last_system_act.getActType() == SystemActType.REQUEST:
|
|
if last_user_act == UserActType.NEGATE:
|
|
slot_type = last_system_act.getActParams()[0]
|
|
if slot_type not in ['date', 'time']:
|
|
self.DST.insert_empty_slot(slot_type)
|
|
if 'date' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'time' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['time'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'place' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['place'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'participants' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['participants'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'description' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['description'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
system_act = SystemAct(SystemActType.CONFIRM_DOMAIN, slots)
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif last_system_act.getActType() == SystemActType.CONFIRM_DOMAIN:
|
|
if last_user_act == UserActType.CONFIRM:
|
|
system_act = SystemAct(SystemActType.AFFIRM, ['create_meeting'])
|
|
# implementacja wpisywanie spotkania do bazy
|
|
self.DST.clear()
|
|
return system_act
|
|
elif last_user_act == UserActType.NEGATE:
|
|
self.DST.clear()
|
|
return SystemAct(SystemActType.REQMORE, ['create_meeting'])
|
|
else:
|
|
return SystemAct(SystemActType.NOT_UNDERSTOOD, [])
|
|
# stan edycji spotkania
|
|
elif dialogue_state == UserActType.UPDATE_MEETING:
|
|
if not last_system_act:
|
|
if 'date' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date', False])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'time' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['time', False])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
# implementacja wyszukiwania odpowiedniego spotkania w bazie
|
|
return SystemAct(SystemActType.CONFIRM_DOMAIN, ['meeting_to_update'])
|
|
elif last_system_act.getActType() == SystemActType.REQUEST:
|
|
if not self.meeting_to_update:
|
|
if 'date' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date', False])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'time' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['time', False])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
# implementacja wyszukiwania odpowiedniego spotkania w bazie
|
|
system_act = SystemAct(SystemActType.CONFIRM_DOMAIN, ['meeting_to_update'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
if last_user_act == UserActType.NEGATE:
|
|
slot_type = last_system_act.getActParams()[0]
|
|
self.DST.insert_empty_slot(slot_type)
|
|
if 'date' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'time' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['time', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'place' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['place', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'description' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['description', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'participants' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['participants', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
system_act = SystemAct(SystemActType.CONFIRM_DOMAIN, slots)
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
if 'date' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'time' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['time', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'place' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['place', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'description' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['description', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'participants' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['participants', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
system_act = SystemAct(SystemActType.CONFIRM_DOMAIN, slots)
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif last_system_act.getActType() == SystemActType.CONFIRM_DOMAIN:
|
|
if self.meeting_to_update:
|
|
if last_user_act == UserActType.CONFIRM:
|
|
self.DST.clear()
|
|
return SystemAct(SystemActType.AFFIRM, ['update_meeting'])
|
|
elif last_user_act == UserActType.NEGATE:
|
|
self.DST.clear()
|
|
return SystemAct(SystemActType.REQMORE, ['meeting_to_update'])
|
|
self.meeting_to_update = False
|
|
if not self.meeting_to_update:
|
|
if last_user_act == UserActType.CONFIRM:
|
|
self.meeting_to_update = True
|
|
self.DST.clear_slots()
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date', True])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif last_user_act == UserActType.NEGATE:
|
|
self.DST.clear()
|
|
return SystemAct(SystemActType.REQMORE, ['meeting_to_update'])
|
|
else:
|
|
return SystemAct(SystemActType.NOT_UNDERSTOOD, [])
|
|
# stan anulowania spotkania
|
|
elif dialogue_state == UserActType.CANCEL_MEETING:
|
|
if not last_system_act:
|
|
if 'date' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'time' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['time'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
# implementacja wyszukiwania odpowiedniego spotkania w bazie
|
|
system_act = SystemAct(SystemActType.CONFIRM_DOMAIN, ['meeting_to_cancel'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif last_system_act.getActType() == SystemActType.REQUEST:
|
|
if 'date' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif 'time' not in slots:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['time'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
else:
|
|
# implementacja wyszukiwania odpowiedniego spotkania w bazie
|
|
system_act = SystemAct(SystemActType.CONFIRM_DOMAIN, ['meeting_to_cancel'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
elif last_system_act.getActType() == SystemActType.CONFIRM_DOMAIN:
|
|
if last_user_act == UserActType.CONFIRM:
|
|
system_act = SystemAct(SystemActType.AFFIRM, ['cancel_meeting'])
|
|
# implementacja usuwania spotkania z bazy
|
|
self.DST.clear()
|
|
return system_act
|
|
elif last_user_act == UserActType.NEGATE:
|
|
self.DST.clear()
|
|
return SystemAct(SystemActType.REQMORE, ['cancel_meeting'])
|
|
else:
|
|
return SystemAct(SystemActType.NOT_UNDERSTOOD, [])
|
|
# stan prośby o listę spotkań
|
|
elif dialogue_state == UserActType.MEETING_LIST:
|
|
if last_user_act == UserActType.NEGATE:
|
|
self.DST.clear()
|
|
return SystemAct(SystemActType.REQMORE, ['meeting_list'])
|
|
else:
|
|
if 'date' in slots:
|
|
system_act = SystemAct(SystemActType.MEETING_LIST, {'date': slots['date']})
|
|
return system_act
|
|
else:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
# stan prośby o czas wolny
|
|
elif dialogue_state == UserActType.FREE_TIME:
|
|
if last_user_act == UserActType.NEGATE:
|
|
self.DST.clear()
|
|
return SystemAct(SystemActType.REQMORE, ['free_time'])
|
|
else:
|
|
if 'date' in slots:
|
|
system_act = SystemAct(SystemActType.FREE_TIME, {'date': slots['date']})
|
|
return system_act
|
|
else:
|
|
system_act = SystemAct(SystemActType.REQUEST, ['date'])
|
|
self.DST.system_update(system_act)
|
|
return system_act
|
|
# brak określonego stanu
|
|
else:
|
|
if last_user_act == UserActType.HELLO:
|
|
return SystemAct(SystemActType.WELCOME_MSG, [])
|
|
elif last_user_act == UserActType.BYE:
|
|
return SystemAct(SystemActType.BYE, [])
|
|
elif last_user_act == UserActType.THANKYOU:
|
|
return SystemAct(SystemActType.REQMORE, [])
|
|
else:
|
|
return SystemAct(SystemActType.NOT_UNDERSTOOD, [])
|