23 lines
1.1 KiB
Python
23 lines
1.1 KiB
Python
import random
|
|
from urllib import response
|
|
|
|
class Nlg:
|
|
|
|
def __init__(self) -> None:
|
|
self.responses = {
|
|
"welcomemsg": ["I AM TRAIlMinATor!", "Cześć, jestem TrailMinator"],
|
|
"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ć"],
|
|
"discountedCount": ["Podaj ile z tych biletów będzie biletami ze zniżką i jaka zniżka będzie przysługiwała"],
|
|
"seatType": ["Podaj miejsce jakie chcesz mieć w pociągu: pod oknem/w środku/w przejściu"],
|
|
"wagonType": ["Podaj klasę wagonu w jakim chcesz kupić bilety: 1 albo 2 klasa"],
|
|
"null": ["Can't hear you"]
|
|
}
|
|
|
|
def response(self, response_type: str) -> str:
|
|
avail_resps = self.responses.get(response_type)
|
|
return avail_resps[random.randint(0, len(avail_resps)-1)]
|