Updated readme; tidied leaderboard

README.md

@@ -21,8 +21,8 @@ short_description: Arena for playing Four-in-a-row between LLMs
 It has been great fun making this Arena and watching LLMs duke it out!
 
 Quick links:
+- The [Live Arena](https://edwarddonner.com/connect-four/)  courtesy of amazing HuggingFace Spaces
 - The [GitHub repo](https://github.com/ed-donner/connect) for the code
-- The [HuggingFace Spaces](https://huggingface.co/spaces/ed-donner/connect) where it's running
 - My [LinkedIn](https://www.linkedin.com/in/eddonner/) - I love connecting!
 
 If you'd like to learn more about this:  
@@ -37,6 +37,8 @@ If you'd like to learn more about this:
 4. Activate your environment with either `venv\Scripts\activate` on Windows, or `source venv/bin/activate` on Mac/Linux
 5. Then run `pip install -r requirements.txt` to install the packages
 
+If you wish to play with the prototype, run `jupyter lab` to launch the lab then look at the notebook **prototype.ipynb**.
+
 ## Setting up your API keys
 
 Please create a file with the exact name `.env` in the project root directory (connect).

arena/c4.py

@@ -2,6 +2,7 @@ from arena.game import Game
 from arena.board import RED, YELLOW
 from arena.llm import LLM
 import gradio as gr
+import pandas as pd
 
 
 css = """
@@ -35,17 +36,25 @@ def message_html(game) -> str:
 
 def format_records_for_table(games):
     """
-    Turn the results objects into a list of lists for the Gradio Dataframe
+    Turn the results objects into a pandas DataFrame for the Gradio Dataframe
     """
-    return [
+    df = pd.DataFrame(
         [
-            game.when,
-            game.red_player,
-            game.yellow_player,
-            "Red" if game.red_won else "Yellow" if game.yellow_won else "Draw",
-        ]
-        for game in reversed(games)
-    ]
+            [
+                game.when,
+                game.red_player,
+                game.yellow_player,
+                "Red" if game.red_won else "Yellow" if game.yellow_won else "Draw",
+            ]
+            for game in reversed(games)
+        ],
+        columns=["When", "Red Player", "Yellow Player", "Winner"],
+    )
+
+    # Remove microseconds while preserving datetime format
+    df["When"] = pd.to_datetime(df["When"]).dt.floor("s")
+
+    return df
 
 
 def format_ratings_for_table(ratings):

requirements.txt

@@ -8,4 +8,5 @@ google.generativeai
 anthropic
 groq
 black
-pymongo
+pymongo
+pandas