Trying to bring back the games tab and the files download

app.py

@@ -14,11 +14,34 @@ from typing import Dict
 llm_models = list(LLM_REGISTRY.keys())
 
 # Define game list manually (for now)
-games_list = list(GAMES_REGISTRY.keys())
+#games_list = list(GAMES_REGISTRY.keys())
+games_list = [
+    "rock_paper_scissors",
+    "prisoners_dilemma",
+    "tic_tac_toe",
+    "connect_four",
+    "matching_pennies",
+    "kuhn_poker",
+]
 
 # File to persist results
 RESULTS_TRACKER_FILE = "results_tracker.json"
 
+def generate_stats_file(model_name: str):
+    """Generate a JSON file with detailed statistics for the selected LLM model."""
+    file_path = f"{model_name}_stats.json"
+    with open(file_path, "w") as f:
+        json.dump(results_tracker.get(model_name, {}), f, indent=4)
+    return file_path
+
+def provide_download_file(model_name):
+    """Creates a downloadable JSON file with stats for the selected model."""
+    return generate_stats_file(model_name)
+
+def refresh_leaderboard():
+    """Manually refresh the leaderboard."""
+    return calculate_leaderboard(game_dropdown.value)
+
 # Load or initialize the results tracker
 if os.path.exists(RESULTS_TRACKER_FILE):
     with open(RESULTS_TRACKER_FILE, "r") as f:
@@ -37,10 +60,10 @@ def save_results_tracker():
 
 def calculate_leaderboard(selected_game: str) -> pd.DataFrame:
     """Generate a structured leaderboard table for the selected game."""
-    leaderboard_df = pd.DataFrame(index=llm_models, 
-                                  columns=["# games", "moves/game", 
+    leaderboard_df = pd.DataFrame(index=llm_models,
+                                  columns=["# games", "moves/game",
                                            "illegal-moves", "win-rate", "vs Random"])
-    
+
     for llm in llm_models:
         game_stats = results_tracker[llm].get(selected_game, {})
         leaderboard_df.loc[llm] = [
@@ -50,7 +73,7 @@ def calculate_leaderboard(selected_game: str) -> pd.DataFrame:
             f"{game_stats.get('win-rate', 0):.1f}%",
             f"{game_stats.get('vs Random', 0):.1f}%"
         ]
-    
+
     leaderboard_df = leaderboard_df.reset_index()
     leaderboard_df.rename(columns={"index": "LLM Model"}, inplace=True)
     return leaderboard_df
@@ -73,7 +96,7 @@ def play_game(game_name, player1_type, player2_type, player1_model, player2_mode
         legal_moves = state.legal_actions(current_player)
         board = str(state)
         game_states.append(f"Current Player: {current_player}\nBoard:\n{board}\nLegal Moves: {legal_moves}")
-    
+
     results = simulator.simulate(rounds=int(rounds), log_fn=log_fn)
     return "\n".join(game_states) + f"\nGame Result: {results}"
 
@@ -81,7 +104,7 @@ def play_game(game_name, player1_type, player2_type, player1_model, player2_mode
 with gr.Blocks() as interface:
     with gr.Tab("Game Arena"):
         gr.Markdown("# LLM Game Arena\nSelect a game and players to play against LLMs.")
-        
+
         game_dropdown = gr.Dropdown(choices=games_list, label="Select a Game", value=games_list[0])
         player1_dropdown = gr.Dropdown(choices=["human", "random_bot", "llm"], label="Player 1 Type", value="llm")
         player2_dropdown = gr.Dropdown(choices=["human", "random_bot", "llm"], label="Player 2 Type", value="random_bot")
@@ -99,17 +122,18 @@ with gr.Blocks() as interface:
 
     with gr.Tab("Leaderboard"):
         gr.Markdown("# LLM Model Leaderboard\nTrack performance across different games!")
-        
+
         game_dropdown = gr.Dropdown(choices=games_list, label="Select Game", value=games_list[0])
         leaderboard_table = gr.Dataframe(value=calculate_leaderboard(games_list[0]), label="Leaderboard")
         model_dropdown = gr.Dropdown(choices=llm_models, label="Select LLM Model")
         download_button = gr.File(label="Download Statistics File")
         refresh_button = gr.Button("Refresh Leaderboard")
-        
+
         def update_leaderboard(selected_game):
             """Updates the leaderboard table based on the selected game."""
             return calculate_leaderboard(selected_game)
-        
+
+        model_dropdown.change(fn=provide_download_file, inputs=[model_dropdown], outputs=[download_button])
         game_dropdown.change(fn=update_leaderboard, inputs=[game_dropdown], outputs=[leaderboard_table])
         refresh_button.click(fn=update_leaderboard, inputs=[game_dropdown], outputs=[leaderboard_table])