YchKhan commited on
Commit
e6ab6b4
1 Parent(s): 21909a6

Delete users_management.py

Browse files
Files changed (1) hide show
  1. users_management.py +0 -98
users_management.py DELETED
@@ -1,98 +0,0 @@
1
- import json
2
- import os
3
- from hashlib import sha256
4
- #import gradio as gr
5
-
6
-
7
- def save_user_data_to_json(username, file_path='users.json'):
8
- user = users[username]
9
- # Load existing data
10
- if os.path.exists(file_path):
11
- with open(file_path, 'r') as file:
12
- try:
13
- existing_data = json.load(file)
14
- except json.JSONDecodeError:
15
- existing_data = {}
16
- else:
17
- existing_data = {}
18
-
19
- # Update or add the user's data, including hashed password
20
- user_data = {
21
- "name": username,
22
- "hashed_password": user['hashed_password'],
23
- "history": user['history']
24
- }
25
-
26
- existing_data[username] = user_data
27
-
28
- # Save updated data back to file
29
- with open(file_path, 'w') as file:
30
- json.dump(existing_data, file, indent=4)
31
-
32
- def load_from_json(file_path='users.json'):
33
- """
34
- Load user data from a JSON file and return it as a dictionary.
35
-
36
- :param file_path: The path to the JSON file from which to load user data.
37
- :return: A dictionary containing the user data loaded from the JSON file.
38
- """
39
- if os.path.exists(file_path):
40
- with open(file_path, 'r') as file:
41
- try:
42
- user_data = json.load(file)
43
- return user_data
44
- except json.JSONDecodeError as e:
45
- print(f"Error decoding JSON from {file_path}: {e}")
46
- return {}
47
- else:
48
- print(f"File {file_path} not found.")
49
- return {}
50
-
51
-
52
- def add_user_pref(username, input_value, users, input_type='keywords'):
53
- # Ensure the user exists
54
- user = users[username]
55
-
56
- if username not in users:
57
- print(f"User {username} not found in {users}")
58
- return
59
-
60
- # Initialize the history as a list of dictionaries if empty
61
- if not user['history']:
62
- user['history'] = {"keywords": [], "prompts": []}
63
-
64
- # Add the input value to the corresponding list
65
- for word in input_value:
66
- if word not in user['history'][input_type]:
67
- user['history'][input_type].append(word)
68
-
69
-
70
- def update_json(user, prompt, keywords):
71
- print(f"TYPE PROMPT = {type(prompt)}")
72
- username = user['name']
73
-
74
- if username != 'Guest':
75
- add_user_pref(username, [prompt], users,input_type='prompts')
76
- add_user_pref(username, keywords, users,input_type='keywords')
77
- save_user_data_to_json(username)
78
-
79
- with open('users.json', 'r') as file:
80
- saved_data = json.load(file)
81
-
82
-
83
- def auth_user(username, password):
84
- if username in users and sha256(password.encode()).hexdigest() == users[username]['hashed_password']:
85
- #mistral_api_key = os.environ[username]
86
- user = users.get(username)
87
- else:
88
- username = 'Guest'
89
- user = users.get(username)
90
- return user, f"## Hi {username}!", gr.update(choices=user['history']['prompts']), gr.update(choices=user['history']['keywords'])
91
-
92
- def logout():
93
- username = 'Guest'
94
- user = users.get(username)
95
- return user, f"## Hi {username}!", gr.update(choices=user['history']['prompts']), gr.update(choices=user['history']['keywords'])
96
-
97
-
98
- #users = load_from_json()