Update app.py

app.py

@@ -1,5 +1,8 @@
 import gradio as gr
 import openai
+import requests
+from PIL import Image
+from io import BytesIO
 
 def generate_image(api_key, prompt):
     openai.api_key = api_key
@@ -9,7 +12,13 @@ def generate_image(api_key, prompt):
         size="1024x1024"
     )
     image_url = response['data'][0]['url']
-    return image_url
+    
+    # Download the image and open it with PIL
+    response = requests.get(image_url)
+    image = Image.open(BytesIO(response.content))
+    
+    return image
+
 
 def generate_story(api_key, names, story_title):
     openai.api_key = api_key
@@ -31,12 +40,12 @@ iface = gr.Interface(
     ],
     outputs=[
         "text",
-        gr.components.Image(label="Image 1", type="url"),
-        gr.components.Image(label="Image 2", type="url"),
-        gr.components.Image(label="Image 3", type="url"),
-        gr.components.Image(label="Image 4", type="url")
+        gr.components.Image(label="Image 1", type="pil"),
+        gr.components.Image(label="Image 2", type="pil"),
+        gr.components.Image(label="Image 3", type="pil"),
+        gr.components.Image(label="Image 4", type="pil")
     ],
     live=True
 )
 
-iface.launch()
+iface.launch()