43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
from service.dialog_state_monitor import DialogStateMonitor
|
|
from service.dialog_policy import DialogPolicy
|
|
from service.natural_languag_understanding import NaturalLanguageUnderstanding
|
|
from service.natural_language_generation import NaturalLanguageGeneration, parse_frame
|
|
from service.templates import templates
|
|
from convlab.dialog_agent import PipelineAgent
|
|
|
|
# initialize classes
|
|
nlu = NaturalLanguageUnderstanding() # NLU
|
|
monitor = DialogStateMonitor() # DSM
|
|
dialog_policy = DialogPolicy() # DP
|
|
language_generation = NaturalLanguageGeneration(templates) # NLG
|
|
|
|
|
|
# Main loop
|
|
dial_num = 0
|
|
print("CTRL+C aby zakończyć program.")
|
|
while True:
|
|
monitor.reset()
|
|
|
|
print(f"\n==== Rozpoczynasz rozmowę nr {dial_num} ====\n")
|
|
user_input = input("Możesz zacząć pisać.\n")
|
|
|
|
while True:
|
|
# NLU
|
|
frame = nlu.predict(user_input)
|
|
print("Frame: ", frame)
|
|
|
|
# DSM
|
|
monitor.update(frame)
|
|
|
|
# DP
|
|
system_action = dialog_policy.predict(monitor)
|
|
print("System action: ", system_action)
|
|
# NLG
|
|
act, slots = parse_frame(frame)
|
|
response = language_generation.generate(act, slots)
|
|
print(response)
|
|
|
|
if frame.act == "bye":
|
|
break
|
|
|
|
user_input = input(">\n") |