SD-project-22/trailminator/nlg.py

import random
from urllib import response

from scipy import rand

class Nlg:
    def __init__(self, response_type: str) -> None:
        self._response_type = response_type
        self.responses = {"welcomemsg": ["I AM TRAIlMinATor!", "Cześć, jestem TrailMinator"],
                          "null": "Can't hear you"}

    def response(self) -> str:
        avail_resps = self.responses.get(self._response_type)
        return avail_resps[random.randint(0, len(avail_resps)-1)]