SD-project-22/trailminator/nlg.py

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)]