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

lets start slashing!

Browse source
This commit is contained in:
insert username here 2022-04-23 20:25:26 -04:00 committed by GitHub
parent aed66203bb
commit df5d850e6d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 46 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
aibot.py
View file

@ -1,10 +1,10 @@
from textgenrnn import textgenrnn
import discord
from discord.ext import commands
import disnake
from disnake.ext import commands
import regex as re
import functools
from aitextgen import aitextgen
import subprocess
import subprocess
from chatterbot import ChatBot
textgen = textgenrnn('insert3_weights.hdf5')
@ -17,9 +17,10 @@ bot = ChatBot(
],
database_uri='sqlite:///database.sqlite3'
)
client = commands.Bot(command_prefix='.')
client = commands.Bot(command_prefix='.', test_guilds=[732793772697583623])
boton = True
def add_to_train(clean_content):
with open("train.txt", "a") as train:
train.write(f"{clean_content} \n")
@ -38,6 +39,12 @@ async def on_message(message):
pass
elif message.content.lower().startswith("p!"):
pass
elif message.content.lower().startswith("_"):
pass
elif message.channel.id == 817397092376117248:
pass
elif message.channel.id == 779060858449821716:
pass
elif message.content.lower().startswith("."):
pass
elif message.content.lower().startswith("?"):
@ -51,11 +58,40 @@ async def on_message(message):
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)
if message.channel.id == 934533774249713664:
bot_response = bot.get_response(message.clean_content)
await message.channel.send(bot_response)
@client.slash_command(description="get a quote from the ai")
async def aiquote(inter):
await inter.response.defer()
try:
textgen.generate_to_file('insert3.txt', n=1)
with open("insert3.txt") as f:
quote = f.read()
await inter.response.send_message(content=f"{quote}")
except:
textgen.generate_to_file('insert3.txt', n=1)
with open("insert3.txt") as f:
quote = f.read()
await inter.edit_original_message(content=f"{quote}")
@client.slash_command(description="give the ai a prompt to finish off")
async def aiprompt(inter, prompt: str):
await inter.response.defer()
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 inter.edit_original_message(content=f"{finalmsg}")
@client.command()
async def aiquote(ctx):
try:
@ -67,7 +103,7 @@ async def aiquote(ctx):
textgen.generate_to_file('insert3.txt', n=1)
with open("insert3.txt") as f:
quote = f.read()
await ctx.send(quote)
await ctx.send(quote)
@client.command()
async def aiprompt(ctx):
@ -84,8 +120,6 @@ async def aiprompt(ctx):
finalmsg = nor2.decode("utf-8")
print(finalmsg)
await ctx.send(finalmsg)
@client.command()
async def aidel(ctx):
@ -102,4 +136,4 @@ async def aidel(ctx):
await ctx.send(quote)
client.run('bot token')
client.run('bottoken')