crate/garfbot
0
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

small refactor
All checks were successful
Garfbot CI/CD Deployment / Deploy (push) Successful in 5s
Details

Browse Source
This commit is contained in:
crate
2025-06-07 21:11:10 -05:00
parent 13d2f09fe3
commit 8c08eed842
3 changed files with 70 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
garfpy/garfai.py
View File

@ -16,17 +16,15 @@ class GarfAI:
self.imgmodel = config.IMG_MODEL
self.image_request_queue = asyncio.Queue()
async def garfpic(self, message, prompt):
await self.image_request_queue.put({"message": message, "prompt": prompt})
async def garfpic(self, ctx, prompt):
await self.image_request_queue.put({"ctx": ctx, "prompt": prompt})
async def generate_image(self, prompt):
client = AsyncOpenAI(api_key=self.openaikey)
try:
client = AsyncOpenAI(api_key=self.openaikey)
response = await client.images.generate(
model=self.imgmodel, prompt=prompt, n=1, size="1024x1024"
)
image_url = response.data[0].url
return image_url
except openai.BadRequestError as e:
return f"`GarfBot Error: ({e.status_code}) - Your request was rejected as a result of our safety system.`"
except openai.InternalServerError as e:
@ -35,32 +33,48 @@ class GarfAI:
except Exception as e:
logger.error(e)
return "`GarfBot Error: Lasagna`"
data = getattr(response, "data", None)
if not data:
logger.error("No data in response")
return "`GarfBot Error: No images generated`"
first_image = data[0] if len(data) > 0 else None
if not first_image:
logger.error("No image in response data")
return "`GarfBot Error: No images generated`"
image_url = getattr(first_image, "url", None)
if not image_url:
logger.error("No URL in image response")
return "`GarfBot Error: No image URL returned`"
return image_url
async def process_image_requests(self):
async with aiohttp.ClientSession() as session:
while True:
request = await self.image_request_queue.get()
message = request["message"]
ctx = request["ctx"]
prompt = request["prompt"]
image_url = await self.generate_image(prompt)
if "GarfBot Error" not in image_url:
if image_url and "GarfBot Error" not in image_url:
logger.info("Downloading & sending image...")
async with session.get(image_url) as resp:
if resp.status == 200:
image_data = await resp.read()
image = io.BytesIO(image_data)
image.seek(0)
timestamp = message.created_at.strftime("%Y%m%d%H%M%S")
timestamp = ctx.message.created_at.strftime("%Y%m%d%H%M%S")
filename = f"{timestamp}_generated_image.png"
sendfile = discord.File(fp=image, filename=filename)
try:
await message.channel.send(file=sendfile)
await ctx.send(file=sendfile)
except Exception as e:
logger.error(e)
else:
await message.channel.send("`GarfBot Error: Odie`")
await ctx.send("`GarfBot Error: Odie`")
else:
await message.channel.send(image_url)
await ctx.send(image_url)
self.image_request_queue.task_done()
await asyncio.sleep(2)
@ -78,15 +92,16 @@ class GarfAI:
],
max_tokens=400,
)
answer = response.choices[0].message.content
answer = str(response.choices[0].message.content)
return answer.replace("an AI language model", "a cartoon animal")
except openai.BadRequestError as e:
logger.error(e)
return f"`GarfBot Error: {e}`"
except openai.APIError as e:
logger.info(e, flush=True)
logger.error(e)
return "`GarfBot Error: Monday`"
except Exception as e:
logger.info(e, flush=True)
logger.error(e)
return "`GarfBot Error: Lasagna`"
async def wikisum(self, query):