Update app.py

app.py

@@ -2,10 +2,7 @@ import gradio as gr
 import openai
 
 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,
@@ -15,35 +12,29 @@ def generate_image(api_key, prompt):
     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=generate_story,
     inputs=[
-        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")
+        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.outputs.Image(label="Image 1"),
-        gr.outputs.Image(label="Image 2"),
-        gr.outputs.Image(label="Image 3"),
-        gr.outputs.Image(label="Image 4")
+        gr.components.Image(label="Image 1", type="url"),
+        gr.components.Image(label="Image 2", type="url"),
+        gr.components.Image(label="Image 3", type="url"),
+        gr.components.Image(label="Image 4", type="url")
     ],
     live=True
 )