Update app.py

app.py

@@ -1,25 +1,14 @@
-import gradio as gr
 import openai
 
-def generate_story(names, story_title, api_key):
+def test_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 an absurd and funny journey. {names}",
+        prompt=f"Title: {story_title}\n\nIn a fantasy world, {names} embarked on a journey.",
         max_tokens=150
     )
     story = response.choices[0].text.strip()
-    return story
+    print(story)
 
-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()
+# Call the function with some test values
+test_generate_story("Alice, Bob", "The Enchanted Forest", "YOUR_OPENAI_API_KEY_HERE")