Update app.py

app.py

@@ -2,66 +2,96 @@ import gradio as gr
 import os
 from openai import OpenAI
 
-# Initialize OpenAI client using API key from environment
+# Load API key from Hugging Face secret
 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 = [
+# Initial state
+initial_scenario = "You wake up in a strange forest. Paths lead in all directions."
+default_choices = [
     "Walk north into the dense forest.",
     "Go south back towards a distant town.",
     "Explore east towards the sound of water.",
     "Move west where the land rises."
 ]
 
-# Conversation history starts with a system prompt and the opening scene
-history = [
-    {
-        "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}
-]
+# System prompt for role instruction
+system_prompt = {
+    "role": "system",
+    "content": (
+        "You are a text-based RPG engine. Continue the story based on the player's input. "
+        "Include vivid descriptions and always end your response with a few suggested choices. "
+        "Format them as a numbered list (1., 2., etc). Do not break character or give explanations."
+    )
+}
+
+# Initialize history
+history = [system_prompt, {"role": "user", "content": initial_scenario}]
+story_log = initial_scenario
+
+
+def parse_choices_from_response(text):
+    lines = text.strip().split("\n")
+    options = [line.strip() for line in lines if line.strip().startswith(("1.", "2.", "3.", "4."))]
+    return options[:4] if options else default_choices
 
-# Function to generate story continuation from the OpenAI API
-def generate_continuation(choice):
-    history.append({"role": "user", "content": f"> {choice}"})
+
+def update_story(player_input):
+    global story_log
+
+    history.append({"role": "user", "content": f"> {player_input}"})
 
     response = client.chat.completions.create(
-        model="gpt-3.5-turbo",  # You can switch to "gpt-4" if needed
+        model="gpt-3.5-turbo",
         messages=history,
-        temperature=0.8,
-        max_tokens=150,
+        temperature=0.9,
+        max_tokens=300
     )
 
     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
-        prefix = "You: " if msg["role"] == "user" else ""
-        story_text += f"{prefix}{msg['content']}\n"
+    story_log += f"\n\nYou: {player_input}\n{continuation}"
+    new_choices = parse_choices_from_response(continuation)
+
+    return story_log, new_choices
+
+
+def handle_button(choice_text):
+    return update_story(choice_text)
+
+
+def handle_custom_input(text_input):
+    if not text_input.strip():
+        return story_log, default_choices
+    return update_story(text_input.strip())
+
+
+def restart_game():
+    global history, story_log
+    history = [system_prompt, {"role": "user", "content": initial_scenario}]
+    story_log = initial_scenario
+    return story_log, default_choices
 
-    return story_text.strip()
 
-# Gradio interface
 with gr.Blocks() as app:
+    gr.Markdown("## 🌲 Text-Based RPG Adventure")
+    story_display = gr.Textbox(value=initial_scenario, lines=20, interactive=False, label="Your Story")
+
     with gr.Row():
-        story_display = gr.Textbox(value=initial_scenario, lines=15, interactive=False, label="Story")
-    with gr.Row():
-        choice_buttons = [gr.Button(choice) for choice in choices]
+        choice_btns = [gr.Button(choice) for choice in default_choices]
+
+    custom_input = gr.Textbox(lines=1, placeholder="Or type your action here...", label="Your Action")
+    restart_btn = gr.Button("🔄 Restart Game")
+
+    # Link button events
+    for btn in choice_btns:
+        btn.click(fn=handle_button, inputs=btn, outputs=[story_display, *choice_btns])
 
-    def handle_choice(choice_index):
-        chosen_action = choices[choice_index]
-        updated_story = generate_continuation(chosen_action)
-        return updated_story
+    # Link custom text input
+    custom_input.submit(fn=handle_custom_input, inputs=custom_input, outputs=[story_display, *choice_btns])
 
-    for i, button in enumerate(choice_buttons):
-        button.click(fn=lambda x=i: handle_choice(x), inputs=None, outputs=story_display)
+    # Restart
+    restart_btn.click(fn=restart_game, inputs=None, outputs=[story_display, *choice_btns])
 
 app.launch()