Update app.py

app.py

@@ -3,9 +3,10 @@ import openai
 import requests
 from PIL import Image
 from io import BytesIO
+import textwrap
 
-def create_dalle_prompt(story_title, names):
-    return f"Fantasy illustration of '{story_title}' featuring {names} in a magical adventure."
+def create_dalle_prompts(story_title, paragraph):
+    return f"Illustration of '{story_title}' - {textwrap.shorten(paragraph, width=50)}"
 
 def generate_image(api_key, prompt):
     openai.api_key = api_key
@@ -19,11 +20,11 @@ def generate_image(api_key, prompt):
     image = Image.open(BytesIO(response.content))
     return image
 
-def generate_story_and_images(names, story_title, api_key):
+def generate_story_and_images(name, story_title, api_key):
     openai.api_key = api_key
     messages = [
         {"role": "system", "content": "You are a magical storyteller crafting a children's fantasy tale."},
-        {"role": "user", "content": f"In the mystical land of Fantasia, two young adventurers named {names} set out on a quest to find the lost city of Eloria. Along the way, they encounter magical creatures, enchanted forests, and mysterious puzzles. Tell us their whimsical and heartwarming tale based on the title: '{story_title}'."}
+        {"role": "user", "content": f"In the mystical land of Fantasia, a young adventurer named {name} sets out on a quest filled with magic and wonder. Tell us their whimsical and heartwarming tale based on the title: '{story_title}'."}
     ]
     
     response = openai.ChatCompletion.create(
@@ -32,15 +33,20 @@ def generate_story_and_images(names, story_title, api_key):
     )
     
     story = response.choices[0].message['content']
-    dalle_prompt = create_dalle_prompt(story_title, names)
     
-    images = [generate_image(api_key, f"{dalle_prompt} - Scene {i}") for i in range(1, 5)]
+    # Split story into four paragraphs
+    paragraphs = story.split('\n')[:4]
+    
+    # Create DALLE prompts for each paragraph
+    dalle_prompts = [create_dalle_prompts(story_title, para) for para in paragraphs]
+    
+    images = [generate_image(api_key, prompt) for prompt in dalle_prompts]
     return story, *images
 
 iface = gr.Interface(
     fn=generate_story_and_images,
     inputs=[
-        gr.components.Textbox(label="Enter names (comma-separated for multiple)"),
+        gr.components.Textbox(label="Enter name"),
         gr.components.Textbox(label="Enter story title"),
         gr.components.Textbox(label="OpenAI API Key", type="password")
     ],