Update app.py

不用user有不同的info；用set和get的方式存储和修改user info

app.py

@@ -47,28 +47,91 @@ predefined_prompts = [
     prompt_4
 ]
 
-user_response = ["pred def" for _ in range(6)]
+_user_response = ["pred def" for _ in range(6)]
+
+# count_i = 1
+count_i_dict = dict()
+
+def init_user_info(uid):
+    global count_i_dict
+    count_i_dict[uid] = dict()
+    count_i_dict[uid]["count_i"] = 1
+    count_i_dict[uid]["user_response"] = _user_response
+
+
+def get_count_i(uid):
+    global count_i_dict
+    if uid not count_i_dict:
+        return 1
+    return count_i_dict[uid]["count_i"]
+
+def set_count_i(uid, delta=1):
+    global count_i_dict
+    if uid not in count_i_dict:
+        # TODO: call user prof init
+        return
+
+    count_i_dict[uid]["count_i"] += delta
+
+
+def get_user_response(uid, i=0):
+    global count_i_dict
+
+    if uid not in count_i_dict:
+        return ""
+
+    return count_i_dict[uid]["user_response"][i]
+
+def set_user_response(uid, i, current_p):
+    global count_i_dict
+    if uid not in count_i_dict:
+        # TODO: call user prof init
+        return
+
+    count_i_dict[uid]["user_response"][i] = current_p
+
+    
 
-count_i = 1
 
 def get_info(p, qid, uid):
-    global predefined_prompts
-    global user_response
-    global count_i
+    global predefined_prompts    
+    # global user_response
+    # global count_i
+    global count_i_dict
+
+    if uid not in count_i_dict:
+        init_user_info(uid)
+        logger.info(f"Init info for user {uid}")
+        
+    # current_count_i = count_i_dict[uid]["count_i"]
+    current_count_i = get_count_i(uid)
+    # count_i_dict[uid]["user_response"][current_count_i-1] = p
+    set_user_response(uid, current_count_i-1, p)
+
+    # user_response[current_count_i-1] = p
+    # user_response[count_i-1] = p
 
-    user_response[count_i-1] = p
     
-    next_prompt = predefined_prompts[count_i]
-    count_i += 1
+    next_prompt = predefined_prompts[current_count_i]
+    set_count_i(uid, 1)
+    
+    # count_i += 1
+    # count_i_dict[uid]["count_i"] += 1
+    
     logger.info(f"In get info loop: get from user: {p}")
-    logger.info(f"In get info loop: Our next prompt: {next_prompt}, count_i={count_i}")
+    logger.info(f"In get info loop: Our next prompt: {next_prompt}, uid={uid}, count_i={get_count_i(uid)}")
     return ["text", next_prompt]
     
 
 
 def chat(p, qid, uid):
-    global user_response
-    global count_i
+    # global user_response
+    # global count_i
+
+    global count_i_dict
+
+    count_i = get_count_i(uid)
+    # count_i = count_i_dict[uid]["count_i"]
     
     if count_i <= 4:
         ret = get_info(p, qid, uid)
@@ -77,7 +140,8 @@ def chat(p, qid, uid):
 
 
     if count_i == 5:  # 5
-        user_response[4] = p
+        # user_response[4] = p
+        set_user_response(uid, count_i-1, p)
         prompt_0_response = user_response[0]
         prompt_1_response = user_response[1]
         prompt_2_response = user_response[2]
@@ -101,7 +165,10 @@ def chat(p, qid, uid):
     logger.info(f"history: {msgs}")
     logger.info(f"p: {p}")
     logger.info(f"response: {response}")
-    count_i += 1
+    # count_i_dict[uid]["count_i"] += 1
+    set_count_i(uid,1)
+
+    
     return ["text", response]