Update app.py

app.py

@@ -1,7 +1,25 @@
 import gradio as gr
 import openai
+import requests
+from PIL import Image
+from io import BytesIO
 
-def generate_story(names, story_title, api_key):
+def create_dalle_prompt(story_title, names):
+    return f"Fantasy illustration of '{story_title}' featuring {names} in a magical adventure."
+
+def generate_image(api_key, prompt):
+    openai.api_key = api_key
+    response = openai.Image.create(
+        prompt=prompt,
+        n=1,
+        size="1024x1024"
+    )
+    image_url = response['data'][0]['url']
+    response = requests.get(image_url)
+    image = Image.open(BytesIO(response.content))
+    return image
+
+def generate_story_and_images(names, story_title, api_key):
     openai.api_key = api_key
     messages = [
         {"role": "system", "content": "You are a magical storyteller crafting a children's fantasy tale."},
@@ -14,17 +32,25 @@ def generate_story(names, story_title, api_key):
     )
     
     story = response.choices[0].message['content']
-    return story
-
+    dalle_prompt = create_dalle_prompt(story_title, names)
+    
+    images = [generate_image(api_key, f"{dalle_prompt} - Scene {i}") for i in range(1, 5)]
+    return story, *images
 
 iface = gr.Interface(
-    fn=generate_story,
+    fn=generate_story_and_images,
     inputs=[
         gr.components.Textbox(label="Enter names (comma-separated for multiple)"),
         gr.components.Textbox(label="Enter story title"),
         gr.components.Textbox(label="OpenAI API Key", type="password")
     ],
-    outputs="text",
+    outputs=[
+        "text",
+        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
 )