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

Zaktualizuj 'modules/NLG.py'

Browse Source 
fix
This commit is contained in:
Julian Zabłoński 2022-06-08 08:19:10 +02:00
parent a5bb41834a
commit 18841bcad3
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
modules/NLG.py
View File

@ -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. '