diff --git a/NaturalLanguageUnderstanding.py b/NaturalLanguageUnderstanding.py index f79afaf..b9e95e8 100644 --- a/NaturalLanguageUnderstanding.py +++ b/NaturalLanguageUnderstanding.py @@ -10,8 +10,9 @@ class NLU: def get_dialog_act(self, rule): slots = [] self.get_slots(rule.expansion, slots) - print(rule.grammar.name) - return UserAct(UserActType.valueOf(rule.grammar.name.upper()), slots) + print(rule) + print(slots) + return UserAct(UserActType[rule.name.upper()], slots) def get_slots(self, expansion, slots): if expansion.tag != '': diff --git a/UserActType.py b/UserActType.py index e40a708..2d0cc32 100644 --- a/UserActType.py +++ b/UserActType.py @@ -8,4 +8,6 @@ class UserActType(Enum): CREATE_MEETING = 2 CANCEL_MEETING = 3 CHANGE_MEETING = 4 + MEETING_LIST = 5 + CONFIRM = 6 INVALID = -1 diff --git a/book.jsgf b/book.jsgf index 54d2dc7..cb71c65 100644 --- a/book.jsgf +++ b/book.jsgf @@ -2,26 +2,20 @@ grammar book; -public = (zapisać | wpisać | dodać | utworzyć | umówić) nowe* spotkanie; +public = chciałbym (zapisać | wpisać | dodać | utworzyć | umówić) nowe* spotkanie; -public = (anulować | odwołać | usunąć) spotkanie; +public = chciałbym (anulować | odwołać | usunąć) spotkanie; -public = (przesunąć | zmienić | edytować) spotkanie; +public = chciałbym (przesunąć | zmienić | edytować) spotkanie; -public = chciałbym zarezerwować stolik ; +public = jakie spotkania mam {date}; - = na {day}; + = ( | dzisiaj | jutro | pojutrze | ); - = dzisiaj | jutro | poniedziałek | wtorek | środę | czwartek | piątek | sobotę | niedzielę; + = 1|2|3|4|5|6|7|8|9|10|11|12|13|14|15|16|17|18|19|20|21|22|23|24|25|26|27|28|29|30|31; - = na [godzinę] {hour}; + = stycznia | lutego | marca | kwietnia | maja | czerwca | lipca | sierpnia | września | października | listopada | grudnia; - = []; + = w (poniedziałek | środę | czwartek | piątek | sobotę | niedzielę) | we wtorek; - = dziewiątą | dziesiątą | jedenastą | dwunastą; - - = pietnaście | trzydzieści; - - = (na | dla) {size} osób; - - = dwie | dwóch | trzy | trzech | cztery | czterech | pięć | pieciu; \ No newline at end of file +public = potwierdzam | tak \ No newline at end of file