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

Zaktualizuj 'modules/NLG.py'

Browse Source 
fix NLG
This commit is contained in:
Julian Zabłoński 2022-06-08 01:25:36 +02:00
parent 8dc0774ca1
commit 3e87606e84
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/NLG.py
View File

@ -1,6 +1,6 @@
import json import json
from xml import dom 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')) # json.load(open('value_dict.json'))
class NLG: class NLG:
@ -64,6 +64,9 @@ class NLG:
self.answer += f'W rzędzie 4 jest wolne miejsce {value}. ' self.answer += f'W rzędzie 4 jest wolne miejsce {value}. '
else: else:
self.answer += f'W rzędzie 4 nie ma wolnego miejsce {value}. ' 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': elif domain == 'cinema' and intent =='request' and slot == 'seat':
if value in value_dict['train']['seat']: if value in value_dict['train']['seat']:
self.answer += f'Miejsce {value} jest wolne. ' self.answer += f'Miejsce {value} jest wolne. '