From 3e87606e84aa4df9c75cab2692feeef12a6b5443 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Julian=20Zab=C5=82o=C5=84ski?= Date: Wed, 8 Jun 2022 01:25:36 +0200 Subject: [PATCH] Zaktualizuj 'modules/NLG.py' fix NLG --- modules/NLG.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/modules/NLG.py b/modules/NLG.py index ae7efc5..613ec36 100644 --- a/modules/NLG.py +++ b/modules/NLG.py @@ -1,6 +1,6 @@ import json from xml import dom -value_dict = json.load(open('value_dict.json')) +value_dict = json.load(open(r'C:\Users\User\VisualStudio\Diagramy\aitech-sd-lab\modules\value_dict.json')) # json.load(open('value_dict.json')) class NLG: @@ -64,6 +64,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': + if value in value_dict['train']['seat']: + self.answer += 'Sala jest obłożona w 15%. ' elif domain == 'cinema' and intent =='request' and slot == 'seat': if value in value_dict['train']['seat']: self.answer += f'Miejsce {value} jest wolne. '