Reset match history after load

background_task.py

@@ -6,13 +6,20 @@ from datetime import datetime
 from huggingface_hub import HfApi, Repository
 
 DATASET_REPO_URL = "https://huggingface.co/datasets/CarlCochet/BotFightData"
+DATASET_TEMP_REPO_URL = "https://huggingface.co/datasets/huggingface-projects/temp-match-results"
 ELO_FILENAME = "soccer_elo.csv"
+TEMP_FILENAME = "results.csv"
 ELO_DIR = "soccer_elo"
+TEMP_DIR = "temp"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
 repo = Repository(
     local_dir=ELO_DIR, clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
 )
+repo_temp = Repository(
+    local_dir=ELO_DIR, clone_from=DATASET_TEMP_REPO_URL, use_auth_token=HF_TOKEN
+)
+
 api = HfApi()
 os.chmod('./SoccerTows.x86_64', 0o755)
 
@@ -96,6 +103,12 @@ class Matchmaking:
                 self.matches["model2"].append(model2.name)
                 self.matches["result"].append(result)
                 self.matches["timestamp"].append(row["timestamp"])
+        data_dict = {"model1": [], "model2": [], "timestamp": [], "result": []}
+        df = pd.DataFrame(data_dict)
+        print(df.head())
+        repo_temp.git_pull()
+        df.to_csv(os.path.join(TEMP_DIR, TEMP_FILENAME), index=False)
+        repo_temp.push_to_hub(commit_message="Reset results.csv")
 
     def find_model(self, author, name):
         """ Find a model in the models list. """