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 import json
from xml import dom 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: class NLG:
def __init__(self): def __init__(self):
self.answer = '' self.answer = ''
@ -28,6 +27,9 @@ class NLG:
self.answer += f'Proszę podać e-mail na który była rezerwacja miejsce. ' self.answer += f'Proszę podać e-mail na który była rezerwacja miejsce. '
else: else:
self.answer += f'Rezerwacja nie została anulowana. ' 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': elif domain == 'cinema' and intent == 'inform' and slot == 'movie':
if value == '': if value == '':
self.answer += f'Nie znalazłem filmu. ' self.answer += f'Nie znalazłem filmu. '
@ -64,9 +66,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': elif domain == 'cinema' and intent =='offer' and slot == 'seat':
if value in value_dict['train']['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': 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. '