upload AI scores too

app.py

@@ -881,11 +881,11 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
                 "round_number": round_idx,
                 "actual_location": {"lat": rnd.get('lat'), "lng": rnd.get('lng')},
                 "human_guess": rnd.get('human_guess'),
-                "human_distance_km": rnd.get('human_distance_km'),
-                "human_score": rnd.get('human_score'),
+                "human_distance_km": round(rnd.get('human_distance_km', 0), 2),
+                "human_score": int(round(rnd.get('human_score', 0))),
                 "ai_guess": rnd.get('ai_guess'),
-                "ai_distance_km": rnd.get('ai_distance_km'),
-                "ai_score": rnd.get('ai_score'),
+                "ai_distance_km": round(rnd.get('ai_distance_km', 0), 2) if rnd.get('ai_distance_km') else None,
+                "ai_score": int(round(rnd.get('ai_score', 0))) if rnd.get('ai_score') else 0,
                 "ai_analysis": rnd.get('ai_analysis', ''),
             }
             data_manager.update_game_record(username, round_data=round_record)
@@ -962,8 +962,8 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
             total_human = sum(float(r.get('human_score', 0.0)) for r in sess.get('rounds', []))
             total_ai = sum(float(r.get('ai_score', 0.0)) for r in sess.get('rounds', []))
             
-            # Update the final score (rounds already recorded incrementally)
-            data_manager.update_game_record(username, final_score=total_human)
+            # Update the final scores (rounds already recorded incrementally)
+            data_manager.update_game_record(username, final_score=total_human, final_ai_score=total_ai)
 
             winner_message = "It's a tie!"
             if total_human > total_ai:

data_manager.py

@@ -54,7 +54,7 @@ def get_user_game_today(username: str, todays_games: list) -> dict:
             return game
     return None
 
-def update_game_record(username: str, round_data: dict = None, final_score: float = None):
+def update_game_record(username: str, round_data: dict = None, final_score: float = None, final_ai_score: float = None):
     """
     Updates or creates a game record for a user after each round.
     This ensures data is recorded incrementally and prevents abuse.
@@ -62,7 +62,8 @@ def update_game_record(username: str, round_data: dict = None, final_score: floa
     Args:
         username: The player's username
         round_data: Single round details to append to the record
-        final_score: Final total score (only set when game is complete)
+        final_score: Final total human score (only set when game is complete)
+        final_ai_score: Final total AI score (only set when game is complete)
     """
     write_token = os.getenv("HF_TOKEN", "")
     if not write_token:
@@ -83,13 +84,16 @@ def update_game_record(username: str, round_data: dict = None, final_score: floa
                     existing_game["rounds"] = []
                 existing_game["rounds"].append(round_data)
             if final_score is not None:
-                existing_game["score"] = final_score
+                existing_game["human_score"] = int(round(final_score))
                 existing_game["completed"] = True
+            if final_ai_score is not None:
+                existing_game["ai_score"] = int(round(final_ai_score))
         else:
             # Create new game record
             game_record = {
                 "username": username,
-                "score": 0,  # Will be updated when game completes
+                "human_score": 0,  # Will be updated when game completes
+                "ai_score": 0,  # Will be updated when game completes
                 "timestamp": datetime.now(timezone.utc).isoformat(),
                 "rounds": [round_data] if round_data else [],
                 "completed": False
@@ -118,9 +122,9 @@ def update_game_record(username: str, round_data: dict = None, final_score: floa
         if 'tmp_file_path' in locals() and os.path.exists(tmp_file_path):
             os.remove(tmp_file_path)
 
-def record_game(username: str, score: float, rounds_data: list = None):
+def record_game(username: str, score: float, rounds_data: list = None, ai_score: float = None):
     """
     Legacy function - now just calls update_game_record with final score.
     Kept for backwards compatibility.
     """
-    update_game_record(username, final_score=score)
+    update_game_record(username, final_score=score, final_ai_score=ai_score)