add auto_save and input limit (#1)

- add auto save and input limit (507564e2311ab6cf14438a84d7b4e50a65409428)
- fix bugs on auto save (a6f3675b7adf940b88b965a489c11f5393d700b2)
- fix bugs on auto save (7f4fb866dd92fc76859713846c331cce8e7d881c)
- Fix bugs (cf0da09860bba6c99300151c6b1c2233691091c6)
- delete unnecessary code andswitch off local mode (b7dc23fbebcc87f7599d5af409399e56c28b5f75)
- launch app to public (755d8263f5b6b9509a9c6244205713d1af9f9df9)
- minor fix (ec8338e2ff3058ebe2fbac74cc7147ae342e17b2)
- Merge branch 'main' into pr/1 (fef4c918b59c3f58d450d2a49564bdb234e65bba)
- Solve environment conflict and merge with new UI layout (1d1684354c83e9faa97c13ffe8e73e97309e7ff0)

app.py

@@ -2,8 +2,10 @@ import gradio as gr
 from utils import *
 from save_data import add_new_data, 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
 
+
 class SessionManager:
     def __init__(self):
         self.sessions = []
@@ -58,19 +60,26 @@ def handle_create_sequential(task, human_input, session_manager, api_key, identi
     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')
+    if word_limit_validation(human_input):
+        output = word_limit_validation(human_input)
+    else:
+        output = merge_texts_sequential(task, human_input, api_key)
+    session_manager.update(session_index, output, 'ai_modificated_output')
     return output, session_index
 
 
 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)
+    if word_limit_validation(human_input):
+        ai_initial_answer = word_limit_validation(human_input)
+        final_answer = word_limit_validation(human_input)
+    else:
+        ai_initial_answer = generate_ai_initial_answer(task, api_key)
+        final_answer = merge_texts_parallel(task, human_input, ai_initial_answer, api_key)
     session_manager.update(session_index, human_input, 'human_initial_answer')
     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
@@ -87,7 +96,10 @@ def handle_create_reverse_sequential(task, session_manager, api_key, identificat
 def handle_modify_reverse_sequential(session_index, modification_suggestions, session_manager, api_key):
     session = session_manager.get_session(session_index)
     session_manager.update(session_index, modification_suggestions, 'human_modifications')
-    final_answer = modify_with_suggestion(session['task'], modification_suggestions, api_key)
+    if word_limit_validation(modification_suggestions):
+        final_answer = word_limit_validation(modification_suggestions)
+    else:
+        final_answer = modify_with_suggestion(session['task'], modification_suggestions, api_key)
     session_manager.update(session_index, final_answer, 'final_answer')
     return final_answer, session_index
 
@@ -119,6 +131,14 @@ def login(identification_code):
     else:
         return update_content(None)
 
+def word_limit_validation(human_input):
+    words = human_input.split()
+    if len(words) < 50:
+        return f"Error: Your input '{human_input}'is less than 50 words. Please enter at least 50 words."
+    return None
+
+def on_textbox_change(session_index, session_manager, service, SHEET_ID):
+    return save_data(session_index, session_manager, service, SHEET_ID)
 
 if __name__ == "__main__":
     api_key = get_api_key(local=False)
@@ -166,7 +186,7 @@ if __name__ == "__main__":
         
         with group_a_content:
             with gr.Row():
-                human_input = gr.Textbox(label="Human Input")
+                human_input = gr.Textbox(label="Human Input (At least 50 words)")
             with gr.Row():
                 submit_btn = gr.Button("Create")
             with gr.Row():
@@ -175,10 +195,15 @@ if __name__ == "__main__":
 
             submit_btn.click(
                 fn=lambda task, human_input, id: handle_create_sequential(task, human_input, session_manager, api_key, id),
-                inputs=[task, human_input, identification_code],
+                inputs=[description, human_input, identification_code],
                 outputs=[ai_output, session_index]
             )
 
+            ai_output.change(
+                fn = lambda session_index: on_textbox_change(session_index, session_manager, service, SHEET_ID1),
+                inputs = [session_index]
+            )
+
             # evaluate same for every group
             evaluate_btn = gr.Button("Evaluate")
             evaluation_result = gr.Textbox(label="Evaluation Result")
@@ -189,9 +214,15 @@ if __name__ == "__main__":
                 outputs=[evaluation_result]
             )
 
-            save_btn = gr.Button("Save Data")
+            evaluation_result.change(
+                fn = lambda session_index: on_textbox_change(session_index, session_manager, service, SHEET_ID1),
+                inputs = [session_index]
+            )
+
+            save_btn = gr.Button("Save Data", elem_id="save_btn")
             save_result = gr.Label()
 
+
             save_btn.click(
                 fn=lambda session_index: save_data(session_index, session_manager, service, SHEET_ID1),
                 inputs=[session_index],
@@ -203,7 +234,7 @@ if __name__ == "__main__":
 
         with group_b_content:
             with gr.Row():
-                human_input = gr.Textbox(label="Human Input")
+                human_input = gr.Textbox(label="Human Input(At least 50 words)")
             with gr.Row():
                 create_btn = gr.Button("Create")
             with gr.Row(): 
@@ -213,10 +244,15 @@ if __name__ == "__main__":
 
             create_btn.click(
                 fn=lambda task, human_input, id: handle_create_parallel(task, human_input, session_manager, api_key, id),
-                inputs=[task, human_input, identification_code],
+                inputs=[description, human_input, identification_code],
                 outputs=[ai_initial_output, final_output, session_index]
             )
 
+            ai_initial_output.change(
+                fn = lambda session_index: on_textbox_change(session_index, session_manager, service, SHEET_ID2),
+                inputs = [session_index]
+            )
+
 
             evaluate_btn = gr.Button("Evaluate")
             evaluation_result = gr.Textbox(label="Evaluation Result")
@@ -227,7 +263,12 @@ if __name__ == "__main__":
                 outputs=[evaluation_result]
             )
 
-            save_btn = gr.Button("Save Data")
+            evaluation_result.change(
+                fn = lambda session_index: on_textbox_change(session_index, session_manager, service, SHEET_ID2),
+                inputs = [session_index]
+            )
+
+            save_btn = gr.Button("Save Data",elem_id="save_btn")
             save_result = gr.Label()
 
             save_btn.click(
@@ -244,7 +285,7 @@ if __name__ == "__main__":
                 create_initial_btn = gr.Button("Create")
             with gr.Row():
                 initial_answer = gr.Textbox(label="AI Initial Answer")
-                modification_suggestions = gr.Textbox(label="Modification Suggestions")
+                modification_suggestions = gr.Textbox(label="Modification Suggestions(At least 50 words)")
             with gr.Row():
                 create_final_btn = gr.Button("Modify")
             with gr.Row():
@@ -254,16 +295,26 @@ if __name__ == "__main__":
 
             create_initial_btn.click(
                 fn=lambda task, id: handle_create_reverse_sequential(task, session_manager, api_key, id),
-                inputs=[task, identification_code],
+                inputs=[description, identification_code],
                 outputs=[initial_answer, session_index]
             )
 
+            initial_answer.change(
+                fn = lambda session_index: on_textbox_change(session_index, session_manager, service, SHEET_ID3),
+                inputs = [session_index]
+            )
+
             create_final_btn.click(
                 fn=lambda session_index, modification_suggestions: handle_modify_reverse_sequential(session_index, modification_suggestions, session_manager, api_key),
                 inputs=[session_index, modification_suggestions],
                 outputs=[final_answer, session_index]
             )
 
+            final_answer.change(
+                fn = lambda session_index: on_textbox_change(session_index, session_manager, service, SHEET_ID3),
+                inputs = [session_index]
+            )
+
             evaluate_btn = gr.Button("Evaluate")
             evaluation_result = gr.Textbox(label="Evaluation Result")
 
@@ -273,7 +324,12 @@ if __name__ == "__main__":
                 outputs=[evaluation_result]
             )
 
-            save_btn = gr.Button("Save Data")
+            evaluation_result.change(
+                fn = lambda session_index: on_textbox_change(session_index, session_manager, service, SHEET_ID3),
+                inputs = [session_index]
+            )
+
+            save_btn = gr.Button("Save Data", elem_id="save_btn")
             save_result = gr.Label()
 
             save_btn.click(

save_data.py

@@ -10,7 +10,7 @@ def load_envs(local=False):
     """Load the environment variables."""
     if local:
         from dotenv import load_dotenv
-        load_dotenv()
+        load_dotenv(override=True)
     service_account_info = json.loads(os.environ['GOOGLE_APPLICATION_CREDENTIALS_JSON'])
     SHEET_ID1 = os.environ['SHEET_ID1'] # human-ai-sequential
     SHEET_ID2 = os.environ['SHEET_ID2'] # human-ai-parallel
@@ -40,6 +40,14 @@ def col_letter(col_num):
         letter = chr(65 + remainder) + letter
     return letter
 
+def col_letter(col_num):
+    """Convert a column number to its corresponding Excel-style letter."""
+    letter = ''
+    while col_num > 0:
+        col_num, remainder = divmod(col_num - 1, 26)
+        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. """
@@ -50,23 +58,49 @@ def add_new_data(new_row, service, SPREADSHEET_ID, num_of_columns = 5):
         spreadsheetId=SPREADSHEET_ID,
         range=range_to_read
     ).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]
-    range_to_write = f'Sheet1!A{number_of_rows + 1}'
+    if not match_found:
+        range_to_write = f'Sheet1!A{number_of_rows + 1}'
 
     request_body = {
         'values': new_row
     }
-    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}")
+    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}")