diff --git a/src/components/DP.py b/src/components/DP.py index 826fd80..7f5feed 100644 --- a/src/components/DP.py +++ b/src/components/DP.py @@ -1,9 +1,10 @@ # Martyna class DP: - def __self__(self): - pass + + def __init__(self, slots): + self.slots = slots - def getAct(self, slots): + def getAct(slots): # iterate over slots # find empty # returns system act diff --git a/src/components/DST.py b/src/components/DST.py index 302504d..039ade7 100644 --- a/src/components/DST.py +++ b/src/components/DST.py @@ -1,10 +1,5 @@ # Adrian class DST: - def __self__(self): - pass - - # is it the only one slot that we need? - slots = {'name': None} def getDialogueState(self, userActs): # iterate over speech acts diff --git a/src/components/NLG.py b/src/components/NLG.py index d017344..c142988 100644 --- a/src/components/NLG.py +++ b/src/components/NLG.py @@ -3,5 +3,5 @@ class NLG: def __self__(self): pass - def getResponse(self, systemAct): - return "Witaj, nazywam się Cheddar.\n" \ No newline at end of file + def getResponse(systemAct): + return "Witaj, nazywam się Nachos.\n" \ No newline at end of file diff --git a/src/components/NLU.py b/src/components/NLU.py index 29bb65f..b562355 100644 --- a/src/components/NLU.py +++ b/src/components/NLU.py @@ -2,6 +2,7 @@ from jsgf import PublicRule, Grammar import re class NLU: + def get_str_cleaned(str_dirty): punctuation = '!"#$%&\'()*+,-./:;<=>?@[\\\\]^_`{|}~' new_str = str_dirty.lower() @@ -25,7 +26,3 @@ class NLU: return NLU.getDialogAct(matched[0]) else: return {'act': 'null', 'slots': []} - - - -print(nlu.nlu("Cześć, jak masz na imię?")) \ No newline at end of file diff --git a/src/dialogue_system.py b/src/dialogue_system.py index 5803d13..c4a5e21 100644 --- a/src/dialogue_system.py +++ b/src/dialogue_system.py @@ -1,5 +1,19 @@ from components.NLU import NLU from components.NLG import NLG + +slots = [ + ("title", None), + ("date", None), + ("time", None), + ("quantity", None), + ("location", None), + ("seats", None), + ("reservation_id", None), + ("goal", None), + ("area", None), + ] + + def generate_response(input): result = NLU.nlu(input) return result