Updating the JSON files

app.py

@@ -56,6 +56,13 @@ def calculate_leaderboard(selected_game: str) -> pd.DataFrame:
     leaderboard_df.rename(columns={"index": "LLM Model"}, inplace=True)
     return leaderboard_df
 
+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
+
 # Gradio Interface
 with gr.Blocks() as interface:
     with gr.Tab("Game Arena"):
@@ -66,11 +73,18 @@ with gr.Blocks() as interface:
         
         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")
         
         def update_leaderboard(selected_game):
             """Updates the leaderboard table based on the selected game."""
             return calculate_leaderboard(selected_game)
         
+        def provide_download_file(model_name):
+            """Creates a downloadable JSON file with stats for the selected model."""
+            return generate_stats_file(model_name)
+        
         game_dropdown.change(fn=update_leaderboard, inputs=[game_dropdown], outputs=[leaderboard_table])
+        model_dropdown.change(fn=provide_download_file, inputs=[model_dropdown], outputs=[download_button])
 
 interface.launch()