diff --git a/dialogue_system.py b/dialogue_system.py index 457dcb3..9b13df3 100644 --- a/dialogue_system.py +++ b/dialogue_system.py @@ -39,11 +39,32 @@ def main(): # WIP while True: user_input = input('> ') - if user_input == '/pomoc': + user_input_lr = user_input.lower() + if user_input_lr == '/pomoc': 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!') 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: slots = nlu.predict_slots(user_input) intent = nlu.predict_intent(user_input)