Update app.py

app.py

@@ -12,27 +12,33 @@ def generate_image(api_key, prompt):
         size="1024x1024"
     )
     image_url = response['data'][0]['url']
-    
+
     # Download the image and open it with PIL
     response = requests.get(image_url)
     image = Image.open(BytesIO(response.content))
-    
+
     return image
 
+def create_dalle_prompt(story, story_title):
+    return f"Visual representation of the fantasy story titled '{story_title}': {story}"
 
-def generate_story(api_key, names, story_title):
+def generate_story_and_images(api_key, names, story_title):
     openai.api_key = api_key
     response = openai.Completion.create(
-        model="text-davinci-003",
+        model="gpt-3.5-turbo",
         prompt=f"Title: {story_title}\n\nIn a fantasy world, {names} embarked on an absurd and funny journey. {names}",
         max_tokens=150
     )
     story = response.choices[0].text.strip()
-    images = [generate_image(api_key, f"Fantasy image inspired by {story_title} - Scene {i}") for i in range(1, 5)]
+
+    # Create a DALLE prompt using the story
+    dalle_prompt = create_dalle_prompt(story, story_title)
+    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="OpenAI API Key", type="password"),
         gr.components.Textbox(label="Enter names (comma-separated for multiple)"),
@@ -48,4 +54,4 @@ iface = gr.Interface(
     live=True
 )
 
-iface.launch()
+iface.launch()