Update app.py

app.py

@@ -2,11 +2,11 @@ import gradio as gr
 import os
 from openai import OpenAI
 
-# Load API key from Hugging Face secret
+# Load API key from environment variable
 api_key = os.getenv("OPENAI_API_KEY")
 client = OpenAI(api_key=api_key)
 
-# Initial state
+# Initial game scenario and default choices
 initial_scenario = "You wake up in a strange forest. Paths lead in all directions."
 default_choices = [
     "Walk north into the dense forest.",
@@ -15,7 +15,7 @@ default_choices = [
     "Move west where the land rises."
 ]
 
-# System prompt for role instruction
+# System prompt to guide the assistant behavior
 system_prompt = {
     "role": "system",
     "content": (
@@ -25,18 +25,20 @@ system_prompt = {
     )
 }
 
-# Initialize history
+# Global story and history state
 history = [system_prompt, {"role": "user", "content": initial_scenario}]
 story_log = initial_scenario
 
 
 def parse_choices_from_response(text):
+    """Extract numbered options from the assistant's response"""
     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
 
 
 def update_story(player_input):
+    """Add user input, get AI response, update story and extract new choices"""
     global story_log
 
     history.append({"role": "user", "content": f"> {player_input}"})
@@ -54,7 +56,10 @@ def update_story(player_input):
     story_log += f"\n\nYou: {player_input}\n{continuation}"
     new_choices = parse_choices_from_response(continuation)
 
-    return story_log, new_choices
+    while len(new_choices) < 4:
+        new_choices.append("...")
+
+    return [story_log] + new_choices[:4]
 
 
 def handle_button(choice_text):
@@ -63,7 +68,7 @@ def handle_button(choice_text):
 
 def handle_custom_input(text_input):
     if not text_input.strip():
-        return story_log, default_choices
+        return [story_log] + default_choices
     return update_story(text_input.strip())
 
 
@@ -71,27 +76,31 @@ 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_log] + default_choices
 
 
+# Gradio UI
 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():
-        choice_btns = [gr.Button(choice) for choice in default_choices]
+        btn1 = gr.Button(default_choices[0])
+        btn2 = gr.Button(default_choices[1])
+        btn3 = gr.Button(default_choices[2])
+        btn4 = gr.Button(default_choices[3])
 
     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])
-
-    # Link custom text input
-    custom_input.submit(fn=handle_custom_input, inputs=custom_input, outputs=[story_display, *choice_btns])
+    # Bind buttons to handlers
+    btn1.click(fn=handle_button, inputs=btn1, outputs=[story_display, btn1, btn2, btn3, btn4])
+    btn2.click(fn=handle_button, inputs=btn2, outputs=[story_display, btn1, btn2, btn3, btn4])
+    btn3.click(fn=handle_button, inputs=btn3, outputs=[story_display, btn1, btn2, btn3, btn4])
+    btn4.click(fn=handle_button, inputs=btn4, outputs=[story_display, btn1, btn2, btn3, btn4])
 
-    # Restart
-    restart_btn.click(fn=restart_game, inputs=None, outputs=[story_display, *choice_btns])
+    custom_input.submit(fn=handle_custom_input, inputs=custom_input, outputs=[story_display, btn1, btn2, btn3, btn4])
+    restart_btn.click(fn=restart_game, inputs=None, outputs=[story_display, btn1, btn2, btn3, btn4])
 
 app.launch()