src/components/DST.py

@@ -1,15 +1,12 @@
 # Adrian
 class DST:
+    def __init__(self, slots):
+        self.slots = slots
 
     def getDialogueState(self, userActs):
         # iterate over speech acts
-
-        """for k, v in userActs['inform'].items():
-            for slot in self.slots:
-                if slot == k:
-                    self.slots['name'] = v
-        print(self.slots)"""
         if userActs['act'] == 'book':
+            if userActs['slots']:
                 for (k, v) in userActs['slots']:
                     print(k)
                     if k in self.slots:
@@ -19,7 +16,7 @@ class DST:
         return self.slots
 
 
-dst = DST()
+#dst = DST()
 # userActs = {'inform': {'name': 'edyta', 'age': 18}, 'reqmore': {'date', 'time'}}
-ivona_acts={'act':'book','slots':[('name','ewa'), ('hour','dziesiąta'), ('size','trzech')]}
-print(dst.getDialogueState(ivona_acts))
+#ivona_acts = {'act': 'book', 'slots': [('name', 'ewa'), ('hour', 'dziesiąta'), ('size', 'trzech')]}
+#print(dst.getDialogueState(ivona_acts))

src/dialogue_system.py

@@ -1,5 +1,6 @@
 from components.NLU import NLU
 from components.NLG import NLG
+from components.DST import DST
 
 slots = [
     ("title", None),
@@ -15,8 +16,15 @@ slots = [
 
 
 def generate_response(input):
-    result = NLU.nlu(input)
-    return result
+    # nlu
+    nlu = NLU.nlu(input)
+
+    # dst
+    dst_obj = DST(slots)
+    dst = dst_obj.getDialogueState(nlu)
+
+    return dst
+
 
 inputText = 'Cześć, jak masz na imię?'
 print(NLG.getResponse(generate_response(inputText)))