Switched to Access Tokens

app.py

@@ -4,25 +4,27 @@ import os
 
 openai.api_key = os.environ.get("OPENAI_API_KEY")
 
-# Dalle API
+# Generate an image from text using DALL-E
 def open_ai_txt2img(prompt):
-    response = openai.Image.create(
-        prompt=prompt,
-        n=1,
-        size="1024x1024",
-    )
-    return response["data"][0]["url"]
+    try:
+        response = openai.Image.create(
+            prompt=prompt,
+            n=3,
+            size="1024x1024",
+        )
+        return [img["url"] for img in response["data"]]
+    except openai.error.OpenAIError as e:
+        print(e.http_status)
+        print(e.error)
 
 # Gradio Interface
 def generator(prompt):
-    return open_ai_txt2img(prompt), open_ai_txt2img(prompt), open_ai_txt2img(prompt)
+    return open_ai_txt2img(prompt)
 
 with gr.Blocks() as demo:
     prompt = gr.Textbox(label="Prompt")
     submit = gr.Button(label="Generate")
-    image1 = gr.Image()
-    image2 = gr.Image()
-    image3 = gr.Image()
+    image1, image2, image3 = gr.Image(), gr.Image(), gr.Image()
     submit.click(generator, inputs=[prompt], outputs=[image1, image2, image3], api_name="mmd")
 
 # demo = gr.Interface(fn=generator, inputs="text", outputs="image")