From 18841bcad3ff12abe310f0c6204c61d72f14feda Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Julian=20Zab=C5=82o=C5=84ski?= Date: Wed, 8 Jun 2022 08:19:10 +0200 Subject: [PATCH] Zaktualizuj 'modules/NLG.py' fix --- modules/NLG.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/modules/NLG.py b/modules/NLG.py index 613ec36..27f8063 100644 --- a/modules/NLG.py +++ b/modules/NLG.py @@ -1,8 +1,7 @@ import json from xml import dom -value_dict = json.load(open(r'C:\Users\User\VisualStudio\Diagramy\aitech-sd-lab\modules\value_dict.json')) +value_dict = json.load(open('value_dict.json')) -# json.load(open('value_dict.json')) class NLG: def __init__(self): self.answer = '' @@ -28,6 +27,9 @@ class NLG: self.answer += f'Proszę podać e-mail na który była rezerwacja miejsce. ' else: self.answer += f'Rezerwacja nie została anulowana. ' + elif domain == 'cinema' and intent == 'request' and slot == 'movie': + if value == '': + self.answer += f'Na jaki film chciałbyś/abyś pójść. ' elif domain == 'cinema' and intent == 'inform' and slot == 'movie': if value == '': self.answer += f'Nie znalazłem filmu. ' @@ -64,9 +66,9 @@ class NLG: self.answer += f'W rzędzie 4 jest wolne miejsce {value}. ' else: self.answer += f'W rzędzie 4 nie ma wolnego miejsce {value}. ' - elif domain == 'cinema' and intent =='inform' and slot == 'seat': + elif domain == 'cinema' and intent =='offer' and slot == 'seat': if value in value_dict['train']['seat']: - self.answer += 'Sala jest obłożona w 15%. ' + self.answer += 'Sala jest obłożona w 15%. Głównie w rzędach 4, 5 i 6. ' elif domain == 'cinema' and intent =='request' and slot == 'seat': if value in value_dict['train']['seat']: self.answer += f'Miejsce {value} jest wolne. '