System_Dialogowy_Janet/Code/Modules/DP_module.py

from collections import defaultdict
import copy
import json
from copy import deepcopy

class DP():
    """
    Moduł decydujący o wyborze kolejnego aktu, który ma podjąć system prowadząc rozmowę.

    Wejście: Reprezentacja stanu dialogu (rama)

    Wyjście: Akt systemu (rama)
    """
    

    def predict(self, state):
        self.results = []
        system_action = defaultdict(list)
        user_action = defaultdict(list)

        for intent, domain, slot, value in state['user_action']:
            user_action[(domain, intent)].append((slot, value))

        for user_act in user_action:
            self.update_system_action(user_act, user_action, state, system_action)
            

        # # Reguła 3
        # if any(True for slots in user_action.values() for (slot, _) in slots if slot in ['Stay', 'Day', 'People']):
        #     if self.results:
        #         system_action = {('Booking', 'Book'): [["Ref", self.results[0].get('Ref', 'N/A')]]}

        # system_acts = [[intent, domain, slot, value] for (domain, intent), slots in system_action.items() for slot, value in slots]
        # state['system_action'] = system_acts
        # return system_acts

    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)
        print(constraints)

        # Reguła 1

        if domain == 'appointment':
            if len(constraints) == 0:
                    system_action[(domain, 'NoOffer')] = []
            elif intent == 'appointment/create_appointment':
                for x in constraints:
                    if(x[0] == 'doctor'):
                        system_action[(domain, 'book_appointent')].append([x[0], x[1]])
                for y in constraints:
                    if(y[0] == 'datetime'):
                        system_action[(domain, 'book_appointent')].append([y[0], y[1]])
            elif intent == 'appointment/set_date':
                for y in constraints:
                    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_v2:
                    if(x[0] == 'doctor'):
                        system_action[(domain, 'date_info')].append([x[0], x[1]])
                for y in constraints_v2:
                    if(y[0] == 'datetime'):
                        system_action[(domain, 'date_info')].append([y[0], y[1]])
            elif intent == 'request_information/doctors':
                system_action[(domain, 'Doctors_list')] = []
            elif intent == 'request_information/location':
                system_action[(domain, 'Location')] = []
            elif intent == 'request_information/cost':
                system_action[(domain, 'Cost')] = []
            elif intent == 'request_information/opening_hours':
                system_action[(domain, 'Opening_Hours')] = []
            elif len(constraints) == 0:
                system_action[(domain, 'NoOffer')] = []

        #################################################################################################
        # Reguła 3
        if domain == 'end_conversation':
            system_action[(domain, 'End_Conversation')] = []
        
        #################################################################################################
        # Reguła 4
        if domain == 'greeting':
            system_action[(domain, 'Greeting')] = []

        #################################################################################################
        # 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_v2:
                    if(x[0] == 'prescription/type'):
                        system_action[(domain, 'create_prescription')].append([x[0], x[1]])
                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_v2:
                    if(x[0] == 'prescription/type'):
                        system_action[(domain, 'collect_prescription')].append([x[0], x[1]])
            elif intent == 'prescription/type':
                for x in constraints_v2:
                    if(x[0] == 'prescription/type_info'):
                        system_action[(domain, 'prescription type')].append([x[0], x[1]])
            if len(constraints_v2) == 0:
                system_action[(domain, 'NoOffer')] = []

        print(system_action)
                     
        ## Brakuje: Rejestracja, Login, Hasło, affirm, deny
                
                



            # if len(constraints) == 0:
            #     system_action[(domain, 'NoOffer')] = []
            # else:
            #     for slot in user_action[user_act]:
            #         kb_slot_name = REF_SYS_DA[domain].get(slot[0], slot[0])

            #         if kb_slot_name in self.results[0]:
            #             system_action[(domain, 'Inform')].append([slot[0], self.results[0].get(kb_slot_name, 'unknown')])

        # # Reguła 2
        # elif intent == 'Inform':
        #     if len(self.results) == 0:
        #         system_action[(domain, 'NoOffer')] = []
        #     else:
        #         system_action[(domain, 'Inform')].append(['Choice', str(len(self.results))])
        #         choice = self.results[0]

        #         if domain in ["Hotel", "Attraction", "Police", "Restaurant"]:
        #             system_action[(domain, 'Recommend')].append(['Name', choice['name']])