Update app.py

app.py

@@ -1,25 +1,34 @@
 import os
 import gradio as gr
 import openai
+import requests
 from PIL import Image
 import io
 import numpy as np
 
 openai.api_key = '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' # your api key
-openai.Model.list()
 
 def DALLE(user_input):
     prompt = user_input
     response = prompt
     
-    DALLE_reply = openai.Image.create(
-        prompt=prompt,
-        n=2,
-        size="1024x1024"
-    )
+    # Create the DALL-E image using the OpenAI API
+    data = {
+        "model": "image-alpha-001",
+        "prompt": f"{prompt}:::",
+        "n": 1,
+        "size": "1024x1024",
+        "response_format": "url"
+    }
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {openai.api_key}"
+    }
+    response = requests.post("https://api.openai.com/v1/images/generations", json=data, headers=headers)
+    response.raise_for_status()
     
-    # Convert the OpenAI output to an image format that Gradio can recognize
-    img_bytes = io.BytesIO(DALLE_reply.get()['data'])
+    # Convert the image to a format that Gradio can display
+    img_bytes = io.BytesIO(requests.get(response.json()["data"][0]["url"]).content)
     img = Image.open(img_bytes)
     img_arr = np.array(img)
     
@@ -27,3 +36,4 @@ def DALLE(user_input):
 
 iface = gr.Interface(fn=DALLE, inputs="text", outputs="image", title="bhAI")
 iface.launch()
+