Zaktualizuj 'nlu.py'

nlu.py

@@ -11,4 +11,27 @@ matched = order_grammar.find_matching_rules(utterance)
 def get_dialog_act(rule):
     slots = []
     get_slots(rule.expansion, slots)
-    return {'act': rule.grammar.name, 'slots': slots}
+    return {'act': rule.grammar.name, 'slots': slots}
+
+
+def get_slots(expansion, slots):
+    if expansion.tag != '':
+        slots.append((expansion.tag, expansion.current_match))
+        return
+
+    for child in expansion.children:
+        get_slots(child, slots)
+
+    if not expansion.children and isinstance(expansion, jsgf.NamedRuleRef):
+        get_slots(expansion.referenced_rule.expansion, slots)
+
+
+def nlu(_utterance):
+    _matched = order_grammar.find_matching_rules(_utterance)
+
+    if _matched:
+        return get_dialog_act(matched[0])
+    else:
+        return {'act': 'null', 'slots': []}
+
+print(nlu('chciałbym zamowic pizze vesuvio XXL na dwie osoby'))