mirror of
https://github.com/zulip/zulip.git
synced 2025-11-16 03:41:58 +00:00
5480f6fd0f
The send_reply function makes it easier for bots to send messages. This commit updates all bots to make use of this function, when possible.
124 lines
4.0 KiB
Python
124 lines
4.0 KiB
Python
import json
|
|
import os
|
|
import sys
|
|
|
|
from random import choice
|
|
|
|
try:
|
|
from chatterbot import ChatBot
|
|
from chatterbot.trainers import ChatterBotCorpusTrainer, ListTrainer
|
|
except ImportError:
|
|
raise ImportError("""It looks like you are missing chatterbot.
|
|
Please: pip install chatterbot""")
|
|
|
|
CONTRIB_BOTS_DIR = os.path.dirname(os.path.abspath(__file__))
|
|
os.chdir(os.path.dirname(CONTRIB_BOTS_DIR))
|
|
sys.path.insert(0, os.path.dirname(CONTRIB_BOTS_DIR))
|
|
|
|
JOKES_PATH = os.path.join(CONTRIB_BOTS_DIR, 'assets/var/jokes.json')
|
|
DATABASE_PATH = os.path.join(CONTRIB_BOTS_DIR, 'assets/var/database.db')
|
|
DIRECTORY_PATH = os.path.join(CONTRIB_BOTS_DIR, 'assets')
|
|
VAR_PATH = os.path.join(CONTRIB_BOTS_DIR, 'assets/var')
|
|
|
|
if not os.path.exists(DIRECTORY_PATH):
|
|
os.makedirs(DIRECTORY_PATH)
|
|
|
|
if not os.path.exists(VAR_PATH):
|
|
os.makedirs(VAR_PATH)
|
|
|
|
# Create a new instance of a ChatBot
|
|
def create_chat_bot(no_learn):
|
|
return ChatBot("John",
|
|
storage_adapter="chatterbot.storage.JsonFileStorageAdapter",
|
|
logic_adapters=
|
|
[
|
|
"chatterbot.logic.MathematicalEvaluation",
|
|
{
|
|
"import_path": "chatterbot.logic.BestMatch",
|
|
"response_selection_method": "chatterbot.response_selection.get_random_response",
|
|
"statement_comparison_function": "chatterbot.comparisons.levenshtein_distance"
|
|
}],
|
|
output_adapter="chatterbot.output.OutputAdapter",
|
|
output_format='text',
|
|
database=DATABASE_PATH,
|
|
silence_performance_warning="True",
|
|
read_only=no_learn)
|
|
|
|
bot = create_chat_bot(False)
|
|
bot.set_trainer(ListTrainer)
|
|
|
|
bot.train([
|
|
"I want to contribute",
|
|
"""Contributors are more than welcomed! Please read
|
|
https://github.com/zulip/zulip#how-to-get-involved-with-contributing-to-zulip
|
|
to learn how to contribute.""",
|
|
])
|
|
|
|
bot.train([
|
|
"What is Zulip?",
|
|
"""Zulip is a powerful, open source group chat application. Written in Python
|
|
and using the Django framework, Zulip supports both private messaging and group
|
|
chats via conversation streams. You can learn more about the product and its
|
|
features at https://www.zulip.org.""",
|
|
])
|
|
|
|
bot.train([
|
|
"I would like to request a remote dev instance",
|
|
"""Greetings! You should receive a response from one of our mentors soon.
|
|
In the meantime, why don't you learn more about running Zulip on a development
|
|
environment? https://zulip.readthedocs.io/en/latest/using-dev-environment.html""",
|
|
])
|
|
|
|
bot.train([
|
|
"Joke!",
|
|
"Only if you ask nicely!",
|
|
])
|
|
|
|
bot.train([
|
|
"What is your name?",
|
|
"I am John, my job is to assist you with Zulip.",
|
|
])
|
|
|
|
bot.train([
|
|
"What can you do?",
|
|
"I can provide useful information and jokes if you follow etiquette.",
|
|
])
|
|
|
|
with open(JOKES_PATH) as data_file:
|
|
for joke in json.load(data_file):
|
|
bot.train([
|
|
"Please can you tell me a joke?",
|
|
joke['joke'],
|
|
])
|
|
|
|
bot.set_trainer(ChatterBotCorpusTrainer)
|
|
|
|
bot.train(
|
|
"chatterbot.corpus.english"
|
|
)
|
|
|
|
bota = create_chat_bot(True)
|
|
|
|
class JohnHandler(object):
|
|
'''
|
|
This bot aims to be Zulip's virtual assistant. It
|
|
finds the best match from a certain input.
|
|
Also understands the English language and can
|
|
mantain a conversation, joke and give useful information.
|
|
'''
|
|
|
|
def usage(self):
|
|
return '''
|
|
This bot aims to be Zulip's virtual assistant. It
|
|
finds the best match from a certain input.
|
|
Also understands the English language and can
|
|
mantain a conversation, joke and give useful information.
|
|
'''
|
|
|
|
def handle_message(self, message, client, state_handler):
|
|
original_content = message['content']
|
|
bot_response = str(bota.get_response(original_content))
|
|
client.send_reply(message, bot_response)
|
|
|
|
handler_class = JohnHandler
|