23 lines
762 B
Python
23 lines
762 B
Python
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)
|