Update users_management.py

users_management.py

@@ -4,8 +4,8 @@ from hashlib import sha256
 import gradio as gr
 
 
-def save_user_data_to_json(username, file_path='users.json'):
-    user = users[username]
+def save_user_data_to_json(user, file_path='users.json'):
+    username = user['name']
     # Load existing data
     if os.path.exists(file_path):
         with open(file_path, 'r') as file:
@@ -16,14 +16,8 @@ def save_user_data_to_json(username, file_path='users.json'):
     else:
         existing_data = {}
 
-    # Update or add the user's data, including hashed password
-    user_data = {
-        "name": username,
-        "hashed_password": user['hashed_password'],
-        "history": user['history']
-    }
     
-    existing_data[username] = user_data
+    existing_data[username] = user
 
     # Save updated data back to file
     with open(file_path, 'w') as file:
@@ -49,35 +43,36 @@ def load_from_json(file_path='users.json'):
         return {}
 
 
-def add_user_pref(username, input_value, users, input_type='keywords'):
-    # Ensure the user exists
-    user = users[username]
-
-    if username not in users:
-        print(f"User {username} not found in {users}")
-        return
+def add_user_pref(user, input_value, input_type='keywords'):
 
-    # Initialize the history as a list of dictionaries if empty
     if not user['history']:
         user['history'] = {"keywords": [], "prompts": []}
 
-    # Add the input value to the corresponding list
     for word in input_value:
         if word not in user['history'][input_type]:
             user['history'][input_type].append(word)
+    return user
 
 
 def update_json(user, prompt, keywords):
-    print(f"TYPE PROMPT = {type(prompt)}")
     username = user['name']
+    users = load_from_json()
+    user = users.get(username)
+
+    print(f"TYPE PROMPT = {type(prompt)}")
+    
 
     if username != 'Guest':
-        add_user_pref(username, prompt, users,input_type='prompts')
-        add_user_pref(username, keywords, users,input_type='keywords')
-        save_user_data_to_json(username)
-        
-    return gr.update(choices=user['history']['prompts']), gr.update(choices=user['history']['keywords'])
+        user = add_user_pref(user, prompt, input_type='prompts')
+        user = add_user_pref(user, keywords, input_type='keywords')
+        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
 
 def auth_user(username, password):
     if username in users and sha256(password.encode()).hexdigest() == users[username]['hashed_password']: