Update users_management.py

users_management.py

@@ -4,11 +4,11 @@ from hashlib import sha256
 import gradio as gr
 
 
-def save_user_data_to_json(user, file_path='users.json'):
+def save_user_data_to_json(user, conf_file_path='users.json'):
     username = user['name']
     # Load existing data
-    if os.path.exists(file_path):
-        with open(file_path, 'r') as file:
+    if os.path.exists(conf_file_path):
+        with open(conf_file_path, 'r') as file:
             try:
                 existing_data = json.load(file)
             except json.JSONDecodeError:
@@ -20,9 +20,11 @@ def save_user_data_to_json(user, file_path='users.json'):
     existing_data[username] = user
 
     # Save updated data back to file
-    with open(file_path, 'w') as file:
+    with open(conf_file_path, 'w') as file:
         json.dump(existing_data, file, indent=4)
 
+    return conf_file_path
+
 def load_from_json(file_path='users.json'):
     """
     Load user data from a JSON file and return it as a dictionary.
@@ -65,14 +67,14 @@ def update_json(user, prompt, keywords):
     if username != 'Guest':
         user = add_user_pref(user, prompt, input_type='prompts')
         user = add_user_pref(user, keywords, input_type='keywords')
-        save_user_data_to_json(user)
+        conf_file_path = save_user_data_to_json(user)
 
         # with open('users.json', 'r') as file:
         #     saved_data = json.load(file)
 
     users = load_from_json()
     user = users.get(username)
-    return gr.update(choices=user['history']['prompts']), gr.update(choices=user['history']['keywords']), user
+    return gr.update(choices=user['history']['prompts']), gr.update(choices=user['history']['keywords']), user, conf_file_path
 
 def auth_user(username, password):
     users = load_from_json()