diff --git a/modules/NLG.py b/modules/NLG.py index a564f0e..63b18b7 100644 --- a/modules/NLG.py +++ b/modules/NLG.py @@ -3,11 +3,13 @@ from xml import dom value_dict = json.load(open('modules/value_dict.json')) class NLG: - def __init__(self): + def __init__(self,dst): self.answer = '' + self.dst = dst def update(self, system_act): self.answer = '' + self.dst.state['system_action'].append(system_act) for domain, intent, slot, value in system_act: if domain == 'cinema' and intent == 'inform' and slot == 'seat': if value == '':