login_route
This commit is contained in:
parent
7ccc919e03
commit
e52091f206
25
main.py
25
main.py
|
@ -38,7 +38,8 @@ conn = psycopg2.connect(
|
|||
|
||||
cur = conn.cursor()
|
||||
|
||||
dm = None
|
||||
dms = {}
|
||||
users = {}
|
||||
|
||||
def vre_fetch():
|
||||
while True:
|
||||
|
@ -59,14 +60,27 @@ def user_interest_decay():
|
|||
def health():
|
||||
return "Success", 200
|
||||
|
||||
@app.route("/api/dm", methods=['GET'])
|
||||
@app.route("/api/dm", methods=['POST'])
|
||||
def init_dm():
|
||||
dm = DM()
|
||||
return "Sucess", 200
|
||||
username = request.get_json().get("username")
|
||||
token = '2c1e8f88-461c-42c0-8cc1-b7660771c9a3-843339462'
|
||||
i = 1
|
||||
while username in dms:
|
||||
username = username + str(i)
|
||||
i+=1
|
||||
dms[username] = DM()
|
||||
user = User(username, token)
|
||||
users[username] = user
|
||||
threading.Thread(target=user_interest_decay, name='decayinterest_'+username).start()
|
||||
message = {"assignedname": "your assigned name is " + username}
|
||||
return message
|
||||
|
||||
@app.route("/api/predict", methods=['POST'])
|
||||
def predict():
|
||||
text = request.get_json().get("message")
|
||||
username = request.get_json().get("username")
|
||||
dm = dms[username]
|
||||
user = users[username]
|
||||
message = {}
|
||||
if text == "<HELP_ON_START>":
|
||||
state = {'help': True, 'inactive': False, 'modified_query':"", 'intent':""}
|
||||
|
@ -155,12 +169,9 @@ if __name__ == "__main__":
|
|||
vre.init()
|
||||
index = vre.get_index()
|
||||
db = vre.get_db()
|
||||
user = User("ahmed", token)
|
||||
|
||||
threading.Thread(target=vre_fetch, name='updatevre').start()
|
||||
|
||||
threading.Thread(target=user_interest_decay, name='decayinterest').start()
|
||||
|
||||
rec = Recommender(retriever)
|
||||
|
||||
rg = ResponseGenerator(index,db, rec, generators, retriever)
|
||||
|
|
Loading…
Reference in New Issue