Update app.py

app.py

@@ -1,28 +1,47 @@
 import streamlit as st
 st.set_page_config(layout="wide")
 import streamlit_authenticator as stauth
-import pandas as pd
-import numpy as np
+from uuid import uuid4
 import model_comparison as MCOMP
 import model_loading as MLOAD
 import model_inferencing as MINFER
 import user_evaluation_variables
+from pathlib import Path
 import tab_manager
 import yaml
 from yaml.loader import SafeLoader
 from PIL import Image
+from huggingface_hub import CommitScheduler
+
 AUTHENTICATOR = None
 TBYB_LOGO = Image.open('./assets/TBYB_logo_light.png')
 USER_LOGGED_IN = False
 USER_DATABASE_PATH = './data/user_database.yaml'
+
+
+USER_DATABASE_DIR = Path("user_database")
+USER_DATABASE_DIR.mkdir(parents=True, exist_ok=True)
+
+USER_DATABASE_PATH = USER_DATABASE_DIR / f"tbyb-users-{uuid4()}.json"
+
+USER_DATABASE_UPDATE_SCHEDULER = CommitScheduler(
+    repo_id="try-before-you-bias-data",
+    repo_type="dataset",
+    folder_path=USER_DATABASE_DIR,
+    path_in_repo="data",
+    every=2,
+)
 def create_new_user(authenticator, users):
     try:
         if authenticator.register_user('Register user', preauthorization=False):
             st.success('User registered successfully')
     except Exception as e:
         st.error(e)
-    with open(USER_DATABASE_PATH, 'w') as file:
-        yaml.dump(users, file, default_flow_style=False)
+    with USER_DATABASE_UPDATE_SCHEDULER.lock:
+        with USER_DATABASE_PATH.open('w') as file:
+            yaml.dump(users, file, default_flow_style=False)
+
+
 def forgot_password(authenticator, users):
     try:
         username_of_forgotten_password, email_of_forgotten_password, new_random_password = authenticator.forgot_password(
@@ -57,7 +76,7 @@ def user_login_create():
     global TBYB_LOGO
     global USER_LOGGED_IN
     users = None
-    with open(USER_DATABASE_PATH) as file:
+    with open(userDataFile) as file:
         users = yaml.load(file, Loader=SafeLoader)
         AUTHENTICATOR = stauth.Authenticate(
             users['credentials'],