diff --git a/Makiety.py b/Makiety.py index c7ad1ae..4476677 100644 --- a/Makiety.py +++ b/Makiety.py @@ -46,9 +46,11 @@ class NLU: #Natural Language Understanding def test_nlu(self, utterance): matched = self.book_grammar.find_matching_rules(utterance) + print(matched) if matched: return self.get_dialog_act(matched[0]) + else: return {'act': 'null', 'slots': []} @@ -134,7 +136,7 @@ class Janet: self.nlg = NLG(self.acts, self.arguments) self.dp = DP(self.acts, self.arguments) self.dst = DST(self.acts, self.arguments) - self.nlu = NLU(self.acts, self.arguments, jsgf.parse_grammar_file('test_book.jsgf')) + self.nlu = NLU(self.acts, self.arguments, jsgf.parse_grammar_file('book.jsgf')) def test(self, command): out = self.nlu.test_nlu(command) @@ -147,7 +149,8 @@ class Janet: return self.nlg.change_to_text(dest_act) janet = Janet() -print(janet.test('chciałbym zarezerwować stolik na jutro na godzinę dziesiątą dla trzech osób')) +print(janet.test('chciałbym się umówić na wizytę do Piotra Pająka na jutro')) #Testowy print na start while(1): + print('\n') text = input("Wpisz tekst: ") print(janet.test(text)) \ No newline at end of file diff --git a/book.jsgf b/book.jsgf index 3a7704b..c29d583 100644 --- a/book.jsgf +++ b/book.jsgf @@ -4,9 +4,13 @@ grammar book; public = hej | cześć | elo | dzień dobry | witam | siemanko | siemanko witam w mojej kuchni | gitara siema; - = chciałbym się umówić na wizytę ; +public = chciałbym się umówić na wizytę ; - = do [imie_lekarza] {doc_list}; +public = chciałbym się umówić na wizytę ; + + = do {doctor_name}; + + = Jana Kolano | Tomasza Łokieto | Piotra Pająka; = na {day};