add modules

modules/AJN.py

@@ -0,0 +1,50 @@
+import jsgf
+from os import listdir
+from os.path import isfile, join
+
+mypath = "./grammar/"
+onlyfiles = [f for f in listdir(mypath) if isfile(join(mypath, f))]
+
+grammars = []
+
+for grammarFile in onlyfiles:
+    grammar = jsgf.parse_grammar_file(mypath + grammarFile)
+    grammars.append(grammar)
+
+class Ajn:        
+ 
+        
+            
+    def __init__(self, grammars = None):
+        self.grammars = grammars
+
+    def get_dialog_act(rule):
+        slots = []
+        get_slots(rule.expansion, 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 = None
+        for grammar in grammars:
+            matched = grammar.find_matching_rules(utterance)
+            if matched:
+                break
+
+        if matched:
+            return get_dialog_act(matched[0])
+        else:
+            return {'act': 'null', 'slots': []}
+
+
+ajn = Ajn()

modules/DST.py

@@ -0,0 +1,27 @@
+class Dst:
+    def __init__(self):
+        self.messages = []
+        self.checklist = {
+            "ilosc": None,
+            "tytul": None,
+            "dzien": None,
+            "godzina": None
+        }
+        self.history = []
+    
+    def store(self, message):
+        self.messages.append(message)
+
+    def get_messages(self):
+        return self.messages
+
+    def get_next_question(self):
+        for key, value in self.checklist.items():
+            if value == None:
+                return key
+
+    def save_answer(self, slots):
+        for slot in slots:
+            self.checklist[slot[0]] = slot[1]
+
+        self.messages.append(slots)