SpiceBot-migrated/sopel_modules/SpiceBot/AI.py
2019-05-31 20:41:03 -04:00

35 lines
970 B
Python

# coding=utf8
from __future__ import unicode_literals, absolute_import, division, print_function
"""
This is the SpiceBot AI system. Based On Chatty cathy
"""
import os
import aiml
class SpiceBot_AI():
def __init__(self):
self.dict = {
"counts": 0,
}
# Load AIML kernel
self.aiml_kernel = aiml.Kernel()
self.aiml_kernel._verboseMode = False
# Learn responses
import sopel_modules
for plugin_dir in set(sopel_modules.__path__):
configsdir = os.path.join(plugin_dir, "SpiceBot_Configs")
aimldir = os.path.join(configsdir, "aiml")
aimlstarter = os.path.join(aimldir, "std-startup.xml")
self.aiml_kernel.learn(aimlstarter)
self.aiml_kernel.respond("LOAD AIML B")
def on_message(self, message):
aiml_response = self.aiml_kernel.respond(message)
return aiml_response
botai = SpiceBot_AI()