Ogólne poprawki

This commit is contained in:
Cezary Gałązkiewicz 2022-06-08 12:58:38 +02:00
parent 887f6249ce
commit 65615ba5c8
3 changed files with 7 additions and 3 deletions

View File

@ -6,8 +6,8 @@ class Nlg:
def __init__(self) -> None: def __init__(self) -> None:
self.responses = { self.responses = {
"welcomemsg": ["I AM TRAIlMinATor!", "Cześć, jestem TrailMinator"], "welcomemsg": ["I AM TRAIlMinATor!", "Cześć, jestem TrailMinator"],
"from": ["Podaj stację poczatkową i końcową", "from": ["Podaj stację poczatkową i końcową przejazdu",
"Podaj stację początkową i końcową zaczynając zdanie od 'chciałbym kupić bilet'"], "Podaj stację początkową i końcową przejazdu zaczynając zdanie od 'chciałbym kupić bilet'"],
"to": ["Podaj stację kocową"], "to": ["Podaj stację kocową"],
"time": ["Podaj w jaki dzień chcesz odbyć podróż"], "time": ["Podaj w jaki dzień chcesz odbyć podróż"],
"passangerCount": ["Podaj ile biletów chcesz kupić"], "passangerCount": ["Podaj ile biletów chcesz kupić"],

View File

@ -58,4 +58,4 @@ class Nlu:
return (act, param) return (act, param)
nlu = Nlu() 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'))

View File

@ -8,6 +8,7 @@ if __name__ == "__main__":
nlu = nlu.Nlu() nlu = nlu.Nlu()
dst = dst.Dst() dst = dst.Dst()
print(nlg.response("welcomemsg"))
response_type = dst.get_next_question() response_type = dst.get_next_question()
while response_type: while response_type:
next_question = nlg.response(response_type) next_question = nlg.response(response_type)
@ -17,6 +18,9 @@ if __name__ == "__main__":
dst.save_answer(res_tokenized['slots']) dst.save_answer(res_tokenized['slots'])
response_type = dst.get_next_question() response_type = dst.get_next_question()
print()
for key, value in dst.checklist.items():
print(f"{key}: {value}")
print(dst.checklist) print(dst.checklist)
# print(dp_params) # print(dp_params)
# nlg_params = dp.Dp(*dp_params).identify() # nlg_params = dp.Dp(*dp_params).identify()