US375432 - update game modes and examples for improved gameplay experience

.gitignore

@@ -3,4 +3,5 @@
 __pycache__/
 *.py[cod]
 *$py.class
-.pytest_cache/
+.pytest_cache/
+geography_data/

app.py

@@ -119,8 +119,21 @@ if ai.LOCAL_MODE:
 else:
     description = "Choose your game mode and I'll think of a location for you to guess with 20 yes or no questions!"
 
-# Common examples and settings - updated to be more generic
-examples = [
+# Mode-specific examples
+examples_states = [
+    ["Is it on the East Coast?"],
+    ["Is it in the Western United States?"],
+    ["Does it border an ocean?"],
+    ["Does it border Canada?"],
+    ["Is it larger than Texas?"],
+    ["Was it one of the original 13 colonies?"],
+    ["Does it have a major city with over 1 million people?"],
+    ["Is it known for agriculture?"],
+    ["Does it have mountains?"],
+    ["Is it in the Mountain Time Zone?"],
+]
+
+examples_countries = [
     ["Is it located in North America?"],
     ["Is it in the Northern Hemisphere?"],
     ["Does it border an ocean?"],
@@ -130,18 +143,28 @@ examples = [
     ["Was it ever a British colony?"],
     ["Is it located on an island?"],
     ["Does it use the US Dollar as currency?"],
-    ["Is it a landlocked location?"],
+    ["Is it a landlocked country?"],
 ]
 
+# Default examples (will be updated based on mode)
+examples = examples_states
+
 # Create wrapper function that handles both local and cloud modes
 # Local mode wrapper function
-def custom_respond(message, history, game_mode_selection):
+def custom_respond(message, history, game_mode_selection, game_type_selection):
     return respond(message, history, "", 4000, 0.3, 0.6, game_mode_selection, None)
 
+def get_examples_for_mode(mode):
+    """Return examples based on the selected game mode"""
+    if mode == game.MODE_STATES:
+        return examples_states
+    else:
+        return examples_countries
+
 with gr.Blocks() as demo:
     
     
-    gr.Markdown("# 20 Questions Geography Game")
+    gr.Markdown("# Geography AI Tutor")
     gr.Markdown(description)
     
     # Game mode selection at the top
@@ -151,14 +174,53 @@ with gr.Blocks() as demo:
         label="Game Mode",
         info="Choose what type of location to guess"
     )
+
+    game.type_dropdown = gr.Dropdown(
+        choices=[game.TYPE_GEOGRAPHY_BEE, game.TYPE_TWENTY_QUESTIONS],
+        value=game.TYPE_GEOGRAPHY_BEE,
+        label="Game Type",
+        info="Choose what type of game to play"
+    )
     
-    
+    # ChatInterface without built-in examples
     chatbot = gr.ChatInterface(
         custom_respond,
         type="messages",
-        examples=examples,
         cache_examples=False,
-        additional_inputs=[game.mode_dropdown],
+        additional_inputs=[game.mode_dropdown, game.type_dropdown],
+    )
+    
+    # Add examples separately using Dataset which can be updated
+    with gr.Row():
+        gr.Markdown("### Example Questions (click to use)")
+    
+    examples_dataset = gr.Dataset(
+        components=[chatbot.textbox],
+        samples=examples_states,
+        type="index"
+    )
+    
+    # Update examples when mode changes
+    def update_examples(mode):
+        return gr.Dataset(samples=get_examples_for_mode(mode))
+    
+    # When an example is clicked, populate the textbox
+    def load_example(index, mode):
+        examples = get_examples_for_mode(mode)
+        if 0 <= index < len(examples):
+            return examples[index][0]
+        return ""
+    
+    game.mode_dropdown.change(
+        fn=update_examples,
+        inputs=[game.mode_dropdown],
+        outputs=[examples_dataset]
+    )
+    
+    examples_dataset.select(
+        fn=load_example,
+        inputs=[examples_dataset, game.mode_dropdown],
+        outputs=[chatbot.textbox]
     )
     gr.Markdown(
         """

game.py

@@ -16,6 +16,9 @@ with open(STATES_JSON_PATH, "r", encoding="utf-8") as f:
 MODE_COUNTRIES = "Countries of the World"
 MODE_STATES = "US States"
 
+TYPE_TWENTY_QUESTIONS = "20 Questions Geography Game"
+TYPE_GEOGRAPHY_BEE = "Geography Bee"
+
 
 current_system = ""
 selected_country = ""