Changes

app.py

@@ -26,6 +26,7 @@ class GameState:
         self.current_image = None
         self.correct_country = None
         self.ai_prediction = None
+        self.ai_top3 = None
         self.options = []
         self.used_indices = []
 
@@ -82,25 +83,21 @@ def new_round():
     game.correct_country = sample["label"]
     game.current_image = image
     
-    # Get AI prediction
+    # Get AI prediction (but don't show it yet!)
     ai_pred, top3 = get_ai_prediction(image)
     game.ai_prediction = ai_pred
+    game.ai_top3 = top3
     
     # Generate options
     game.options = generate_options(game.correct_country, ai_pred)
     
-    # AI confidence message
-    ai_confidence = f"The AI predicts: **{ai_pred}**\n\n"
-    ai_confidence += "Top 3 predictions:\n"
-    for country, prob in top3:
-        ai_confidence += f"- {country}: {prob*100:.1f}%\n"
-    
     return (
         image,
-        "🌍 **Where do you think this image is from?**\n\n" + ai_confidence,
+        "🌍 **Where do you think this image is from?**\n\nMake your choice before seeing the AI's prediction!",
         gr.update(choices=game.options, value=None, visible=True),
         gr.update(visible=True),
-        f"🎮 Player: {game.player_score} | 🤖 AI: {game.ai_score} | 🎯 Rounds: {game.rounds}"
+        f"🎮 Player: {game.player_score} | 🤖 AI: {game.ai_score} | 🎯 Rounds: {game.rounds}",
+        ""  # Clear previous result
     )
 
 def check_answer(player_choice):
@@ -124,6 +121,7 @@ def check_answer(player_choice):
     result = f"## 🎯 Round {game.rounds} Result\n\n"
     result += f"**Correct country:** {game.correct_country}\n\n"
     
+    # Show comparison
     if player_correct and ai_correct:
         result += "🎉 **It's a tie!** Both you and the AI got it right!\n"
     elif player_correct:
@@ -136,11 +134,17 @@ def check_answer(player_choice):
     result += f"\n**Your answer:** {player_choice} {'✅' if player_correct else '❌'}\n"
     result += f"**AI prediction:** {game.ai_prediction} {'✅' if ai_correct else '❌'}\n"
     
+    # Show AI's top 3 predictions
+    result += f"\n**AI's top 3 predictions:**\n"
+    for country, prob in game.ai_top3:
+        result += f"- {country}: {prob*100:.1f}%\n"
+    
     # Calculate win rate
     if game.rounds > 0:
         player_rate = (game.player_score / game.rounds) * 100
         ai_rate = (game.ai_score / game.rounds) * 100
-        result += f"\n**Your accuracy:** {player_rate:.1f}% ({game.player_score}/{game.rounds})\n"
+        result += f"\n---\n"
+        result += f"**Your accuracy:** {player_rate:.1f}% ({game.player_score}/{game.rounds})\n"
         result += f"**AI accuracy:** {ai_rate:.1f}% ({game.ai_score}/{game.rounds})\n"
     
     return result, gr.update(visible=True)
@@ -200,7 +204,7 @@ with gr.Blocks(theme=gr.themes.Soft(), title="GeoGuessr: Player vs AI") as demo:
     start_btn.click(
         fn=new_round,
         inputs=[],
-        outputs=[image_display, question, options, submit_btn, scoreboard]
+        outputs=[image_display, question, options, submit_btn, scoreboard, result]
     )
     
     submit_btn.click(
@@ -223,4 +227,4 @@ with gr.Blocks(theme=gr.themes.Soft(), title="GeoGuessr: Player vs AI") as demo:
     """)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()