insert/ai-bot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Update aibot.py

Browse source 
updates that were wayyyy behinf
This commit is contained in:
insert username here 2022-01-22 16:48:27 -05:00 committed by GitHub
parent e4ddb3b32c
commit 2aa3775529
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 82 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

83
aibot.py
View file

@ -2,10 +2,59 @@ from textgenrnn import textgenrnn
import discord
from discord.ext import commands
import regex as re
import functools
from aitextgen import aitextgen
import subprocess
from chatterbot import ChatBot
textgen = textgenrnn('insert3_weights.hdf5')
ai = aitextgen()
bot = ChatBot(
'insert3',
storage_adapter='chatterbot.storage.SQLStorageAdapter',
logic_adapters=[
'chatterbot.logic.BestMatch'
],
database_uri='sqlite:///database.sqlite3'
)
client = commands.Bot(command_prefix='.')
boton = True
def add_to_train(clean_content):
with open("train.txt", "a") as train:
train.write(f"{clean_content} \n")
@client.event
async def on_message(message):
await client.process_commands(message)
urls = re.findall('http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@.&+]|[!*(),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+',message.content.lower())
if not message.attachments:
if boton:
if message.content.lower().startswith("$"):
pass
elif urls:
pass
elif message.author.bot:
pass
elif message.content.lower().startswith("p!"):
pass
elif message.content.lower().startswith("."):
pass
elif message.content.lower().startswith("?"):
pass
elif message.content.lower().startswith("!"):
pass
elif message.content.lower().startswith("—"):
pass
elif message.content.lower().startswith("--"):
pass
else:
writing_function = functools.partial(add_to_train, message.clean_content)
await client.loop.run_in_executor(None, writing_function)
if message.channel.id == 934533774249713664:
if message.author.bot != True:
bot_response = bot.get_response(message.clean_content)
await message.channel.send(bot_response)
@client.command()
async def aiquote(ctx):
@ -20,5 +69,37 @@ async def aiquote(ctx):
quote = f.read()
await ctx.send(quote)
@client.command()
async def aiprompt(ctx):
prompt = ctx.message.content[10:]
endmsg = subprocess.check_output(['python3','promptai.py',f'{prompt}'])
non = endmsg.replace("%1B[1m".encode(), "**".encode())
nona = non.replace("%1B[0m".encode(), "**".encode())
nor = nona.replace("".encode(), "".encode())
nora = nor.replace("".encode(), "".encode())
norx = nora.replace("\x1b[1m".encode(), "".encode())
nore = norx.replace("\\x1b[1m".encode(), "".encode())
nor1 = nore.replace("\x1b[0m".encode(), "".encode())
nor2 = nor1.replace("\\x1b[0m".encode(), "".encode())
finalmsg = nor2.decode("utf-8")
print(finalmsg)
await ctx.send(finalmsg)
client.run('BOTTOKEN')
@client.command()
async def aidel(ctx):
await ctx.message.delete()
try:
textgen.generate_to_file('insert3.txt', n=1)
with open("insert3.txt") as f:
quote = f.read()
await ctx.send(quote)
except:
textgen.generate_to_file('insert3.txt', n=1)
with open("insert3.txt") as f:
quote = f.read()
await ctx.send(quote)
client.run('bot token')