very basic start

app.py

@@ -151,9 +151,9 @@ 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, game_type_selection):
+def custom_respond(message, history, game_type_selection, game_mode_selection):
     if game_type_selection == game.TYPE_GEOGRAPHY_BEE:
-        return bee.respond(message, history, game_mode_selection)
+        return bee.respond(message, history, 4000, game_mode_selection)
     return respond(message, history, "", 4000, 0.3, 0.6, game_mode_selection, None)
 
 def get_examples_for_mode(mode):

bee.py

@@ -1,15 +1,45 @@
 import gradio as gr
 import game 
+import ai
 from app import format_game_result
+
+def get_system_message(game_mode):
+    return """You are a friendly classroom tutor for 9-10 year-olds.
+Generate one multiple-choice geography question that a 4th-grader would answer.
+
+Requirements:
+1. Question text - one short sentence (or two) that asks the student to identify a place, landmark, or simple map feature.
+2. Four answer options labeled A)-D). All options should be plausible country, state, or landmark names.
+3. After the question and choices, include a short "Answer key" line that states the correct answer (e.g., "Correct answer: A").
+4. If a student gives an incorrect answer, respond with:
+
+   "That's okay! Let's look at why that answer isn't right and see what clues the question gives us. Remember, …"
+
+   Then provide a brief explanation (1-2 sentences) that points the student toward the correct answer without giving it away outright.
+
+5. Keep vocabulary simple, concrete, and age-appropriate.
+6. Do not mention the word "multiple choice" in the output; just give the question, options, answer key, and the incorrect-answer reaction.
+
+Example output (use as a guide only, do NOT copy):
+
+**Question:** Which country is shaped like a boot and is located in southern Europe?
+
+A) France
+B) Italy
+C) Greece
+D) Spain
+
+**Answer key:** Correct answer: B
+
+If the student answers incorrectly:
+"That's okay! Let's look at why that answer isn't right and see what clues the question gives us. Remember, the question mentions the country looks like a boot. Think about which country on a map of Europe really looks like you could wear it on your foot!"
+"""
+
 def respond(
     message,
     history: list[dict[str, str]],
-    system_message,
     max_tokens,
-    temperature,
-    top_p,
     game_mode_selection,
-    hf_token: gr.OAuthToken | None = None,
 ):
     """
     For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
@@ -18,12 +48,8 @@ def respond(
     # If this is the start of a new conversation (empty history), generate a new country/state
     if not history:
         game.guess_number = 0
-        if game_mode_selection == game.MODE_STATES:
-            game.current_system = game.get_system_message(game.MODE_STATES)
-            print(f"🔍 DEBUG - New session started, selected state: {game.selected_country}")
-        else:
-            game.current_system = game.get_system_message(game.MODE_COUNTRIES)
-            print(f"🔍 DEBUG - New session started, selected country: {game.selected_country}")
+        game.current_system = get_system_message(game.MODE_STATES)
+        print(f"🔍 DEBUG - New session started, selected state: {game.selected_country}")
 
     game.guess_number += 1
     messages = [{"role": "system", "content": game.current_system + str(game.guess_number)}]