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

add simple task recognition

Browse Source
This commit is contained in:
Kacper 2022-06-07 16:39:01 +02:00
parent fa345178a8
commit 4b02834daf
1 changed files with 23 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
dialogue_system.py
View File

@ -39,11 +39,32 @@ def main():
# WIP # WIP
while True: while True:
user_input = input('> ') user_input = input('> ')
if user_input == '/pomoc': user_input_lr = user_input.lower()
if user_input_lr == '/pomoc':
print('TEKST_POMOCY_WIP') print('TEKST_POMOCY_WIP')
elif user_input == '/koniec': elif user_input_lr == '/koniec':
print('Dziękuję za skorzystanie z moich usług. Miłego dnia!') print('Dziękuję za skorzystanie z moich usług. Miłego dnia!')
break break
elif 'rezerw' in user_input_lr:
if 'anulo' in user_input_lr:
dst.update([['inform', 'Cinema', 'task', 'cancel_book']])
print(dst.state)
else:
dst.update([['inform', 'Cinema', 'task', 'book']])
print(dst.state)
elif 'kup' in user_input_lr or ('zwr' and 'bilet') in user_input_lr:
if 'anulo' in user_input_lr or 'zwr' in user_input_lr:
dst.update([['inform', 'Cinema', 'task', 'cancel_buy']])
print(dst.state)
else:
dst.update([['inform', 'Cinema', 'task', 'buy']])
print(dst.state)
elif (('jak' or 'któr') and 'film') in user_input_lr or 'repertuar' in user_input_lr:
dst.update([['inform', 'Cinema', 'task', 'show_movies']])
print(dst.state)
elif (('czy' or 'jakie' or 'które') and ('dostęp' or 'woln' or 'zajęt') and 'miejsc') in user_input_lr:
dst.update([['inform', 'Cinema', 'task', 'show_seats']])
print(dst.state)
else: else:
slots = nlu.predict_slots(user_input) slots = nlu.predict_slots(user_input)
intent = nlu.predict_intent(user_input) intent = nlu.predict_intent(user_input)