GOATS/Main.py

from NaturalLanguageAnalyzer import NaturalLanguageAnalyzer
from DialoguePolicy import DialoguePolicy
from DialogueStateTracker import DialogueStateTracker
from NaturalLanguageGeneration import NaturalLanguageGeneration
from convlab.dialog_agent import PipelineAgent
import warnings
warnings.filterwarnings("ignore")


if __name__ == "__main__":
    nlu = NaturalLanguageAnalyzer()
    dst = DialogueStateTracker()
    policy = DialoguePolicy()
    nlg = NaturalLanguageGeneration()

    agent = PipelineAgent(nlu=nlu, dst=dst, policy=policy, nlg=nlg, name='sys')

    print("Witam, jestem systemem do rezerwowania pokoi hotelowych. W czym mogę Ci pomóc?")
    while True:
        text = input()
        response = agent.response(text)
        print(response)