Update app.py

app.py

@@ -1,9 +1,12 @@
 import gradio as gr
-import openai
 import os
+from openai import OpenAI
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Initialize OpenAI client using API key from environment
+api_key = os.getenv("OPENAI_API_KEY")
+client = OpenAI(api_key=api_key)
 
+# Initial game scenario and choices
 initial_scenario = "You wake up in a strange forest. Paths lead in all directions.\n"
 choices = [
     "Walk north into the dense forest.",
@@ -12,34 +15,41 @@ choices = [
     "Move west where the land rises."
 ]
 
+# Conversation history starts with a system prompt and the opening scene
 history = [
-    {"role": "system", "content": "You're an interactive text-based RPG engine. Continue the story based on the user's choices. Keep responses concise and vivid."},
+    {
+        "role": "system",
+        "content": (
+            "You are a text-based RPG engine. Continue the story based on the player's choices. "
+            "Respond in immersive but concise prose. End each turn ready for the next input."
+        )
+    },
     {"role": "user", "content": initial_scenario}
 ]
 
+# Function to generate story continuation from the OpenAI API
 def generate_continuation(choice):
-    # Append player choice
     history.append({"role": "user", "content": f"> {choice}"})
 
-    # Get OpenAI response
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",  # or "gpt-4" if available
+    response = client.chat.completions.create(
+        model="gpt-3.5-turbo",  # You can switch to "gpt-4" if needed
         messages=history,
         temperature=0.8,
         max_tokens=150,
     )
 
-    continuation = response.choices[0].message["content"].strip()
+    continuation = response.choices[0].message.content.strip()
     history.append({"role": "assistant", "content": continuation})
 
     # Build full story text
     story_text = initial_scenario + "\n"
-    for msg in history[2:]:  # Skip system and initial
+    for msg in history[2:]:  # skip system and initial
         prefix = "You: " if msg["role"] == "user" else ""
         story_text += f"{prefix}{msg['content']}\n"
-    
+
     return story_text.strip()
 
+# Gradio interface
 with gr.Blocks() as app:
     with gr.Row():
         story_display = gr.Textbox(value=initial_scenario, lines=15, interactive=False, label="Story")