Update app.py

app.py

@@ -4,6 +4,24 @@ import requests
 from PIL import Image
 from io import BytesIO
 
+def generate_story(api_key, names, story_title):
+    openai.api_key = api_key
+    response = openai.Completion.create(
+        engine="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()
+    return story
+
+def create_dalle_prompt(story, story_title):
+    return f"Visual representation of the fantasy story titled '{story_title}': {story}"
+
+def generate_dalle_images(api_key, prompt):
+    openai.api_key = api_key
+    images = [generate_image(api_key, f"{prompt} - Scene {i}") for i in range(1, 5)]
+    return images
+
 def generate_image(api_key, prompt):
     openai.api_key = api_key
     response = openai.Image.create(
@@ -12,33 +30,18 @@ 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_and_images(api_key, names, story_title):
-    openai.api_key = api_key
-    response = openai.Completion.create(
-        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()
-
-    # 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)]
-
+def main(api_key, names, story_title):
+    story = generate_story(api_key, names, story_title)
+    prompt = create_dalle_prompt(story, story_title)
+    images = generate_dalle_images(api_key, prompt)
     return story, *images
 
 iface = gr.Interface(
-    fn=generate_story_and_images,
+    fn=main,
     inputs=[
         gr.components.Textbox(label="OpenAI API Key", type="password"),
         gr.components.Textbox(label="Enter names (comma-separated for multiple)"),