SystemyDialogowe/modules/NLG.py

import random

unknown = ['Nie rozumiem. Proszę o powtórzenie.',
          'Czy możesz powiedzieć to innymi słowami?',
          "Nie wiem co masz na myśli",
           "Czy możesz powiedzieć to w inny sposób, proszę?"]

bye =  ["Trzymaj się!",
         "Na razie!",
        "Do zobaczenia później!",
         "Do zobaczenia",
        "Pogadamy później!"
       ]




def nlg(query,frame,text):
    if frame['act'] == "hello":
        print("Witamy w systemie rezerwacji biletów kinowych. W czym mogę pomóc?")
    if frame['act'] == "repertuar":
        print("Aktualnie repertuar zawiera takie filmy jak: \"Batman\" \"Ambulans\" \"Bunkier strachu\" \"Córka\" \"Uncharted\" \"Inni ludzie\" \"Śmierć na Nilu\" \"Skarb Mikołajka\"")
    
    if frame['act'] == 'null':
        if text == 'reset':
            print("Resetowanie dialogu.")
        if text == "exit":
            print('Dziękujemy za skorzystanie z naszych usług!')
        elif text != "reset":
            print(random.choice(unknown))
    
    if frame['act'] == "numer":
        if len(text) != 9:
            print("Proszę podać poprawny numer telefonu.")
        elif query == None:
            print("Rezerwacja została dokonana. Potwierdzenie rezerwacji zostanie niebawem przesłane wiadomością SMS. Proszę o przybycie 15 minut przed rozpoczęciem seansu w celu zakupu biletu, w innym przypadku rezerwacja przepada.")
    
    if frame['act'] == "bye" :
        print(random.choice(bye))

        
#     if frame['act'] != "hello" and frame['act'] != "null" and :
    if frame['act'] not in ['hello','bye','null']:
        if query == "miejsce":
            print("Czy miejsca mają być z przodu, z tyłu czy na środku?")
        if query == "numer":
            print("Na jaki numer telefonu ma zostać dokonana rezerwacja?")
        if query == 'dzien':
            print("Na który dzień ma być dokonana rezerwacja?")
        if query == 'godzina':
            print("Na którą godzinę ma być dokonana rezerwacja?")
        if query == 'ilosc':
            print("Ile biletów ma zostać zarezerwowanych?")
        if query == 'tytul':
            print("Na jaki film ma zostać dokonana rezerwacja?")