data save position fixed to id-determined row

app.py

@@ -1,19 +1,19 @@
 import gradio as gr
 from utils import *
-from save_data import add_new_data, get_sheet_service
+from save_data import add_or_update_row_at_fixed_position, get_sheet_service
 from instructions import *
 from user_groups import user_data
-from constants import SDG_DETAILS, GPT_PROMPT_parallel, GPT_PROMPT_sequential, GPT_PROMPT_reverse_sequential
+from constants import SDG_DETAILS, WORD_LIMIT_MIN, GROUP_SEPERATION, LOCAL_PARAMS
 from html_codes import *
 
 class SessionManager:
     def __init__(self):
-        self.sessions = []
+        self.sessions = {}
 
-    def add_session(self, cooperate_style, task):
+    def add_session(self, cooperate_style, task, identification_code):
         if cooperate_style == "sequential":
             session = {
-                "user_identification_code": None,
+                "user_identification_code": identification_code,
                 "task": task,
                 "cooperate_style": cooperate_style,
                 "human_initial_answer": None,
@@ -22,7 +22,7 @@ class SessionManager:
             }
         elif cooperate_style == "reverse_sequential":
             session = {
-                "user_identification_code": None,
+                "user_identification_code": identification_code,
                 "task": task,
                 "cooperate_style": cooperate_style,
                 "ai_initial_answer": None,
@@ -32,7 +32,7 @@ class SessionManager:
             }
         elif cooperate_style == "parallel":
             session = {                
-                "user_identification_code": None,
+                "user_identification_code": identification_code,
                 "task": task,
                 "cooperate_style": cooperate_style,
                 "ai_initial_answer": None,
@@ -40,8 +40,8 @@ class SessionManager:
                 "merged_final_answer": None,
                 "evaluation": None
             }
-        self.sessions.append(session)
-        return len(self.sessions) - 1
+        self.sessions[identification_code] = session
+        return identification_code
 
     def update(self, index, output_content, key='final_output'):
         self.sessions[index][key] = output_content
@@ -51,14 +51,20 @@ class SessionManager:
 
     def save_session_to_sheet(self, index, service, SHEET_ID):
         session = self.sessions[index]
+        row_id = int(index) % GROUP_SEPERATION + 2  # user data starts from row 2
         new_row = list(session.values())
-        add_new_data(new_row, service, SHEET_ID, num_of_columns=len(new_row))  # 动态列数
+        add_or_update_row_at_fixed_position(
+            row_id = row_id, 
+            new_row = new_row, 
+            service = service, 
+            SPREADSHEET_ID = SHEET_ID, 
+            num_of_columns=len(new_row))  
 
 
 
 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_index = session_manager.add_session(task=task, cooperate_style=cooperate_style, identification_code = identification_code)
     session_manager.update(session_index, human_input, 'human_initial_answer')
     session_manager.update(session_index, identification_code, 'user_identification_code')
     if word_limit_validation(human_input):
@@ -71,7 +77,7 @@ def handle_create_sequential(task, human_input, session_manager, api_key, identi
 
 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)
+    session_index = session_manager.add_session(task=task, cooperate_style=cooperate_style, identification_code = identification_code)
     if word_limit_validation(human_input):
         ai_initial_answer = word_limit_validation(human_input)
         final_answer = word_limit_validation(human_input)
@@ -86,7 +92,7 @@ def handle_create_parallel(task, human_input, session_manager, api_key, identifi
 
 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)
+    session_index = session_manager.add_session(task=task, cooperate_style=cooperate_style, identification_code = identification_code)
     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')
@@ -134,7 +140,7 @@ def login(identification_code):
 
 def word_limit_validation(human_input):
     words = human_input.split()
-    if len(words) < 50:
+    if len(words) < WORD_LIMIT_MIN:
         return f"Error: Please enter at least 50 words."
     return None
 
@@ -151,9 +157,8 @@ def check_initial_generated(initial_answer):
     return None
 
 if __name__ == "__main__":
-    deploy_local = True
-    api_key = get_api_key(local=deploy_local)
-    service, SHEET_IDs = get_sheet_service(local=deploy_local)
+    api_key = get_api_key(local=LOCAL_PARAMS)
+    service, SHEET_IDs = get_sheet_service(local=LOCAL_PARAMS)
     SHEET_ID1, SHEET_ID2, SHEET_ID3 = SHEET_IDs
 
     session_manager = SessionManager()
@@ -220,8 +225,7 @@ if __name__ == "__main__":
             )
 
 
-            # evaluate same for every group
-            #evaluate_btn = gr.Button("Evaluate", visible=False)
+
             # Evaluate without showing 
             evaluation_result = gr.Textbox(label="Evaluation Result", visible=False, interactive = False)
             

constants.py

@@ -1,3 +1,7 @@
+WORD_LIMIT_MIN = 50
+GROUP_SEPERATION = 1000
+LOCAL_PARAMS = False
+
 SDG_DETAILS = """
 1. No Poverty: Eradicate all forms of poverty.
 2. Zero Hunger: End hunger and promote sustainable agriculture.

save_data.py

@@ -21,7 +21,7 @@ def load_envs(local=False):
 
 def get_sheet_service(local=False):
     """Get the google sheet service object."""
-    service_account_info, SHEET_ID = load_envs(local=local)
+    service_account_info, SHEET_IDs = load_envs(local=local)
     # verify the service_account_info
     credentials = service_account.Credentials.from_service_account_info(
         service_account_info,
@@ -30,15 +30,8 @@ def get_sheet_service(local=False):
 
     # build the service object
     service = build('sheets', 'v4', credentials=credentials)
-    return service, SHEET_ID
+    return service, SHEET_IDs
 
-def col_letter(col_num):
-    """Convert a column number to a column letter (1-indexed)."""
-    letter = ''
-    while col_num > 0:
-        col_num, remainder = divmod(col_num - 1, 26)
-        letter = chr(65 + remainder) + letter
-    return letter
 
 def col_letter(col_num):
     """Convert a column number to its corresponding Excel-style letter."""
@@ -48,64 +41,36 @@ def col_letter(col_num):
         letter = chr(65 + remainder) + letter
     return letter
 
-def add_new_data(new_row, service, SPREADSHEET_ID, num_of_columns = 5):
-    """Add new data to the spreadsheet.
-    new_row: list of data to be added. """
-    # read the existing data
-    range_to_read = f'Sheet1!A:{col_letter(num_of_columns)}'
-
-    result = service.spreadsheets().values().get(
+def add_or_update_row_at_fixed_position(row_id, new_row, service, SPREADSHEET_ID, num_of_columns=5):
+    """
+    Add or update data at a fixed row position in Google Sheet.
+    
+    Args:
+        row_id: int, the row number (1-based index) where the data should be written.
+        new_row: list, the new data to be added or updated.
+        service: Google Sheets API service instance.
+        SPREADSHEET_ID: str, the spreadsheet ID.
+        num_of_columns: int, the number of columns to consider (default: 5).
+    """
+    # Ensure `new_row` has exactly `num_of_columns` elements
+    new_row = new_row + [""] * (num_of_columns - len(new_row)) if len(new_row) < num_of_columns else new_row[:num_of_columns]
+    
+    # Determine the write range for the specified row
+    range_to_write = f"Sheet1!A{row_id}:{col_letter(num_of_columns)}{row_id}"
+
+    # Directly write the data to the specified row
+    response = service.spreadsheets().values().update(
         spreadsheetId=SPREADSHEET_ID,
-        range=range_to_read
+        range=range_to_write,
+        valueInputOption='RAW',
+        body={"values": [new_row]}
     ).execute()
- 
-
-    # search for first columns match
-    match_found = False
-    update_idx = None
-    for idx, row in enumerate(result.get('values', [])):
-        if row[0] == new_row[0]:
-            match_found = True
-            range_to_write = f'Sheet1!A{idx + 1}:{col_letter(num_of_columns)}{idx + 1}'
-            update_idx = idx + 1
-            break
-
-    values = result.get('values', [])
-    number_of_rows = len(values)
-    new_row = [new_row]
-    if not match_found:
-        range_to_write = f'Sheet1!A{number_of_rows + 1}'
-
-    request_body = {
-        'values': new_row
-    }
-    if match_found:
-        service.spreadsheets().values().clear(
-            spreadsheetId=SPREADSHEET_ID,
-            range=range_to_write,
-            body={},
-        ).execute()
-        response = service.spreadsheets().values().update(
-            spreadsheetId=SPREADSHEET_ID,
-            range=range_to_write,
-            valueInputOption='RAW',
-            body=request_body
-        ).execute()
-        print(f"Updated row at position {update_idx}")
-    else:
-        response = service.spreadsheets().values().append(
-            spreadsheetId=SPREADSHEET_ID,
-            range=range_to_write,
-            valueInputOption='RAW',
-            insertDataOption='INSERT_ROWS',
-            body=request_body
-        ).execute()
-        print(f"Added new row at position {number_of_rows + 1}")
 
+    print(f"Data written to row {row_id}. Response: {response}")
 
 
 if __name__ == "__main__":
-    service, SHEET_ID = get_sheet_service(local=True)
+    service, SHEET_IDs = get_sheet_service(local=True)
     new_row = ["test1", "test2", "test3", "test4", "test5"]
-    add_new_data(new_row, service, SHEET_ID)
+    add_or_update_row_at_fixed_position(12, new_row, service, SHEET_IDs[0])