# pip install chatterbot
from chatterbot import ChatBot
from chatterbot.trainers import ChatterBotCorpusTrainer, ListTrainer
import os
bot = ChatBot("Candice")
trainer = ChatterBotCorpusTrainer(bot)
# train based on english
trainer.train("chatterbot.corpus.english")
# train based on chinese
trainer.train("chatterbot.corpus.chinese")
# use list data train
trainer2 = ListTrainer(bot)
trainer2.train(["What is your name?", "My name is Candice"])
trainer2.train(["Who are you?", "I am a bot"])
## conf
chatbot = ChatBot("Math & Time Bot",
logic_adapters=['chatterbot.logic.MathematicalEvaluation', 'chatterbot.logic.TimeLogicAdapter'],
input_adapter="chatterbot.input.VariableInputTypeAdapter",
output_adapter="chatterbot.output.OutputAdapter")
while True:
message = input("\t\t\tYou:")
if message.strip() != "Bye":
reply = bot.get_response(message)
print("Candice: ", reply)
if message.strip() == "Bye":
print("Candice: Bye")
break