Update app.py

app.py

@@ -1,14 +1,25 @@
+import gradio as gr
 import openai
 
-def test_generate_story(names, story_title, api_key):
+def generate_story(names, story_title, api_key):
     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 a journey.",
+        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()
-    print(story)
+    return story
 
-# Call the function with some test values
-test_generate_story("Alice, Bob", "The Enchanted Forest", "YOUR_OPENAI_API_KEY_HERE")
+iface = gr.Interface(
+    fn=generate_story,
+    inputs=[
+        gr.components.Textbox(label="Enter names (comma-separated for multiple)"),
+        gr.components.Textbox(label="Enter story title"),
+        gr.components.Textbox(label="OpenAI API Key", type="password")
+    ],
+    outputs="text",
+    live=True
+)
+
+iface.launch()