Update app.py

app.py

@@ -39,70 +39,66 @@ def game_logic(action):
     global current_room, inventory, game_started
     room = rooms.get(current_room)
 
-    # --- Main Menu ---
     if not game_started:
         if action == "Start Game":
             game_started = True
             current_room = "dark_room"
             inventory.clear()
-            return f"🎮 Game started!\n{rooms[current_room]['description']}", update_buttons()
+            return f"🎮 Game started!\n{rooms[current_room]['description']}"
         elif action == "Help":
-            return help_text, ["Start Game", "Help", "Exit"]
+            return help_text
         elif action == "Exit":
-            return "Goodbye! Thanks for playing.", []
-        return "Select an option to start.", ["Start Game", "Help", "Exit"]
+            return "Goodbye! Thanks for playing."
+        return "Select an option to start."
 
-    # --- Help Menu ---
     if action == "Help":
-        return help_text, update_buttons()
+        return help_text
 
-    # --- Room Actions ---
     if action.startswith("Look"):
         if current_room == "dark_room" and not room["puzzles"]["candle_lit"]:
-            return "It’s too dark to see much. You might need to light the candle first.", update_buttons()
+            return "It’s too dark to see much. You might need to light the candle first."
         items = ", ".join(room["items"]) if room["items"] else "nothing"
-        return f"{room['description']} You see: {items}.", update_buttons()
+        return f"{room['description']} You see: {items}."
 
     if action.startswith("Pick up "):
         item = action.replace("Pick up ", "")
         if item in room["items"]:
             inventory.append(item)
             room["items"].remove(item)
-            return f"You picked up the {item}.", update_buttons()
-        return f"No {item} here to pick up.", update_buttons()
+            return f"You picked up the {item}."
+        return f"No {item} here to pick up."
 
     if action.startswith("Use "):
         item = action.replace("Use ", "")
         if item in inventory:
             if item == "candle":
                 room["puzzles"]["candle_lit"] = True
-                return "You light the candle. Now you can see everything clearly!", update_buttons()
-            return f"You used the {item}, but nothing happened.", update_buttons()
-        return f"You don't have a {item}.", update_buttons()
+                return "You light the candle. Now you can see everything clearly!"
+            return f"You used the {item}, but nothing happened."
+        return f"You don't have a {item}."
 
     if action in room["exits"]:
         if room["door_locked"] and action == "door":
             if "key" in inventory:
                 room["door_locked"] = False
                 current_room = "hallway"
-                return f"You unlock the door with the key and enter the hallway.", update_buttons()
-            return "The door is locked. You need a key.", update_buttons()
+                return f"You unlock the door with the key and enter the hallway."
+            return "The door is locked. You need a key."
         current_room = action
-        return f"You move into {current_room}. {rooms[current_room]['description']}", update_buttons()
+        return f"You move into {current_room}. {rooms[current_room]['description']}"
 
     if action == "Inventory":
-        return f"You have: {', '.join(inventory) if inventory else 'nothing'}", update_buttons()
+        return f"You have: {', '.join(inventory) if inventory else 'nothing'}"
 
     if action == "Exit":
-        return "Thanks for playing! Goodbye!", []
+        return "Thanks for playing! Goodbye!"
 
-    return "I don't understand that action.", update_buttons()
+    return "I don't understand that action."
 
-# --- Update Buttons Based on Room ---
-def update_buttons():
+# --- Determine Which Buttons to Show ---
+def get_visible_buttons():
     if not game_started:
         return ["Start Game", "Help", "Exit"]
-
     room = rooms[current_room]
     buttons = ["Look", "Inventory", "Help", "Exit"]
     buttons += [f"Pick up {item}" for item in room["items"]]
@@ -111,35 +107,42 @@ def update_buttons():
 
 # --- Gradio UI ---
 with gr.Blocks() as demo:
-    gr.Markdown("# 🏰 Escape Room Adventure (Clickable Buttons)")
+    gr.Markdown("# 🏰 Escape Room Adventure (Dynamic Buttons)")
 
     chatbot = gr.Chatbot(type="messages")
     state = gr.State()
-    buttons_container = gr.Row()
+    msg = gr.Textbox(label="Type your action")
 
-    # --- Function to handle button clicks ---
+    # Pre-create all possible buttons
+    all_button_labels = [
+        "Start Game", "Help", "Exit", "Look", "Inventory",
+        "Pick up candle", "Pick up key", "Pick up painting",
+        "door", "trapdoor", "stairs", "painting", "Use candle"
+    ]
+    buttons_dict = {}
+    for label in all_button_labels:
+        btn = gr.Button(label)
+        buttons_dict[label] = btn
+
+    # Respond function
     def respond(action, history):
         history = history or []
-        bot_reply, buttons = game_logic(action)
-        # Append messages in 'messages' format
+        bot_reply = game_logic(action)
         history.append({"role": "user", "content": action})
         history.append({"role": "assistant", "content": bot_reply})
-        update_buttons_ui(buttons)
-        return history, history
 
-    # --- Update clickable buttons dynamically ---
-    def update_buttons_ui(button_labels):
-        buttons_container.children.clear()
-        for label in button_labels:
-            btn = gr.Button(label)
-            btn.click(respond, inputs=[btn, state], outputs=[chatbot, state])
-        return
+        # Update button visibility
+        visible = get_visible_buttons()
+        for lbl, btn in buttons_dict.items():
+            btn.update(visible=(lbl in visible))
 
-    # Initial buttons
-    update_buttons_ui(update_buttons())
+        return history, history
+
+    # Connect buttons to respond
+    for btn in buttons_dict.values():
+        btn.click(respond, inputs=[btn, state], outputs=[chatbot, state])
 
     # Manual input
-    msg = gr.Textbox(label="Type your action")
     msg.submit(respond, inputs=[msg, state], outputs=[chatbot, state])
 
 demo.launch()