Update app.py

app.py

@@ -32,8 +32,8 @@ def extract_agent_info(filename: str):
         agent_type, model_name = parts[0], "Unknown"
     return agent_type, model_name
 
-def get_available_games() -> List[str]:
-    """Extracts all unique game names from all SQLite databases and includes 'Aggregated Performance'."""
+def get_available_games(include_aggregated=True) -> List[str]:
+    """Extracts all unique game names from all SQLite databases. Includes 'Aggregated Performance' only when required."""
     db_files = find_or_download_db()
     game_names = set()
     
@@ -49,7 +49,8 @@ def get_available_games() -> List[str]:
             conn.close()
     
     game_list = sorted(game_names) if game_names else ["No Games Found"]
-    game_list.insert(0, "Aggregated Performance")  # Ensure 'Aggregated Performance' is always first
+    if include_aggregated:
+        game_list.insert(0, "Aggregated Performance")  # Ensure 'Aggregated Performance' is always first
     return game_list
 
 def extract_leaderboard_stats(game_name: str) -> pd.DataFrame:
@@ -124,7 +125,7 @@ def extract_leaderboard_stats(game_name: str) -> pd.DataFrame:
 with gr.Blocks() as interface:
     with gr.Tab("Game Arena"):
         gr.Markdown("# Play Against LLMs\nChoose a game and an opponent to play!")
-        game_dropdown = gr.Dropdown(get_available_games(), label="Select a Game")
+        game_dropdown = gr.Dropdown(get_available_games(include_aggregated=False), label="Select a Game")
         opponent_dropdown = gr.Dropdown(["Random Bot", "LLM"], label="Choose Opponent")
         play_button = gr.Button("Start Game")
         game_output = gr.Textbox(label="Game Log")
@@ -133,8 +134,7 @@ with gr.Blocks() as interface:
     
     with gr.Tab("Leaderboard"):
         gr.Markdown("# LLM Model Leaderboard\nTrack performance across different games!")
-        available_games = get_available_games()
-        leaderboard_game_dropdown = gr.Dropdown(available_games, label="Select Game", value="Aggregated Performance")
+        leaderboard_game_dropdown = gr.Dropdown(get_available_games(), label="Select Game", value="Aggregated Performance")
         leaderboard_table = gr.Dataframe(value=extract_leaderboard_stats("Aggregated Performance"), headers=["agent_name", "agent_type", "# games", "total rewards", "avg_generation_time (sec)", "win-rate", "win vs_random (%)"], every=5)
     
     interface.launch()