From 995a0090f9bdab074f6f13de5845f89a517d0487 Mon Sep 17 00:00:00 2001
From: unknown <jacobdamkjaer@gmail.com>
Date: Wed, 15 Mar 2023 23:52:42 +0100
Subject: [PATCH] attempting to use endpoint

---
 reginaldCog/reginald.py | 27 ++++++++++++++++-----------
 1 file changed, 16 insertions(+), 11 deletions(-)

diff --git a/reginaldCog/reginald.py b/reginaldCog/reginald.py
index 619b2f8..f10ac8b 100644
--- a/reginaldCog/reginald.py
+++ b/reginaldCog/reginald.py
@@ -137,18 +137,23 @@ class ReginaldCog(commands.Cog):
             await ctx.author.send(f"I apologize, but I am unable to generate an image at this time. Error message: {str(e)}")
 
     async def generate_image(self, api_key, prompt):
-        openai.api_key = api_key
-        model = "image-alpha-001"  # Replace this with the appropriate DALL-E model name
-        response = openai.Image.create(
-            model=model,
-            prompt=prompt,
-            n=1,
-            size="512x512",
-            response_format="url"
-        )
+        url = "https://api.openai.com/v1/images/generations"
+        headers = {
+            "Content-Type": "application/json",
+            "Authorization": f"Bearer {api_key}",
+        }
+        data = {
+            "prompt": prompt,
+            "n": 1,
+            "size": "1024x1024",
+        }
 
-        if response and response.choices and response.choices[0].url:
-            return response.choices[0].url
+        async with aiohttp.ClientSession() as session:
+            async with session.post(url, headers=headers, data=json.dumps(data)) as resp:
+                response = await resp.json()
+
+        if response and "data" in response and len(response["data"]) > 0:
+            return response["data"][0]["url"]
         return None
 
     @reginald.error