Even better DP!!!!

Code/Modules/DP_module.py

@@ -37,6 +37,7 @@ class DP():
     def update_system_action(self, user_act, user_action, state, system_action):
         domain, intent = user_act
         constraints = [(slot, value) for slot, value in state['belief_state'][domain.lower()].items() if value != '']
+        
 
         print(domain)
         print(intent)
@@ -59,15 +60,18 @@ class DP():
                     if(y[0] == 'datetime'):
                         system_action[(domain, 'set_appointment_date')].append([y[0], y[1]])
 
+
+        
         #################################################################################################
         # Reguła 2
 
         if domain == 'request_information':
+            constraints_v2 = [(slot, value) for slot, value in state['request_state'][domain.lower()].items() if value != '']
             if intent == 'request_information/available_dates':
-                for x in constraints:
+                for x in constraints_v2:
                     if(x[0] == 'doctor'):
                         system_action[(domain, 'date_info')].append([x[0], x[1]])
-                for y in constraints:
+                for y in constraints_v2:
                     if(y[0] == 'datetime'):
                         system_action[(domain, 'date_info')].append([y[0], y[1]])
             elif intent == 'request_information/doctors':
@@ -94,22 +98,23 @@ class DP():
         #################################################################################################
         # Reguła 5
         if domain == 'prescription':
+            constraints_v2 = [(slot, value) for slot, value in state['request_state'][domain.lower()].items() if value != '']
             if intent == 'prescription/request':
-                for x in constraints:
+                for x in constraints_v2:
                     if(x[0] == 'prescription/type'):
                         system_action[(domain, 'create_prescription')].append([x[0], x[1]])
-                for y in constraints:
+                for y in constraints_v2:
                     if(y[0] == 'prescription/medicine'):
                         system_action[(domain, 'create_prescription')].append([y[0], y[1]])
             elif intent == 'prescription/collect':
-                for x in constraints:
+                for x in constraints_v2:
                     if(x[0] == 'prescription/type'):
                         system_action[(domain, 'collect_prescription')].append([x[0], x[1]])
             elif intent == 'prescription/type':
-                for x in constraints:
+                for x in constraints_v2:
                     if(x[0] == 'prescription/type_info'):
                         system_action[(domain, 'prescription type')].append([x[0], x[1]])
-            if len(constraints) == 0:
+            if len(constraints_v2) == 0:
                 system_action[(domain, 'NoOffer')] = []
 
         print(system_action)

Code/Modules/DST_module.py

@@ -35,12 +35,22 @@ class Rules_DST(): #Dialogue State Tracker
                 if(domain in slot):
                     slot.replace(domain + "/", '')
 
-                domain_dic = self.state['belief_state'][domain]
-                if slot in domain_dic:
-                    self.state['belief_state'][domain][slot] = value
+                if domain not in self.state['request_state']:
+                    self.state['request_state'][domain] = {}
+                if slot not in self.state['request_state'][domain]:
+                    self.state['request_state'][domain][slot] = value
 
-            elif intent == 'end_conversation':
-                self.state = json.load(open('default_state.json'))
+            if 'request_information' in intent:
+                if (slot == 'request_information'):
+                    continue
+                
+                if(domain in slot):
+                    slot.replace(domain + "/", '')
+
+                if domain not in self.state['request_state']:
+                    self.state['request_state'][domain] = {}
+                if slot not in self.state['request_state'][domain]:
+                    self.state['request_state'][domain][slot] = value
             
         print(self.state)
         return self.state

default_state.json

@@ -19,6 +19,12 @@
         "prescription": {
             "prescription/type": "",
             "prescription/medicine": ""
+        },
+        "request_information":{
+            "doctor": "",
+            "datetime": "",
+            "appointment": "",
+            "appointment/type": ""
         }
     },
     "request_state": {},