format etc

garfpy/garfai.py

@@ -17,16 +17,13 @@ class GarfAI:
         self.image_request_queue = asyncio.Queue()
 
     async def garfpic(self, message, prompt):
-        await self.image_request_queue.put({'message': message, 'prompt': prompt})
+        await self.image_request_queue.put({"message": message, "prompt": prompt})
 
     async def generate_image(self, prompt):
         try:
-            client = AsyncOpenAI(api_key = self.openaikey)
+            client = AsyncOpenAI(api_key=self.openaikey)
             response = await client.images.generate(
-                model=self.imgmodel,
-                prompt=prompt,
-                n=1,
-                size="1024x1024"
+                model=self.imgmodel, prompt=prompt, n=1, size="1024x1024"
             )
             image_url = response.data[0].url
             return image_url
@@ -37,14 +34,14 @@ class GarfAI:
             return f"`GarfBot Error: ({e.status_code}) - Monday`"
         except Exception as e:
             logger.error(e)
-            return f"`GarfBot Error: Lasagna`"
+            return "`GarfBot Error: Lasagna`"
 
     async def process_image_requests(self):
         async with aiohttp.ClientSession() as session:
             while True:
                 request = await self.image_request_queue.get()
-                message = request['message']
-                prompt = request['prompt']
+                message = request["message"]
+                prompt = request["prompt"]
                 image_url = await self.generate_image(prompt)
                 if "GarfBot Error" not in image_url:
                     logger.info("Downloading & sending image...")
@@ -53,7 +50,7 @@ class GarfAI:
                             image_data = await resp.read()
                             ram_image = io.BytesIO(image_data)
                             ram_image.seek(0)
-                            timestamp = message.created_at.strftime('%Y%m%d%H%M%S')
+                            timestamp = message.created_at.strftime("%Y%m%d%H%M%S")
                             filename = f"{timestamp}_generated_image.png"
                             sendfile = discord.File(fp=ram_image, filename=filename)
                             try:
@@ -69,14 +66,17 @@ class GarfAI:
 
     async def generate_chat(self, question):
         try:
-            client = AsyncOpenAI(api_key = self.openaikey)
+            client = AsyncOpenAI(api_key=self.openaikey)
             response = await client.chat.completions.create(
                 model=self.txtmodel,
                 messages=[
-                    {"role": "system", "content": "Pretend you are sarcastic Garfield."},
-                    {"role": "user", "content": f"{question}"}
+                    {
+                        "role": "system",
+                        "content": "Pretend you are sarcastic Garfield.",
+                    },
+                    {"role": "user", "content": f"{question}"},
                 ],
-                max_tokens=400
+                max_tokens=400,
             )
             answer = response.choices[0].message.content
             return answer.replace("an AI language model", "a cartoon animal")
@@ -84,15 +84,17 @@ class GarfAI:
             return f"`GarfBot Error: {e}`"
         except openai.APIError as e:
             logger.info(e, flush=True)
-            return f"`GarfBot Error: Monday`"
+            return "`GarfBot Error: Monday`"
         except Exception as e:
             logger.info(e, flush=True)
-            return f"`GarfBot Error: Lasagna`"
+            return "`GarfBot Error: Lasagna`"
 
     async def wikisum(self, query):
         try:
             summary = wikipedia.summary(query)
-            garfsum = await self.generate_chat(f"Please summarize in your own words: {summary}")
+            garfsum = await self.generate_chat(
+                f"Please summarize in your own words: {summary}"
+            )
             return garfsum
         except Exception as e:
-            return e
+            return e