phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

cleaned up folder setup

Browse source 
added indicators that bot saw a question and is working on it
This commit is contained in:
phixxy 2024-01-30 02:33:47 -08:00
parent ee49ea9520
commit fcb02562f8
1 changed files with 18 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

33
extensions/chatgpt.py
View file

@ -20,22 +20,23 @@ class ChatGPT(commands.Cog):
self.remind_me_loop.start() self.remind_me_loop.start()
def folder_setup(self): async def folder_setup(self):
try: try:
if not os.path.exists(self.working_dir): folders = [
os.mkdir(self.working_dir) self.working_dir,
if not os.path.exists(self.data_dir): self.data_dir,
os.mkdir(self.data_dir) self.data_dir + "config",
if not os.path.exists(self.data_dir + "config"): self.data_dir + "logs",
os.mkdir(self.data_dir + "config") self.data_dir + "dalle"
if not os.path.exists(self.data_dir + "logs"): ]
os.mkdir(self.data_dir + "logs")
if not os.path.exists(self.data_dir + "dalle"): for folder in folders:
os.mkdir(self.data_dir + "dalle") if not os.path.exists(folder):
if not os.path.exists(self.data_dir + "dalle2"): os.mkdir(folder)
os.mkdir(self.data_dir + "dalle2")
except: except Exception as e:
self.bot.logger.exception("ChatGPT failed to make directories") self.bot.logger.exception(f"ChatGPT failed to make directories: {e}")
def create_channel_config(self, filepath): def create_channel_config(self, filepath):
config_dict = { config_dict = {
@ -159,6 +160,7 @@ class ChatGPT(commands.Cog):
brief="Get an answer" brief="Get an answer"
) )
async def question(self, ctx): async def question(self, ctx):
await ctx.send("One moment, let me think...")
question = ctx.message.content.split(" ", maxsplit=1)[1] question = ctx.message.content.split(" ", maxsplit=1)[1]
answer = await self.answer_question(question) answer = await self.answer_question(question)
chunks = [answer[i:i+1999] for i in range(0, len(answer), 1999)] chunks = [answer[i:i+1999] for i in range(0, len(answer), 1999)]
@ -172,6 +174,7 @@ class ChatGPT(commands.Cog):
) )
async def question_gpt4(self, ctx): async def question_gpt4(self, ctx):
if ctx.author.get_role(self.premium_role): if ctx.author.get_role(self.premium_role):
await ctx.send("One moment, let me think...")
question = ctx.message.content.split(" ", maxsplit=1)[1] question = ctx.message.content.split(" ", maxsplit=1)[1]
answer = await self.answer_question(question, "gpt-4") answer = await self.answer_question(question, "gpt-4")
chunks = [answer[i:i+1999] for i in range(0, len(answer), 1999)] chunks = [answer[i:i+1999] for i in range(0, len(answer), 1999)]