48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
import prof
|
|
from chatterbotapi import ChatterBotFactory, ChatterBotType
|
|
|
|
factory = ChatterBotFactory()
|
|
bot = factory.create(ChatterBotType.CLEVERBOT)
|
|
# bot = factory.create(ChatterBotType.JABBERWACKY)
|
|
# bot = factory.create(ChatterBotType.PANDORABOTS, 'b0dafd24ee35a477')
|
|
bot_session = {}
|
|
bot_state = False
|
|
|
|
def prof_post_chat_message_display(jid, message):
|
|
if bot_state:
|
|
if jid not in bot_session:
|
|
bot_session[jid] = bot.create_session()
|
|
response = bot_session[jid].think(message)
|
|
prof.send_line("/msg " + jid + " " + response)
|
|
|
|
def _cmd_chatterbot(state):
|
|
global bot_state
|
|
|
|
if state == "enable":
|
|
prof.cons_show("ChatterBot Activated")
|
|
bot_state = True
|
|
elif state == "disable":
|
|
prof.cons_show("ChatterBot Stopped")
|
|
bot_state = False
|
|
else:
|
|
if bot_state:
|
|
prof.cons_show("ChatterBot is running - current sessions:")
|
|
prof.cons_show(str(bot_session))
|
|
else:
|
|
prof.cons_show("ChatterBot is stopped - /chatterbot enable to activate.")
|
|
|
|
def prof_init(version, status):
|
|
synopsis = [
|
|
"/chatterbot",
|
|
"/chatterbot enable|disable"
|
|
]
|
|
description = "ChatterBot, running with no args will show the current chatterbot status"
|
|
args = [
|
|
[ "enable", "Enable chatterbot" ],
|
|
[ "disable", "Disable chatterbot" ]
|
|
]
|
|
examples = []
|
|
|
|
prof.register_command("/chatterbot", 0, 1, synopsis, description, args, examples, _cmd_chatterbot)
|
|
prof.completer_add("/chatterbot", [ "enable", "disable" ])
|