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

48
aibot.py
View file

@ -1,6 +1,6 @@
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
@ -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("?"):
@ -52,10 +59,39 @@ async def on_message(message):
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.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:
@ -85,8 +121,6 @@ async def aiprompt(ctx):
print(finalmsg)
await ctx.send(finalmsg)
@client.command()
async def aidel(ctx):
await ctx.message.delete()
@ -102,4 +136,4 @@ async def aidel(ctx):
await ctx.send(quote)
client.run('bot token')
client.run('bottoken')