import json from xml import dom 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: def __init__(self): self.answer = '' def update(self, system_act): self.answer = '' for domain, intent, slot, value in system_act: if domain == 'cinema' and intent == 'inform' and slot == 'seat': if value == '': self.answer += f'Gdzie dokładnie chciałabyś/byś zarezezrwować miejsce. ' elif value == "top": self.answer += f'Wolne miejsca na górze w rzędzie 10 miejsca od 5 do 10 i od 12 do 14. ' elif value <= "bottom": self.answer += f'Wolne miejsca na dole w rzędzie 1 miejsca od 2 do 10 i od 12 do 14. ' elif value <= "middle": self.answer += f'Wolne miejsca na środku w rzędzie 6 miejsca od 5 do 10. ' else: self.answer += f'Nie znalazłem miejsca. ' elif domain == 'cinema' and intent == 'inform' and slot == 'cancel_book': if len(value): self.answer += f'Rezerwacja na maila: {value} została anulowana. ' elif value == '': 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 == 'inform' and slot == 'movie': if value == '': self.answer += f'Nie znalazłem filmu. ' else: self.answer += f'Film {value} będzie wyświetlany jutro o godzinie 21:30. ' elif domain == 'cinema' and intent == 'offer' and slot == 'closestscreening': if value in ['Dziś','Jutro','Pojutrze']: self.answer += f'{value} prezentujemy Batmana o 18:30 i Ambulans o 21:00. ' elif value in value_dict['train']['day']: self.answer += f'W {value} prezentujemy Ambulans o 18:30 i To nie wypanda o 20:00. ' elif value in value_dict['train']['date']: self.answer += f'Dnia {value} prezentujemy Batmana o 18:00 i Sing 2 o 20:30. ' elif domain == 'cinema' and intent == 'request' and slot == 'e-mail': if value== '': self.answer += 'Prosze podać e-mail. ' elif domain == 'cinema' and intent == 'request' and slot == 'date': if value== '': self.answer += 'Którego dnia chciałabyś/byś pani obejrzeć ten film. ' elif domain == 'cinema' and intent == 'request' and slot == 'name': if value== '': self.answer += 'Prosze podać imie i nazwisko. ' elif domain == 'cinema' and intent == 'request' and slot == 'phone': if value== '': self.answer += 'Prosze podać numer telefonu. ' elif domain == 'cinema' and intent =='request' and slot == 'row': if value in value_dict['train']['seat_row']: self.answer += f'W rzędzie {value} są wolne miejsca od 1 do 4 i od 12 do 15. ' elif value == '': self.answer += f'W jakim rzędzie chciałbyś/abyś zarezerwować miejsce. ' else: self.answer += f'Nie ma wolnych miejsc w rzędzie {value}. ' elif domain == 'cinema' and intent =='request' and slot == 'seat_place': if value in value_dict['train']['seat_place']: 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 =='request' and slot == 'seat': if value in value_dict['train']['seat']: self.answer += f'Miejsce {value} jest wolne. ' elif value == '': self.answer += f'Jakie miejsce chciałbyś/abyś zarezerwować. ' else: self.answer += f'Miejsce {value} nie jest wolne. ' elif domain == 'cinema' and intent == 'inform' and slot == 'Location': self.answer += 'Kino znajduję się na ulicy Jana Pawła 2/40. ' elif domain == 'cinema' and intent == 'inform' and slot == 'cinema_name': self.answer += 'Kino nazywa sie Hellios' elif domain == 'cinema' and intent == 'offer' and slot == 'time': if value in ['15:00','12:00','18:00','18:30','19:00','20:00','20:30','21:00','21:30']: self.answer += f'Dostępny jest seans na godzine {value}. ' else: self.answer += f'Nie mamy aktualnie seansu na taką godzinę. ' elif domain == 'cinema' and intent == 'request' and slot == 'tickettype': if value == '': self.answer += 'Czy chciałabyś/byś bilet normalny czy ulgowy? ' elif domain == 'cinema' and intent == 'request' and slot == 'quantity': if value == '': self.answer += 'Ile biletów chciałabyś/byś zarezerwować. ' elif domain == 'cinema' and intent == 'request' and slot == 'hour': if value == '': self.answer += 'O jakiej godzinie chciałabyś/byś zobaczyć film. ' elif domain == 'cinema' and intent == 'request' and slot == 'payments': if value == 'online': self.answer += 'Proszę dokonać płatność online. Dziękujemy za skorzystanie z naszych usług. ' else: self.answer += 'Proszę stawić się 30 minut przed seansem aby dokonać płatność. Dziękujemy za skorzystanie z naszych usług. ' return self.answer # nlg = NLG() # print(nlg.update([['cinema', 'offer', 'Closestscreening', "1.02"],['cinema', 'request', 'name', ""]]))