def util_find_reponses(self, naoDialogModel, dialog_scenario):
naoDialogContext = model.NaoDialogContext()
naoDialogUtil = model.NaoDialogUtil()
response_arr = []
for user_step, bot_step in dialog_scenario.items():
response = naoDialogUtil.find_response(naoDialogModel,
naoDialogContext, user_step)
response_arr.append(response)
self.assertEqual(bot_step, response.responseText)
return response_arr
def train_dialog(self):
(gram_str, sphinx_dictionary_str
) = model.NaoDialogUtil().generate_sphinx_resouces(
self.naoDialogModel)
f = open("/tmp/liepa_dialog.gram", "w")
f.write(gram_str)
f.close()
f = open("/tmp/liepa_dialog.dict", "w")
f.write(sphinx_dictionary_str)
f.close()
self.logger.info("[train_dialog] ", gram_str)
def on_phrase_recognized(self, phrase):
#print(phrase)
self.logger.info("[on_phrase_recognized] ", phrase)
naoDialogResponse = model.NaoDialogUtil().find_response(
self.naoDialogModel, self.naoDialogContext, phrase)
self.say(naoDialogResponse.responseText)
eventValue = "NA"
if (naoDialogResponse.eventValue):
eventValue = naoDialogResponse.eventValue
self.memory.raiseEvent("Libot/DialogEvent", eventValue)
self.memory.raiseEvent("Libot/DialogStepDone",
naoDialogResponse.ruleId)
def test_dialog_generate_resouces(self):
dialog_str = """topic: ~test_dialog()
language: Lithuanian
concept:(greetings) ^rand[sveiki "laba diena"]
u:(~greetings) ~greetings
u:(Kaip tau sekasi) Normoje
u:(kuri diena) geroji"""
naoDialogTrainer = trainer.NaoDialogTrainer()
naoDialogModel = naoDialogTrainer.train(dialog_str)
(gram, sphinx_dictionary
) = model.NaoDialogUtil().generate_sphinx_resouces(naoDialogModel)
self.assertEqual(
gram,
"#JSGF V1.0;\n\ngrammar adr_element;\n\npublic <adr_element> =\nsveiki|\nlaba diena|\nkaip tau sekasi|\nkuri diena;"
)
self.assertEqual(
sphinx_dictionary,
"diena\tD I E N A\nkaip\tK A I P\nkuri\tK U R I\nlaba\tL A B A\nsekasi\tS E K A S I\nsveiki\tS V E I K I\ntau\tT A U"
)
def main():
# dialog_str = """topic: ~test_dialog()
# language: ltu
# u:(Labas) Sveiki
# u:(Kaip tau sekasi) Normoje
# u:(kuri diena) geroji
# u:(e:Dialog/Fallback) Neturiu atsakymo"""
f = open("./test/topics/game.top", "r")
# f = open("./test/topics/simple.top", "r")
dialog_str = f.read()
f.close()
naoDialogTrainer = trainer.NaoDialogTrainer()
naoDialogModel = naoDialogTrainer.train(dialog_str)
naoDialogContext = model.NaoDialogContext()
naoDialogUtil = model.NaoDialogUtil()
chart = naoDialogTrainer.generate_dialog_chart(naoDialogModel)
print(chart)
warmup = ["labas",
"Noriu pradėti žaidimą",
"ne",
"ne",
"taip"]
for line in warmup:
naoDialogUtil.find_response(naoDialogModel, naoDialogContext, line)
while True:
val = input("💬 > ")
response=naoDialogUtil.find_response(naoDialogModel, naoDialogContext, val)
print("🤖\t" + response.responseText)
if(response.eventValue):
print("🤖\t\t eventValue: " + response.eventValue)
def setUp(self):
self.naoDialogUtil = model.NaoDialogUtil()
pass