Update app.py

app.py

@@ -1,15 +1,30 @@
+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 an absurd and funny journey. {names}",
-        max_tokens=150
+    messages = [
+        {"role": "system", "content": "You are a helpful assistant."},
+        {"role": "user", "content": f"Title: {story_title}. In a fantasy world, {names} embarked on an absurd and funny journey. {names}"}
+    ]
+    
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages
     )
-    story = response.choices[0].text.strip()
-    print(story)
+    
+    story = response.choices[0].message['content']
+    return story
 
-# Call the function with some test values
-# Test API key
-test_generate_story("Alice, Bob", "The Enchanted Forest", "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")
+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()