login_route

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)