NaturalLanguageGeneration update
This commit is contained in:
parent
f343ebec98
commit
d0f9db73bd
@ -33,14 +33,14 @@ class DP:
|
||||
system_act = SystemAct(SystemActType.REQUEST, ['place'])
|
||||
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
|
||||
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)
|
||||
@ -62,14 +62,14 @@ class DP:
|
||||
system_act = SystemAct(SystemActType.REQUEST, ['place'])
|
||||
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
|
||||
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)
|
||||
@ -82,7 +82,7 @@ class DP:
|
||||
return system_act
|
||||
elif last_user_act == UserActType.NEGATE:
|
||||
self.DST.clear()
|
||||
return SystemAct(SystemActType.REQMORE, [])
|
||||
return SystemAct(SystemActType.REQMORE, ['create_meeting'])
|
||||
else:
|
||||
return SystemAct(SystemActType.NOT_UNDERSTOOD, [])
|
||||
# stan edycji spotkania
|
||||
@ -174,7 +174,7 @@ class DP:
|
||||
return SystemAct(SystemActType.AFFIRM, ['update_meeting'])
|
||||
elif last_user_act == UserActType.NEGATE:
|
||||
self.DST.clear()
|
||||
return SystemAct(SystemActType.REQMORE, [])
|
||||
return SystemAct(SystemActType.REQMORE, ['meeting_to_update'])
|
||||
self.meeting_to_update = False
|
||||
if not self.meeting_to_update:
|
||||
if last_user_act == UserActType.CONFIRM:
|
||||
@ -185,7 +185,7 @@ class DP:
|
||||
return system_act
|
||||
elif last_user_act == UserActType.NEGATE:
|
||||
self.DST.clear()
|
||||
return SystemAct(SystemActType.REQMORE, [])
|
||||
return SystemAct(SystemActType.REQMORE, ['meeting_to_update'])
|
||||
else:
|
||||
return SystemAct(SystemActType.NOT_UNDERSTOOD, [])
|
||||
# stan anulowania spotkania
|
||||
@ -226,14 +226,14 @@ class DP:
|
||||
return system_act
|
||||
elif last_user_act == UserActType.NEGATE:
|
||||
self.DST.clear()
|
||||
return SystemAct(SystemActType.REQMORE, [])
|
||||
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, [])
|
||||
return SystemAct(SystemActType.REQMORE, ['meeting_list'])
|
||||
else:
|
||||
if 'date' in slots:
|
||||
system_act = SystemAct(SystemActType.MEETING_LIST, slots)
|
||||
@ -247,7 +247,7 @@ class DP:
|
||||
elif dialogue_state == UserActType.FREE_TIME:
|
||||
if last_user_act == UserActType.NEGATE:
|
||||
self.DST.clear()
|
||||
return SystemAct(SystemActType.REQMORE, [])
|
||||
return SystemAct(SystemActType.REQMORE, ['free_time'])
|
||||
else:
|
||||
if 'date' in slots:
|
||||
system_act = SystemAct(SystemActType.FREE_TIME, slots)
|
||||
|
||||
@ -1,5 +1,6 @@
|
||||
from SystemAct import SystemAct
|
||||
from SystemActType import SystemActType
|
||||
from UserActType import UserActType
|
||||
|
||||
class NLG:
|
||||
"""
|
||||
@ -8,17 +9,146 @@ class NLG:
|
||||
Wyjście: Tekst
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
pass
|
||||
def __init__(self, dst):
|
||||
self.DST = dst
|
||||
|
||||
def toText(self, systemAct: SystemAct) -> str:
|
||||
if systemAct.getActType() == SystemActType.WELCOME_MSG:
|
||||
return "Cześć"
|
||||
if systemAct.getActType() == SystemActType.INFORM:
|
||||
if "name" in systemAct.getActParams():
|
||||
return "Nazywam się Janusz"
|
||||
if systemAct.getActType() == SystemActType.BYE:
|
||||
return "Do widzenia."
|
||||
if systemAct.getActType() == SystemActType.NOT_UNDERSTOOD:
|
||||
return "Nie rozumiem o czym mówisz."
|
||||
def generateResponse(self, systemAct: SystemAct) -> str:
|
||||
dialogue_state, last_user_act, _ = self.DST.get_dialogue_state()
|
||||
slots = self.DST.get_dialogue_slots()
|
||||
if dialogue_state == UserActType.CREATE_MEETING:
|
||||
if systemAct.getActType() == SystemActType.REQUEST:
|
||||
if "date" in systemAct.getActParams():
|
||||
return "W jakim dniu ma się odbyć to spotkanie?"
|
||||
if "time" in systemAct.getActParams():
|
||||
return "W jakim czasie ma się odbyć to spotkanie?"
|
||||
if "place" in systemAct.getActParams():
|
||||
return "Gdzie ma się odbyć to spotkanie?"
|
||||
if "description" in systemAct.getActParams():
|
||||
return "Czy mam dodać jakiś opis do tego spotkania?"
|
||||
if "participants" in systemAct.getActParams():
|
||||
return "Kto ma wziąć udział w spotkaniu?"
|
||||
if systemAct.getActType() == SystemActType.CONFIRM_DOMAIN:
|
||||
date = slots['date']
|
||||
time = slots['time']
|
||||
place = slots['place']
|
||||
part_list = slots['participants']
|
||||
part = ""
|
||||
for p in part_list:
|
||||
part += p
|
||||
part += ", "
|
||||
part = part[:-2]
|
||||
desc = slots['description']
|
||||
return f'Czy mam dodać te spotkanie do kalendarza?\n' \
|
||||
f'Dzień: {date}\nCzas: {time}\nMiejsce: {place}\nUczestnicy: {part}\nOpis: {desc}'
|
||||
|
||||
# TODO: nie sprawdzone - trudno wejść do tego stanu
|
||||
elif dialogue_state == UserActType.UPDATE_MEETING:
|
||||
if systemAct.getActType() == SystemActType.REQUEST:
|
||||
if "date" in systemAct.getActParams():
|
||||
return "W jakim dniu miało się odbyć to spotkanie?"
|
||||
if "time" in systemAct.getActParams():
|
||||
return "W jakim czasie miało się odbyć to spotkanie?"
|
||||
# TODO dopracować po dodaniu DB
|
||||
if systemAct.getActType() == SystemActType.CONFIRM_DOMAIN:
|
||||
date = slots['date']
|
||||
time = slots['time']
|
||||
place = slots['place']
|
||||
part_list = slots['participants']
|
||||
part = ""
|
||||
for p in part_list:
|
||||
part += p
|
||||
part += ", "
|
||||
part = part[:-2]
|
||||
desc = slots['description']
|
||||
return f'Spotkanie:\n' \
|
||||
f'Dzień: {date}\nCzas: {time}\nMiejsce: {place}\nUczestnicy: {part}\nOpis: {desc}'
|
||||
|
||||
elif dialogue_state == UserActType.CANCEL_MEETING:
|
||||
if systemAct.getActType() == SystemActType.REQUEST:
|
||||
if "date" in systemAct.getActParams():
|
||||
return "W jakim dniu miało się odbyć to spotkanie?"
|
||||
if "time" in systemAct.getActParams():
|
||||
return "W jakim czasie miało się odbyć to spotkanie?"
|
||||
# TODO dopracować po dodaniu DB
|
||||
if systemAct.getActType() == SystemActType.CONFIRM_DOMAIN:
|
||||
date = slots['date']
|
||||
time = slots['time']
|
||||
# place = slots['place']
|
||||
# part_list = slots['participants']
|
||||
# part = ""
|
||||
# for p in part_list:
|
||||
# part += p
|
||||
# part += ", "
|
||||
# part = part[:-2]
|
||||
# desc = slots['description']
|
||||
return f'Spotkanie:\n' \
|
||||
f'Dzień: {date}\nCzas: {time}'
|
||||
|
||||
elif dialogue_state == UserActType.MEETING_LIST:
|
||||
if systemAct.getActType() == SystemActType.REQUEST:
|
||||
if "date" in systemAct.getActParams():
|
||||
return "Z jakiego okresu chcesz przejrzeć spotkania?"
|
||||
# TODO: dopracować po dodaniu DB
|
||||
if systemAct.getActType() == SystemActType.MEETING_LIST:
|
||||
response = ""
|
||||
for s in slots:
|
||||
date = s['date']
|
||||
time = s['time']
|
||||
place = s['place']
|
||||
part_list = s['participants']
|
||||
part = ""
|
||||
for p in part_list:
|
||||
part += p
|
||||
part += ", "
|
||||
part = part[:-2]
|
||||
desc = s['description']
|
||||
response += f'Spotkanie:\nDzień: {date}\nCzas: {time}\nMiejsce: {place}\nUczestnicy: {part}\nOpis: {desc}\n'
|
||||
response += "--------------------"
|
||||
return response
|
||||
|
||||
elif dialogue_state == UserActType.FREE_TIME:
|
||||
if systemAct.getActType() == SystemActType.REQUEST:
|
||||
if "date" in systemAct.getActParams():
|
||||
return "W jakim okresie chcesz znaleźć wolny czas?"
|
||||
# TODO: dopracować po dodaniu DB
|
||||
if systemAct.getActType() == SystemActType.FREE_TIME:
|
||||
response = ""
|
||||
for s in slots:
|
||||
date = s['date']
|
||||
time = s['time']
|
||||
response += f'Spotkanie:\nDzień: {date}\nCzas: {time}\n'
|
||||
return response
|
||||
|
||||
elif systemAct.getActType() == SystemActType.AFFIRM:
|
||||
if "create_meeting" in systemAct.getActParams():
|
||||
return "Spotkanie zostało dodane"
|
||||
if "update_meeting" in systemAct.getActParams():
|
||||
return "Spotkanie zostało zaktualizowane"
|
||||
if "cancel_meeting" in systemAct.getActParams():
|
||||
return "Spotkanie zostało odwołane"
|
||||
|
||||
elif systemAct.getActType() == SystemActType.REQMORE:
|
||||
if "create_meeting" in systemAct.getActParams():
|
||||
return "Spotkanie zostało odrzucone. Mogę pomóc w czymś jeszcze?"
|
||||
if "update_meeting" in systemAct.getActParams():
|
||||
return "Aktualizacja spotkania została anulowana. Mogę pomóc w czymś jeszcze?"
|
||||
if "cancel_meeting" in systemAct.getActParams():
|
||||
return "Odwoływanie spotkania zostało anulowane. Mogę pomóc w czymś jeszcze?"
|
||||
if "meeting_list" in systemAct.getActParams() or "free_time" in systemAct.getActParams():
|
||||
return "Mogę pomóc w czymś jeszcze?"
|
||||
|
||||
else:
|
||||
if systemAct.getActType() == SystemActType.WELCOME_MSG:
|
||||
return "Cześć"
|
||||
if systemAct.getActType() == SystemActType.REQMORE:
|
||||
return "Czy mogę Ci w czymś jeszcze pomóc?"
|
||||
if systemAct.getActType() == SystemActType.BYE:
|
||||
return "Do widzenia."
|
||||
if systemAct.getActType() == SystemActType.NOT_UNDERSTOOD:
|
||||
return "Nie rozumiem o czym mówisz."
|
||||
|
||||
# TODO: Not implemented in DP
|
||||
# if systemAct.getActType() == SystemActType.INFORM:
|
||||
# if "name" in systemAct.getActParams():
|
||||
# return "Nazywam się Janusz"
|
||||
raise Exception("SystemAct:{} not recognized".format(systemAct))
|
||||
|
||||
7
main.py
7
main.py
@ -9,7 +9,7 @@ if __name__ == "__main__":
|
||||
nlu = NLU()
|
||||
dst = DST()
|
||||
dp = DP(dst)
|
||||
nlg = NLG()
|
||||
nlg = NLG(dst)
|
||||
|
||||
while(1):
|
||||
user_input = input("Wpisz tekst: ")
|
||||
@ -21,15 +21,18 @@ if __name__ == "__main__":
|
||||
state, last_user_act, last_system_act = dst.get_dialogue_state()
|
||||
slots = dst.get_dialogue_slots()
|
||||
system_act = dp.chooseTactic()
|
||||
|
||||
print('------ stan ------')
|
||||
print(state, last_user_act, last_system_act)
|
||||
print('------ przechowywane sloty ------')
|
||||
print(slots)
|
||||
print('------ wybrana akcja systemu ------')
|
||||
print(system_act)
|
||||
system_response = nlg.generateResponse(system_act)
|
||||
print('------ wygenerowana odpowiedź systemu ------')
|
||||
print(system_response)
|
||||
print('-----------------------------------')
|
||||
print('-----------------------------------')
|
||||
#text = nlg.toText(system_act)
|
||||
|
||||
if system_act.getActType() == SystemActType.BYE:
|
||||
break
|
||||
|
||||
Loading…
Reference in New Issue
Block a user