2022-04-20 11:25:16 +02:00
|
|
|
import random
|
|
|
|
from urllib import response
|
2022-05-18 12:33:50 +02:00
|
|
|
|
2022-04-13 13:23:47 +02:00
|
|
|
class Nlg:
|
2022-04-20 11:25:16 +02:00
|
|
|
|
2022-06-01 16:25:19 +02:00
|
|
|
def __init__(self) -> None:
|
|
|
|
self.responses = {
|
|
|
|
"welcomemsg": ["I AM TRAIlMinATor!", "Cześć, jestem TrailMinator"],
|
2022-06-07 23:11:29 +02:00
|
|
|
"from": ["Podaj stację poczatkową i końcową",
|
|
|
|
"Podaj stację początkową i końcową 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"]
|
|
|
|
}
|
2022-06-01 16:25:19 +02:00
|
|
|
|
|
|
|
def response(self, response_type: str) -> str:
|
|
|
|
avail_resps = self.responses.get(response_type)
|
2022-05-18 12:34:59 +02:00
|
|
|
return avail_resps[random.randint(0, len(avail_resps)-1)]
|