Kanium-PUBLIC/KaniumCogs
5
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

Told chatgpt to fix itself because I am tired

Browse Source
This commit is contained in:
unknown 2023-03-16 17:33:08 +01:00
parent 3d12dae65d
commit 4a15cef8f7
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
reginaldCog/reginald.py
View File

@ -1,6 +1,6 @@
import discord import discord
import json import json
import openai_async as openai import openai
import os import os
import random import random
import requests import requests
@ -9,6 +9,7 @@ import aiohttp
from io import BytesIO from io import BytesIO
from PIL import Image from PIL import Image
import tempfile import tempfile
from openai import OpenAIError
from redbot.core import Config, commands from redbot.core import Config, commands
@ -23,19 +24,19 @@ class ReginaldCog(commands.Cog):
openai_api_key=None openai_api_key=None
) )
def has_kanium_role(): def has_kanium_role(self):
async def predicate(ctx): async def predicate(ctx):
kanium_role_id = 280260875678515200 kanium_role_id = 280260875678515200
return any(role.id == kanium_role_id for role in ctx.author.roles) return any(role.id == kanium_role_id for role in ctx.author.roles)
return commands.check(predicate) return commands.check(predicate)
def has_admin_role(self):
def has_janitor_or_admin_role():
async def predicate(ctx): async def predicate(ctx):
janitor_role_id = 672156832323600396 #janitor_role_id = 672156832323600396
has_janitor_role = any(role.id == janitor_role_id for role in ctx.author.roles) #has_janitor_role = any(role.id == janitor_role_id for role in ctx.author.roles) # Uncomment this line
has_admin_permission = ctx.author.guild_permissions.administrator has_admin_permission = ctx.author.guild_permissions.administrator
return has_janitor_role or has_admin_permission #return has_janitor_role or has_admin_permission
return has_admin_permission
return commands.check(predicate) return commands.check(predicate)
@ -74,7 +75,7 @@ class ReginaldCog(commands.Cog):
response_text = await self.generate_response(api_key, prompt) response_text = await self.generate_response(api_key, prompt)
for chunk in self.split_response(response_text, 2000): for chunk in self.split_response(response_text, 2000):
await ctx.send(chunk) await ctx.send(chunk)
except openai.error.OpenAIError as e: except OpenAIError as e:
await ctx.send(f"I apologize, but I am unable to generate a response at this time. Error message: {str(e)}") await ctx.send(f"I apologize, but I am unable to generate a response at this time. Error message: {str(e)}")
except commands.CommandOnCooldown as e: except commands.CommandOnCooldown as e:
remaining_seconds = int(e.retry_after) remaining_seconds = int(e.retry_after)
@ -111,7 +112,7 @@ class ReginaldCog(commands.Cog):
return chunks return chunks
@commands.guild_only() @commands.guild_only()
@has_janitor_or_admin_role() @has_admin_role()
@commands.command(help="Ask Reginald to generate an image based on a prompt") @commands.command(help="Ask Reginald to generate an image based on a prompt")
@commands.cooldown(1, 300, commands.BucketType.user) # 5-minute cooldown per user @commands.cooldown(1, 300, commands.BucketType.user) # 5-minute cooldown per user
async def reginaldimagine(self, ctx, *, prompt=None): async def reginaldimagine(self, ctx, *, prompt=None):
@ -154,7 +155,7 @@ class ReginaldCog(commands.Cog):
data = { data = {
"prompt": prompt, "prompt": prompt,
"n": 1, "n": 1,
"size": "1024x1024", "size": "256x256",
} }
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session: