Update app.py

app.py

@@ -1,22 +1,50 @@
 import gradio as gr
+import openai
 
-def create_story(noun1, adjective1, verb1, noun2, adjective2, verb2):
-    story = (f"One day, a {adjective1} {noun1} {verb1} through the forest. "
-             f"It came across a {adjective2} {noun2} and they quickly became friends. "
-             f"Later, they both {verb2} happily ever after.")
-    return story
+def generate_image(api_key, prompt):
+    # Initialize OpenAI with the provided API key
+    openai.api_key = api_key
+
+    # Generate the image
+    response = openai.Image.create(
+        prompt=prompt,
+        n=1,
+        size="1024x1024"
+    )
+    image_url = response['data'][0]['url']
+    return image_url
+
+def generate_story(api_key, names, story_title):
+    # Initialize OpenAI with the provided API key
+    openai.api_key = api_key
+
+    # Generate the story
+    response = openai.Completion.create(
+        model="text-davinci-003",
+        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()
+
+    # Generate 4 images inspired by the story title
+    images = [generate_image(api_key, f"Fantasy image inspired by {story_title} - Scene {i}") for i in range(1, 5)]
+
+    return story, *images
 
 iface = gr.Interface(
-    fn=create_story,
+    fn=generate_story,
     inputs=[
-        gr.inputs.Textbox(label="Enter a noun"),
-        gr.inputs.Textbox(label="Enter an adjective"),
-        gr.inputs.Textbox(label="Enter a verb (past tense)"),
-        gr.inputs.Textbox(label="Enter another noun"),
-        gr.inputs.Textbox(label="Enter another adjective"),
-        gr.inputs.Textbox(label="Enter another verb (past tense)")
+        gr.inputs.Textbox(label="OpenAI API Key", type="password"),
+        gr.inputs.Textbox(label="Enter names (comma-separated for multiple)"),
+        gr.inputs.Textbox(label="Enter story title")
+    ],
+    outputs=[
+        "text",
+        gr.outputs.Image(label="Image 1"),
+        gr.outputs.Image(label="Image 2"),
+        gr.outputs.Image(label="Image 3"),
+        gr.outputs.Image(label="Image 4")
     ],
-    outputs="text",
     live=True
 )