Source code for sugaroid.brain.ok

from chatterbot.logic import LogicAdapter

from sugaroid.brain.ooo import Emotion
from sugaroid.brain.postprocessor import random_response
from sugaroid.brain.preprocessors import normalize
from sugaroid.core.statement import SugaroidStatement


[docs]class OkayAdapter(LogicAdapter): """ Handles statements with a plain old okay """ def __init__(self, chatbot, **kwargs): super().__init__(chatbot, **kwargs)
[docs] def can_process(self, statement): normalized = normalize(str(statement)) if "ok" in normalized or "okay" in normalized: return True else: return False
[docs] def process(self, statement, additional_response_selection_parameters=None): confidence = 0.7 ls = ["😀", "😁", "😂", "😏", "😝"] selected_statement = SugaroidStatement( "ok ok {}".format(random_response(ls)), chatbot=True ) selected_statement.confidence = confidence emotion = Emotion.wink selected_statement.emotion = emotion return selected_statement