Added ability to wipe the logs from the system status page

pages/900_System_Status.py

@@ -3,11 +3,18 @@ from time import sleep
 
 from src.st_helpers import st_setup
 from src.common import *
+from src.architectures import Architecture
 
 
 if st_setup('LLM Arch'):
     st.write("# System Status")
 
+    st.write("## Wipe Trace Logs")
+    if st.button("Wipe logs"):
+        Architecture.wipe_trace()
+
+    st.divider()
+
     st.write("## HF Inference Endpoint Statuses")
     st.write("The following endpoints need to be running to run all the demonstrations.")
 

src/architectures.py

@@ -8,6 +8,7 @@ import json
 import os
 import regex as re
 import requests
+import shutil
 import traceback
 
 from abc import ABC, abstractmethod
@@ -197,6 +198,27 @@ class Architecture:
     save_repo = None
     save_repo_load_error = False
 
+    @classmethod
+    def wipe_trace(cls):
+        repo_url = "https://huggingface.co/datasets/alfraser/llm-arch-trace"
+        trace_dir = "trace"
+        trace_file_name = "trace.json"
+        data_file = os.path.join(trace_dir, trace_file_name)
+        if os.path.exists(trace_dir):
+            shutil.rmtree(trace_dir)
+
+        try:
+            hf_write_token = hf_api_token(write=True)
+            Architecture.save_repo = Repository(local_dir=trace_dir, clone_from=repo_url, token=hf_write_token)
+            test_json = {'tests': []}
+            with open(data_file, 'w') as f:
+                json.dump(test_json, f, indent=2)
+            Architecture.save_repo.push_to_hub()
+        except Exception as err:
+            Architecture.save_repo_load_error = True
+            print(f"Error connecting to the save repo {err} - persistence now disabled")
+
+
     @classmethod
     def load_architectures(cls, force_reload: bool = False) -> None:
         """