def clear_models():
if hash_string(request.form['key']) == delete_key:
for i in models.find():
os.remove('ai_helpers/models/' + i['hash'])
models.delete_many({})
return {}, 200
return {}, 401
def login():
user = users.find_one({'username': request.json['username']})
if not user:
return {}, 401
password = hash_string(request.json['password'] + user['salt'])
if password == user['password']:
token = encode_auth_token(user['_id'])
print(token)
return {"token":token}, 200
return {}, 401
def create_user():
user = {}
salt = str(gensalt())
password = hash_string(request.json['password'] + salt)
user['username'] = request.json['username']
user['password'] = password
user['salt'] = salt
if users.find_one({'username': user['username']}):
return {}, 409
token = encode_auth_token(users.insert_one(user).inserted_id)
return {"token":token}, 201
def train_datasets():
dataset = datasets.find_one({'name': request.form['dataset']})
if not dataset:
return {}, 401
epochs = None
if 'epochs' in request.form:
epochs = request.form['epochs']
else:
epochs = 10
train_dataset = TrainModel(dataset['text'], 10)
created_model = train_dataset.train_model()
created_time = time.time()
hash_name = hash_string(dataset['text'] + str(created_time))
model = {
'hash': hash_name,
'name': request.form['dataset'],
'text': dataset['text'],
'maxlen': train_dataset.maxlen,
'epochs': epochs,
'time': created_time
}
created_model.save('ai_helpers/models/' + hash_name)
info = models.insert_one(model)
details = {
'_id': str(info.inserted_id),
'hash': hash_name,
'name': request.form['dataset'],
'time': created_time
}
return details, 201
def clear_datasets():
if hash_string(request.form['key']) == delete_key:
datasets.delete_many({})
return {}, 200
return {}, 401
def clear_users():
if hash_string(request.json['key']) == DB.delete_key:
users.delete_many({})
return {}, 200
return {}, 401