import jsgf

from Modules.NLG_module import NLG
from Modules.DP_module import DP
from Modules.DST_module import DST
from Modules.Book_NLU_module import Book_NLU
from Modules.ML_NLU_module import ML_NLU

import random
import torch
random.seed(42)
torch.manual_seed(42)

if torch.cuda.is_available():
    torch.cuda.manual_seed(0)
    torch.cuda.manual_seed_all(0)
    torch.backends.cudnn.enabled = False
    torch.backends.cudnn.benchmark = False
    torch.backends.cudnn.deterministic = True

class Janet:
    def __init__(self):
        self.acts={
            0: "greetings",
            1: "request",
        }
        self.arguments={
            0: "name"
        }
        self.nlg = NLG(self.acts, self.arguments)
        self.dp = DP(self.acts, self.arguments)
        self.dst = DST(self.acts, self.arguments)
        self.nlu = Book_NLU(self.acts, self.arguments, jsgf.parse_grammar_file('book.jsgf'))
        self.nlu_v2 = ML_NLU(self.acts, self.arguments)

    def test(self, command):
        out = self.nlu_v2.test_nlu(command)
        return out

    def process(self, command):
        act = self.nlu.analyze(command)
        self.dst.store(act)
        dest_act = self.dp.choose_tactic(self.dst.transfer())
        return self.nlg.change_to_text(dest_act)


def main():
    janet = Janet()
    while(1):
        print('\n')
        text = input("Wpisz tekst: ")
        print(janet.test(text))

if __name__ == "__main__":
    main()