mockup #7

Merged
s444519 merged 19 commits from mockup into master 2022-05-04 12:47:45 +02:00
2 changed files with 32 additions and 27 deletions
Showing only changes of commit 8401deb358 - Show all commits

View File

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

View File

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