Upload 3 files

README.md

@@ -13,7 +13,7 @@ pinned: false
 
 An AI-involved multiple-choice I Spy game from ZygAI Research.
 
-The app uses `meta-llama/Llama-3.1-8B-Instruct` as the game master. Llama creates each clue, four possible answers, the correct answer, and a short explanation. Players answer 10 rounds and get a final score.
+The app uses `meta-llama/Llama-3.1-8B-Instruct` as the game master. When the player starts, Llama generates all 10 clues, answer sets, correct answers, and explanations in one request. The game then plays from that cached set, which keeps the demo cost-effective.
 
 ## Hugging Face Space Setup
 

app.py

@@ -82,7 +82,7 @@ def get_client() -> InferenceClient | None:
     return InferenceClient(model=MODEL_ID, token=HF_TOKEN)
 
 
-def call_llama(messages: list[dict[str, str]], max_tokens: int = 500) -> str:
+def call_llama(messages: list[dict[str, str]], max_tokens: int = 1800) -> str:
     client = get_client()
     if not client:
         raise RuntimeError("HF_TOKEN is not configured.")
@@ -115,7 +115,25 @@ def fallback_round(round_number: int) -> dict[str, Any]:
     return data
 
 
-def generate_round(round_number: int, previous_answers: list[str]) -> dict[str, Any]:
+def validate_round(data: dict[str, Any]) -> dict[str, Any]:
+    clue = clean_text(data["clue"])
+    answer = clean_text(data["answer"])
+    options = [clean_text(option) for option in data["options"]]
+    explanation = clean_text(data.get("explanation", ""))
+
+    if len(options) != 4 or answer not in options:
+        raise ValueError("Round must have four options and include the answer.")
+
+    random.shuffle(options)
+    return {
+        "clue": clue,
+        "answer": answer,
+        "options": options,
+        "explanation": explanation or f"The answer is {answer}.",
+    }
+
+
+def generate_game() -> list[dict[str, Any]]:
     try:
         model_text = call_llama(
             [
@@ -123,37 +141,25 @@ def generate_round(round_number: int, previous_answers: list[str]) -> dict[str,
                 {
                     "role": "user",
                     "content": (
-                        f"Create round {round_number} of {TOTAL_ROUNDS} for I See. "
-                        "Make an I Spy-style clue and four answer options. "
-                        "The answer should be one of the options. "
-                        "Avoid repeating these previous answers: "
-                        f"{', '.join(previous_answers) or 'none'}. "
+                        f"Create all {TOTAL_ROUNDS} rounds for one I See demo game. "
+                        "Each round needs a short I Spy-style clue, four answer options, "
+                        "one correct answer that appears exactly in options, and one short explanation. "
+                        "Use varied everyday objects. Do not repeat correct answers. "
                         "Return only JSON with this exact shape: "
-                        '{"clue":"I see something that...","answer":"An umbrella",'
+                        '{"rounds":[{"clue":"I see something that...","answer":"An umbrella",'
                         '"options":["An umbrella","A chair","A candle","A spoon"],'
-                        '"explanation":"The umbrella keeps rain off you."}'
+                        '"explanation":"The umbrella keeps rain off you."}]}'
                     ),
                 },
             ]
         )
         data = extract_json(model_text)
-        clue = clean_text(data["clue"])
-        answer = clean_text(data["answer"])
-        options = [clean_text(option) for option in data["options"]]
-        explanation = clean_text(data.get("explanation", ""))
-
-        if len(options) != 4 or answer not in options:
-            raise ValueError("Round must have four options and include the answer.")
-
-        random.shuffle(options)
-        return {
-            "clue": clue,
-            "answer": answer,
-            "options": options,
-            "explanation": explanation or f"The answer is {answer}.",
-        }
+        rounds = [validate_round(round_data) for round_data in data["rounds"][:TOTAL_ROUNDS]]
+        if len(rounds) != TOTAL_ROUNDS:
+            raise ValueError("Game must include exactly 10 rounds.")
+        return rounds
     except Exception:
-        return fallback_round(round_number)
+        return [fallback_round(round_number) for round_number in range(1, TOTAL_ROUNDS + 1)]
 
 
 def option_text(round_data: dict[str, Any], index: int) -> str:
@@ -179,13 +185,13 @@ def render_question(state: dict[str, Any] | None) -> tuple[str, str, str, str, s
 
 
 def start_game() -> tuple[dict[str, Any], str, str, str, str, str, str, str, str]:
-    first_round = generate_round(1, [])
+    rounds = generate_game()
     state = {
         "round_number": 1,
         "score": 0,
         "answered": False,
-        "previous_answers": [first_round["answer"]],
-        "round": first_round,
+        "rounds": rounds,
+        "round": rounds[0],
     }
     return state, *render_question(state), "", gr.update(visible=False)
 
@@ -226,13 +232,11 @@ def next_round(state: dict[str, Any] | None) -> tuple[dict[str, Any] | None, str
         return state, message, "FINISHED", f"SCORE: {score}", "A", "B", "C", "D", "Start a new game to play again.", gr.update(visible=False)
 
     round_number = state["round_number"] + 1
-    round_data = generate_round(round_number, state["previous_answers"])
     state.update(
         {
             "round_number": round_number,
             "answered": False,
-            "round": round_data,
-            "previous_answers": state["previous_answers"] + [round_data["answer"]],
+            "round": state["rounds"][round_number - 1],
         }
     )
     return state, *render_question(state), "", gr.update(visible=False)