diff --git a/trailminator/nlg.py b/trailminator/nlg.py index c385a4f..39af81f 100644 --- a/trailminator/nlg.py +++ b/trailminator/nlg.py @@ -6,8 +6,8 @@ class Nlg: def __init__(self) -> None: self.responses = { "welcomemsg": ["I AM TRAIlMinATor!", "Cześć, jestem TrailMinator"], - "from": ["Podaj stację poczatkową i końcową", - "Podaj stację początkową i końcową zaczynając zdanie od 'chciałbym kupić bilet'"], + "from": ["Podaj stację poczatkową i końcową przejazdu", + "Podaj stację początkową i końcową przejazdu zaczynając zdanie od 'chciałbym kupić bilet'"], "to": ["Podaj stację kocową"], "time": ["Podaj w jaki dzień chcesz odbyć podróż"], "passangerCount": ["Podaj ile biletów chcesz kupić"], diff --git a/trailminator/nlu.py b/trailminator/nlu.py index feaea91..2361e80 100644 --- a/trailminator/nlu.py +++ b/trailminator/nlu.py @@ -58,4 +58,4 @@ class Nlu: return (act, param) nlu = Nlu() -print(nlu.tokenize('chciałbym kupić bilet na pociąg z Poznan do Krakow')) +#print(nlu.tokenize('chciałbym kupić bilet na pociąg z Poznan do Krakow')) diff --git a/trailminator/trailminator.py b/trailminator/trailminator.py index 6e39e1a..df9b7ce 100644 --- a/trailminator/trailminator.py +++ b/trailminator/trailminator.py @@ -8,6 +8,7 @@ if __name__ == "__main__": nlu = nlu.Nlu() dst = dst.Dst() + print(nlg.response("welcomemsg")) response_type = dst.get_next_question() while response_type: next_question = nlg.response(response_type) @@ -17,6 +18,9 @@ if __name__ == "__main__": dst.save_answer(res_tokenized['slots']) response_type = dst.get_next_question() + print() + for key, value in dst.checklist.items(): + print(f"{key}: {value}") print(dst.checklist) # print(dp_params) # nlg_params = dp.Dp(*dp_params).identify()