Update users_management.py

users_management.py

@@ -1,5 +1,7 @@
 import json
 import os
+from hashlib import sha256
+
 
 def save_user_data_to_json(username, file_path='users.json'):
     user = users[username]
@@ -46,7 +48,7 @@ def load_from_json(file_path='users.json'):
         return {}
 
 
-def add_user_pref(username, input_type, input_value, users):
+def add_user_pref(username, input_value, users, input_type='keywords'):
     # Ensure the user exists
     user = users[username]
 
@@ -64,12 +66,13 @@ def add_user_pref(username, input_type, input_value, users):
             user['history'][input_type].append(word)
 
 
-def update_json(user, values, type='keywords'):
-    # users = load_from_json()
+def update_json(user, prompt, keywords):
+    print(f"TYPE PROMPT = {type(prompt)}")
     username = user['name']
 
     if username != 'Guest':
-        add_user_pref(username, type, values, users)
+        add_user_pref(username, [prompt], users,input_type='prompts')
+        add_user_pref(username, keywords, users,input_type='keywords')
         save_user_data_to_json(username)
 
         with open('users.json', 'r') as file:
@@ -83,11 +86,9 @@ def auth_user(username, password):
     else:
         username = 'Guest'
         user = users.get(username)
-    return user, f"## Hi {username}!", gr.update(choices=user['history']['keywords']) , gr.update(choices=user['history']['prompts'])
+    return user, f"## Hi {username}!", gr.update(choices=user['history']['prompts'], gr.update(choices=user['history']['keywords']))
 
 def logout():
     username = 'Guest'
     user = users.get(username)
-    return user, f"## Hi {username}!", gr.update(choices=user['history']['keywords']) , gr.update(choices=user['history']['prompts'])
-
-
+    return user, f"## Hi {username}!", gr.update(choices=user['history']['prompts'], gr.update(choices=user['history']['keywords']))