GOATS/Main.py

25 lines
704 B
Python

from NaturalLanguageAnalyzer import NaturalLanguageAnalyzer
from DialogueStateTracker import DialogueStateTracker
from DialoguePolicy import DialoguePolicy
from NaturalLanguageGeneration import NaturalLanguageGeneration
if __name__ == "__main__":
text = "chciałbym zarezerwować pokój z balkonem 1 stycznia w Warszawie"
nla = NaturalLanguageAnalyzer()
user_act = nla.process(text)
print(user_act)
# dst = DialogueStateTracker()
# state = dst.dst(user_act)
# print(state)
#
# dp = DialoguePolicy()
# system_act = dp.policy(state)
# print(system_act)
#
# nlg = NaturalLanguageGeneration()
# response = nlg.nlg(system_act)
# print(response)