add auto_save and input limit

app.py

@@ -3,6 +3,8 @@ 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):
@@ -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
 
@@ -109,12 +121,24 @@ def save_data(session_index, session_manager, service, SHEET_ID):
     return "Data has been saved to Google Sheets."
 
 def login(identification_code):
-    user_info = user_data.get(identification_code)
-    if user_info:
-        return update_content(user_info["group"])
+    groups = ["A", "B", "C"]
+    if not identification_code:
+        return update_content(None)
+    
+    user_group_id = int(identification_code)//1000 
+    if user_group_id in range(3):
+        return update_content(groups[user_group_id])
     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)
@@ -130,17 +154,22 @@ if __name__ == "__main__":
         login_status = gr.Textbox(label="Next Tasks", interactive=False)
         group = gr.State()
 
-        task = gr.Textbox(label="Task Description", 
-                          value = default_task_discription(), 
-                          visible=False,
-                          lines=10)
+        with gr.Column(visible=False) as task:
+            description = gr.Textbox(label="Task Description", 
+                            value = default_task_discription(), 
+                            lines=10)
         
+            with gr.Accordion(label = "17 Sustainable Development Goals", 
+                              open=False):
+                gr.Markdown(SDG_DETAILS)
         # initialization of different group contents
         group_a_content = gr.Group(visible=False)
         group_b_content = gr.Group(visible=False)
         group_c_content = gr.Group(visible=False)
 
         
+        
+        
         def update_content(group):
             if group == "A":
                 return gr.update(visible=True), gr.update(visible=True), gr.update(visible=False), gr.update(visible=False), group_a_instructions()
@@ -157,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():
@@ -166,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")
@@ -180,18 +214,27 @@ 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],
                 outputs=[save_result]
             )
+            
+            with gr.Accordion(label="Appendix: AI instructions", open=False):
+                gr.Markdown(GPT_PROMPT_sequential)
 
         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(): 
@@ -201,11 +244,14 @@ 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")
@@ -217,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(
@@ -226,12 +277,15 @@ if __name__ == "__main__":
                 outputs=[save_result]
             )
 
+            with gr.Accordion(label="Appendix: AI instructions", open=False):
+                gr.Markdown(GPT_PROMPT_parallel)
+
         with group_c_content:
             with gr.Row():
                 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():
@@ -241,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")
 
@@ -260,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(
@@ -268,6 +337,8 @@ if __name__ == "__main__":
                 inputs=[session_index],
                 outputs=[save_result]
             )
-        
+
+            with gr.Accordion(label="Appendix: AI instructions", open=False):
+                gr.Markdown(GPT_PROMPT_reverse_sequential)
 
         app.launch(share=True)

constants.py

@@ -0,0 +1,53 @@
+SDG_DETAILS = """
+1. No Poverty: End poverty in all its forms everywhere.
+2. Zero Hunger: End hunger, achieve food security and improved nutrition, and promote sustainable agriculture.
+3. Good Health and Well-being: Ensure healthy lives and promote well-being for all at all ages.
+4. Quality Education: Ensure inclusive and equitable quality education and promote lifelong learning opportunities for all.
+5. Gender Equality: Achieve gender equality and empower all women and girls.
+6. Clean Water and Sanitation: Ensure availability and sustainable management of water and sanitation for all.
+7. Affordable and Clean Energy: Ensure access to affordable, reliable, sustainable and modern energy for all.
+8. Decent Work and Economic Growth: Promote sustained, inclusive and sustainable economic growth, full and productive employment and decent work for all.
+9. Industry, Innovation and Infrastructure: Build resilient infrastructure, promote inclusive and sustainable industrialization and foster innovation.
+10. Reduced Inequality: Reduce inequality within and among countries.
+11. Sustainable Cities and Communities: Make cities and human settlements inclusive, safe, resilient and sustainable.
+12. Responsible Consumption and Production: Ensure sustainable consumption and production patterns.
+13. Climate Action: Take urgent action to combat climate change and its impacts.
+14. Life Below Water: Conserve and sustainably use the oceans, seas and marine resources for sustainable development.
+15. Life on Land: Protect, restore and promote sustainable use of terrestrial ecosystems, sustainably manage forests, combat desertification, and halt and reverse land degradation and halt biodiversity loss.
+16. Peace and Justice Strong Institutions: Promote peaceful and inclusive societies for sustainable development, provide access to justice for all and build effective, accountable and inclusive institutions at all levels.
+17. Partnerships to achieve the Goal: Strengthen the means of implementation and revitalize the Global Partnership for Sustainable Development.
+"""
+
+GPT_PROMPT_parallel="""
+For AI generted/modified parts, our base model is GPT-3.5-turbo. 
+
+For direct generation, we use instructions like following:
+"Given the task as: [task_description], provide an answer: "
+
+For the merging, we use instructions like following:
+"Given the task as: [task_description], there are two answers provided:
+The first answer: [human_text]
+The second answer: [ai_text]
+Merge the two answers into one in a coherent way: "
+"""
+
+GPT_PROMPT_sequential="""
+For AI generted/modified parts, our base model is GPT-3.5-turbo. 
+
+For the modification, we use instructions like following:
+"Given the task as :[task_description, the human answer is: [human_text]
+Provide an answer of your own but make sure it is coherent and should be based on the human answer: "
+"""
+
+GPT_PROMPT_reverse_sequential="""
+For AI generted/modified parts, our base model is GPT-3.5-turbo. 
+
+For direct generation, we use instructions like following:
+"Given the task as: [task_description], provide an answer: "
+
+For the merging, we use instructions like following:
+"Given the task as: [task_description], there are two answers provided:
+The first answer: [human_text]
+The second answer: [ai_text]
+Merge the two answers into one in a coherent way: "
+"""

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}")