Update app.py

app.py

@@ -32,6 +32,24 @@ 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."""
+    db_files = find_or_download_db()
+    game_names = set()
+    
+    for db_file in db_files:
+        conn = sqlite3.connect(db_file)
+        try:
+            query = "SELECT DISTINCT game_name FROM moves"
+            df = pd.read_sql_query(query, conn)
+            game_names.update(df["game_name"].tolist())
+        except Exception:
+            pass  # Ignore errors if table doesn't exist
+        finally:
+            conn.close()
+    
+    return sorted(game_names) if game_names else ["No Games Found"]
+
 def extract_leaderboard_stats(game_name: str) -> pd.DataFrame:
     """Extract and aggregate leaderboard stats from all SQLite databases."""
     db_files = find_or_download_db()
@@ -41,16 +59,11 @@ def extract_leaderboard_stats(game_name: str) -> pd.DataFrame:
         conn = sqlite3.connect(db_file)
         agent_type, model_name = extract_agent_info(db_file)
         
-        if game_name == "Total Performance":
-            query = "SELECT game_name, COUNT(DISTINCT episode) AS games_played, " \
-                    "AVG(generation_time) AS avg_gen_time, SUM(reward) AS total_rewards " \
-                    "FROM game_results GROUP BY game_name"
-        else:
-            query = "SELECT COUNT(DISTINCT episode) AS games_played, " \
-                    "AVG(generation_time) AS avg_gen_time, SUM(reward) AS total_rewards " \
-                    "FROM game_results WHERE game_name = ?"
+        query = "SELECT COUNT(DISTINCT episode) AS games_played, " \
+                "AVG(generation_time) AS avg_gen_time, SUM(reward) AS total_rewards " \
+                "FROM game_results WHERE game_name = ?"
         
-        df = pd.read_sql_query(query, conn, params=(game_name,) if game_name != "Total Performance" else None)
+        df = pd.read_sql_query(query, conn, params=(game_name,))
         df["agent_name"] = model_name
         df["agent_type"] = agent_type
         all_stats.append(df)
@@ -61,7 +74,8 @@ def extract_leaderboard_stats(game_name: str) -> pd.DataFrame:
 
 def generate_leaderboard_json():
     """Generate a JSON file containing leaderboard stats."""
-    leaderboard = extract_leaderboard_stats("Total Performance").to_dict(orient="records")
+    available_games = get_available_games()
+    leaderboard = extract_leaderboard_stats(available_games[0]).to_dict(orient="records")
     json_file = "results/leaderboard_stats.json"
     with open(json_file, "w", encoding="utf-8") as f:
         json.dump({"timestamp": datetime.utcnow().isoformat(), "leaderboard": leaderboard}, f, indent=4)
@@ -70,16 +84,15 @@ def generate_leaderboard_json():
 with gr.Blocks() as interface:
     with gr.Tab("Leaderboard"):
         gr.Markdown("# Leaderboard")
-        leaderboard_game_dropdown = gr.Dropdown(["Total Performance", "tic_tac_toe"], label="Select Game")
+        available_games = get_available_games()
+        leaderboard_game_dropdown = gr.Dropdown(available_games, label="Select Game", value=available_games[0])
         leaderboard_table = gr.Dataframe()
-        download_button = gr.File(label="Download Leaderboard")
+        generate_button = gr.Button("Generate Leaderboard JSON")
+        download_component = gr.File(label="Download Leaderboard JSON")
         refresh_button = gr.Button("Refresh Leaderboard")
+        
         leaderboard_game_dropdown.change(extract_leaderboard_stats, inputs=[leaderboard_game_dropdown], outputs=[leaderboard_table])
         refresh_button.click(extract_leaderboard_stats, inputs=[leaderboard_game_dropdown], outputs=[leaderboard_table])
-        generate_button = gr.Button("Generate Leaderboard JSON")
-        download_component = gr.File(label="Download Leaderboard JSON")
-
         generate_button.click(generate_leaderboard_json, outputs=[download_component])
 
-
 interface.launch()