kdudzic/aitech-sd-lab
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Zaktualizuj 'dialogue_system.py'

Browse Source 
fix
This commit is contained in:
Julian Zabłoński 2022-06-22 10:37:10 +02:00
parent 06fc1dd32c
commit 520f262ecc
1 changed files with 19 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
dialogue_system.py
View File

@ -59,16 +59,29 @@ def main():
formatted_prediction = format_prediction(slots, intent) formatted_prediction = format_prediction(slots, intent)
for slot in formatted_prediction: for slot in formatted_prediction:
if len(formatted_prediction)<2: if len(formatted_prediction)<2:
if dst.state['system_action'][-1] != slot: try:
formatted_prediction.remove(slot) if dst.state['system_action'][-1] != slot:
formatted_prediction.append([slot[0],slot[1],dst.state['system_action'][-1][0][2],slot[3]]) formatted_prediction.remove(slot)
formatted_prediction.append([slot[0],slot[1],dst.state['system_action'][-1][0][2],slot[3]])
except:
pass
if slot[2]=='seat': if slot[2]=='seat':
if ',' in slot[3]: if ',' in slot[3]:
seat,row = slot[3].split(',') seat,row = slot[3].split(',')
formatted_prediction.remove(['inform', 'Cinema', 'seat', slot[3]]) formatted_prediction.remove(['inform', 'Cinema', 'seat', slot[3]])
formatted_prediction.append(['inform', 'Cinema', 'seat', seat]) formatted_prediction.append(['inform', 'Cinema', 'seat', seat])
formatted_prediction.append(['inform', 'Cinema', 'row', f' {row}']) formatted_prediction.append(['inform', 'Cinema', 'row', f' {row}'])
if 'rzęd' in user_input_lr: if ('czy' or 'jakie' or 'które') and ('dostęp' or 'woln' or 'zajęt' or 'są') and 'miejsc' in user_input_lr:
dst.update([['offer', 'Cinema', 'task', 'show_seats']])
flag=False
for slot in slots:
if slot[0] in value_dict['train']['seat']:
print(nlg.update([['cinema','offer','seat',slot[0]]]))
flag = True
if not flag:
print(nlg.update([['cinema','offer','seat','']]))
elif 'rzęd' in user_input_lr:
formatted_prediction_tmp = formatted_prediction.copy() formatted_prediction_tmp = formatted_prediction.copy()
user_input_splited = user_input_lr.split() user_input_splited = user_input_lr.split()
for slots in formatted_prediction: for slots in formatted_prediction:
@ -88,7 +101,6 @@ def main():
formatted_prediction.append(['inform', 'Cinema', 'row', user_input_splited[i+1]]) formatted_prediction.append(['inform', 'Cinema', 'row', user_input_splited[i+1]])
elif 'miejsc' in user_input_lr and 'bilet' not in user_input_lr: elif 'miejsc' in user_input_lr and 'bilet' not in user_input_lr:
for i in range(len(user_input_splited)-1): for i in range(len(user_input_splited)-1):
print(user_input_splited[i],user_input_splited[i+1])
if 'rzęd' in user_input_splited[i]: if 'rzęd' in user_input_splited[i]:
try: try:
if isinstance(int(user_input_splited[i+1]),int)and i==0: if isinstance(int(user_input_splited[i+1]),int)and i==0:
@ -109,7 +121,6 @@ def main():
formatted_prediction.append(['inform', 'Cinema', 'seat', user_input_splited[i+1]]) formatted_prediction.append(['inform', 'Cinema', 'seat', user_input_splited[i+1]])
elif 'miejsc' not in user_input_lr and 'bilet' in user_input_lr: elif 'miejsc' not in user_input_lr and 'bilet' in user_input_lr:
for i in range(len(user_input_splited)-1): for i in range(len(user_input_splited)-1):
print(user_input_splited[i],user_input_splited[i+1])
if 'rzęd' in user_input_splited[i]: if 'rzęd' in user_input_splited[i]:
try: try:
if isinstance(int(user_input_splited[i+1]),int)and i==0: if isinstance(int(user_input_splited[i+1]),int)and i==0:
@ -159,7 +170,7 @@ def main():
formatted_prediction.append(['inform', 'Cinema', 'quantity', user_input_splited[i+1]]) formatted_prediction.append(['inform', 'Cinema', 'quantity', user_input_splited[i+1]])
if user_input_lr == '/pomoc': elif user_input_lr == '/pomoc':
print('System Usher pozwala na:\n1. Rezerwację biletu\n2. Anulowanie rezerwacji biletu\n3. Zakup biletu\n4. Anulowanie zakupu biletu\n5. Sprawdzenie repertuaru\n6. Sprawdzenie dostępności miejsc') print('System Usher pozwala na:\n1. Rezerwację biletu\n2. Anulowanie rezerwacji biletu\n3. Zakup biletu\n4. Anulowanie zakupu biletu\n5. Sprawdzenie repertuaru\n6. Sprawdzenie dostępności miejsc')
elif user_input_lr == '/koniec': elif user_input_lr == '/koniec':
@ -240,24 +251,11 @@ def main():
if not flag: if not flag:
print(nlg.update([['cinema','offer','closestscreening','']])) print(nlg.update([['cinema','offer','closestscreening','']]))
elif ('czy' or 'jakie' or 'które') and ('dostęp' or 'woln' or 'zajęt') in user_input_lr:
if 'miejsce' in user_input_lr:
dst.update([['offer', 'Cinema', 'task', 'show_seats']])
flag=False
for slot in slots:
if slot[0] in value_dict['train']['seat']:
print(nlg.update([['cinema','offer','seat',slot[0]]]))
flag = True
if not flag:
print(nlg.update([['cinema','offer','seat','']]))
elif len(formatted_prediction)<1: elif len(formatted_prediction)<1:
print('Nie do końca zrozumiałem mógłbyś/mogłabyś powtórzyć? ') print('Nie do końca zrozumiałem mógłbyś/mogłabyś powtórzyć? ')
else: else:
# print(formatted_prediction) # NLU output
dst.update(formatted_prediction) dst.update(formatted_prediction)
#DP
fl = False fl = False
for action in dst.state['system_action']: for action in dst.state['system_action']:
if 'cancel_book' == action[0][2]: if 'cancel_book' == action[0][2]:
@ -267,7 +265,6 @@ def main():
if fl: if fl:
print("Anulowano. Czy w czymś jeszcze mogę Ci pomóc?") print("Anulowano. Czy w czymś jeszcze mogę Ci pomóc?")
else: else:
for slot,value in dst.state['belief_state']['cinema']['book'].items(): for slot,value in dst.state['belief_state']['cinema']['book'].items():
if value == '': if value == '':