Blackfur/SheepBotFrame
1
1
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Fixed Cleanup

Browse source
This commit is contained in:
BuildTools 2021-01-13 19:54:25 +01:00
parent 04d35d77c1
commit 7e3a644d3f
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
run.py
View file

@ -10,13 +10,12 @@ config = {"botname": "Sheepy",
"filterpings": True, "filterpings": True,
"usechannels": [717758904527880253, 772839872557088769, 773980809471197235], "usechannels": [717758904527880253, 772839872557088769, 773980809471197235],
"trainchannels": [[492412566430154783, 1000], [761462125696385047, 1000]], "trainchannels": [[492412566430154783, 1000], [761462125696385047, 1000]],
"corpus": ["chatterbot.corpus.english", "chatterbot.corpus.uswest"]} "corpus": ["chatterbot.corpus.english"]}
with open("token", "r") as tokenfile: with open("token", "r") as tokenfile:
token = tokenfile.read() token = tokenfile.read()
bot = discord.Client() bot = discord.Client()
chatbot = ChatBot(config["botname"])
tasks = [False, False] tasks = [False, False]
@ -45,7 +44,6 @@ async def download():
async def train(): async def train():
os.system("rm db.sqlite3")
corpustrainer = ChatterBotCorpusTrainer(chatbot) corpustrainer = ChatterBotCorpusTrainer(chatbot)
listtrainer = ListTrainer(chatbot) listtrainer = ListTrainer(chatbot)
for corpus in config["corpus"]: for corpus in config["corpus"]:
@ -82,7 +80,11 @@ async def on_ready():
if tasks[0]: if tasks[0]:
await download() await download()
if tasks[1]: if tasks[1]:
os.system("rm db.sqlite3")
global chatbot
chatbot = ChatBot(config["botname"])
await train() await train()
else:
chatbot = ChatBot(config["botname"])
bot.run(token) bot.run(token)