System_Dialogowy_Janet/Code/Modules/DST_module.py

import json

class Rules_DST(): #Dialogue State Tracker
    """
    Moduł odpowiedzialny za śledzenie stanu dialogu. Przechowuje informacje o tym jakie dane zostały uzyskane od użytkownika w toku prowadzonej konwersacji.

    Wejście: Akt użytkownika (rama)

    Wyjście: Reprezentacja stanu dialogu (rama)
    """
    def __init__(self):
        self.state = json.load(open('default_state.json'))

    def update(self, user_acts=None):
        for intent, domain, slot, value in user_acts:
            self.state["user_action"].append([intent, domain, slot, value])
            if domain in ['password', 'name', 'email', 'enter_email', 'enter_name']:
                return

            if 'appointment' in intent:
                if (slot == 'appointment'):
                    continue
                
                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 'prescription' in intent:
                if (slot == 'prescription'):
                    continue
                
                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

            elif intent == 'end_conversation':
                self.state = json.load(open('default_state.json'))
            
        print(self.state)
        return self.state