Update app.py

app.py

@@ -19,13 +19,13 @@ login(token=os.environ.get("HF_TOKEN"), write_permission=True)
 AUTHENTICATOR = None
 TBYB_LOGO = Image.open('./assets/TBYB_logo_light.png')
 USER_LOGGED_IN = False
-USER_DATABASE_PATH = './data/user_database.yaml'
+base_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_PATH = USER_DATABASE_DIR / f"tbyb-users-{uuid4()}.yaml"
 
 USER_DATABASE_UPDATE_SCHEDULER = CommitScheduler(
     repo_id="try-before-you-bias-data",
@@ -79,6 +79,8 @@ def user_login_create():
     global TBYB_LOGO
     global USER_LOGGED_IN
     users = None
+    if not os.path.exists(USER_DATABASE_PATH):
+        USER_DATABASE_PATH=base_user_database_path
     with open(USER_DATABASE_PATH) as file:
         users = yaml.load(file, Loader=SafeLoader)
         AUTHENTICATOR = stauth.Authenticate(
@@ -88,6 +90,7 @@ def user_login_create():
             users['cookie']['expiry_days'],
             users['preauthorized']
         )
+        
     with st.sidebar:
         st.image(TBYB_LOGO, width=70)
         loginTab, registerTab, detailsTab = st.tabs(["Log in", "Register", "Account details"])