Initial

app.py

@@ -1,27 +1,25 @@
 import gradio as gr
-import openai
+from openai import OpenAI
 import os
 
-openai.api_key = os.environ['OPENAI_API_KEY']
-
+client = OpenAI(api_key=os.environ['OPENAI_API_KEY'])
 
 def generate_story(topic):
     prompt = f"Write a short, creative, and fun children's story about the following topic:\n\nTopic: {topic}\n\nStory:"
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model="gpt-3.5-turbo",
         messages=[{"role": "user", "content": prompt}],
-        max_tokens=2048,
-        temperature=0.1
+        max_tokens=500,
+        temperature=0.8
     )
     story = response.choices[0].message.content.strip()
     return story
 
-
 iface = gr.Interface(
     fn=generate_story,
     inputs=gr.Textbox(label="Story Topic", placeholder="E.g., Adventures of a brave rabbit..."),
     outputs=gr.Textbox(label="Generated Story"),
-    title="✨ UmaiMother ✨",
+    title="✨ Storyteller ✨",
     description="Enter a topic, and a fun story will be automatically generated for you!"
 )