SD-project-22/trailminator/nlg.py

15 lines
482 B
Python

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