Update app.py

app.py

@@ -1,10 +1,7 @@
 import gradio as gr
 import openai
-import requests
-from PIL import Image
-from io import BytesIO
 
-def generate_story(api_key, names, story_title):
+def generate_story(names, story_title, api_key):
     openai.api_key = api_key
     response = openai.Completion.create(
         engine="gpt-3.5-turbo",
@@ -14,46 +11,14 @@ def generate_story(api_key, names, story_title):
     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(
-        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 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=main,
+    fn=generate_story,
     inputs=[
-        gr.components.Textbox(label="OpenAI API Key", type="password"),
         gr.components.Textbox(label="Enter names (comma-separated for multiple)"),
-        gr.components.Textbox(label="Enter story title")
-    ],
-    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")
+        gr.components.Textbox(label="Enter story title"),
+        gr.components.Textbox(label="OpenAI API Key", type="password")
     ],
+    outputs="text",
     live=True
 )