log user identification code as well

app.py

@@ -11,14 +11,16 @@ class SessionManager:
     def add_session(self, cooperate_style, task):
         if cooperate_style == "sequential":
             session = {
+                "user_identification_code": None,
                 "task": task,
-                "human_initial_answer": None,
                 "cooperate_style": cooperate_style,
+                "human_initial_answer": None,
                 "ai_modificated_output": None,
                 "evaluation": None
             }
         elif cooperate_style == "reverse_sequential":
             session = {
+                "user_identification_code": None,
                 "task": task,
                 "cooperate_style": cooperate_style,
                 "ai_initial_answer": None,
@@ -27,7 +29,8 @@ class SessionManager:
                 "evaluation": None
             }
         elif cooperate_style == "parallel":
-            session = {
+            session = {                
+                "user_identification_code": None,
                 "task": task,
                 "cooperate_style": cooperate_style,
                 "ai_initial_answer": None,
@@ -51,16 +54,17 @@ class SessionManager:
 
 
 
-def handle_create_sequential(task, human_input, session_manager, api_key):
+def handle_create_sequential(task, human_input, session_manager, api_key, identification_code):
     cooperate_style = "sequential"
     session_index = session_manager.add_session(task=task, cooperate_style=cooperate_style)
     session_manager.update(session_index, human_input, 'human_initial_answer')
     output = merge_texts_sequential(task, human_input, api_key)
     session_manager.update(session_index, output, 'ai_modificated_output')
+    session_manager.update(session_index, identification_code, 'user_identification_code')
     return output, session_index
 
 
-def handle_create_parallel(task, human_input, session_manager, api_key):
+def handle_create_parallel(task, human_input, session_manager, api_key, identification_code):
     cooperate_style = "parallel"
     session_index = session_manager.add_session(task=task, cooperate_style=cooperate_style)
     ai_initial_answer = generate_ai_initial_answer(task, api_key)
@@ -68,13 +72,15 @@ def handle_create_parallel(task, human_input, session_manager, api_key):
     session_manager.update(session_index, ai_initial_answer, 'ai_initial_answer')
     final_answer = merge_texts_parallel(task, human_input, ai_initial_answer, api_key)
     session_manager.update(session_index, final_answer, 'merged_final_answer')
+    session_manager.update(session_index, identification_code, 'user_identification_code')
     return ai_initial_answer, final_answer, session_index
 
-def handle_create_reverse_sequential(task, session_manager, api_key):
+def handle_create_reverse_sequential(task, session_manager, api_key, identification_code):
     cooperate_style = "reverse_sequential"
     session_index = session_manager.add_session(task=task, cooperate_style=cooperate_style)
     ai_initial_answer = generate_ai_initial_answer(task, api_key)
     session_manager.update(session_index, ai_initial_answer, 'ai_initial_answer')
+    session_manager.update(session_index, identification_code, 'user_identification_code')
     return ai_initial_answer, session_index
 
 
@@ -114,8 +120,7 @@ if __name__ == "__main__":
     api_key = get_api_key(local=False)
     service, SHEET_IDs = get_sheet_service(local=False)
     SHEET_ID1, SHEET_ID2, SHEET_ID3 = SHEET_IDs
-    # print(api_key)
-    # print(SHEET_ID1, SHEET_ID2, SHEET_ID3)
+
     session_manager = SessionManager()
 
     with gr.Blocks() as app:
@@ -157,8 +162,8 @@ if __name__ == "__main__":
                 session_index = gr.Number(label="Session Index", visible=False)
 
             submit_btn.click(
-                fn=lambda task, human_input: handle_create_sequential(task, human_input, session_manager, api_key),
-                inputs=[task, human_input],
+                fn=lambda task, human_input, id: handle_create_sequential(task, human_input, session_manager, api_key, id),
+                inputs=[task, human_input, identification_code],
                 outputs=[ai_output, session_index]
             )
 
@@ -193,8 +198,8 @@ if __name__ == "__main__":
                 session_index = gr.Number(label="Session Index", visible=False)
 
             create_btn.click(
-                fn=lambda task, human_input: handle_create_parallel(task, human_input, session_manager, api_key),
-                inputs=[task, human_input],
+                fn=lambda task, human_input, id: handle_create_parallel(task, human_input, session_manager, api_key),
+                inputs=[task, human_input, identification_code],
                 outputs=[ai_initial_output, final_output, session_index]
             )
 
@@ -235,8 +240,8 @@ if __name__ == "__main__":
             
 
             create_initial_btn.click(
-                fn=lambda task: handle_create_reverse_sequential(task, session_manager, api_key),
-                inputs=[task],
+                fn=lambda task, id: handle_create_reverse_sequential(task, session_manager, api_key, id),
+                inputs=[task, identification_code],
                 outputs=[initial_answer, session_index]
             )