update saving to each iteration

app.py

@@ -266,8 +266,23 @@ def generate_image(participant, scenario, prompt, active_tab, like_image, dislik
             img = infer(refined_prompts[i])
             gallery_images.append(img)
             yield gallery_images
+
+def save_response_to_sheet(participant, method, scenario, active_tab, round, like_image, dislike_image):
+    global responses_memory
+    gc = gspread.service_account(filename='credentials.json')
+    sheet = gc.open("DiverseGen-phase3").sheet1 
+    
+    entry = responses_memory[participant][method][round]
+    print(entry)
+    sheet.append_row([participant, scenario, f"{active_tab}, {method}", round, entry["prompt"], entry["sim_radio"], entry["response"], entry["satisfied_img"], entry["unsatisfied_img"]])
     
-def redesign(participant, scenario, prompt, sim_radio, like_radio, dislike_radio, current_images, history_images, active_tab):
+    # save images in google drive
+    creds = Credentials.from_service_account_file('credentials.json',scopes=scopes)
+    save_image(creds, like_image, dislike_image, f"{participant}_{scenario}_{active_tab}_{method}_round{round}")
+
+    display_info_message("✅ Your answer is saved!")
+
+def redesign(participant, scenario, prompt, sim_radio, like_radio, dislike_radio, current_images, history_images, active_tab, like_image, dislike_image):
     global counter1, counter2, responses_memory, current_task1, current_task2, enable_submit1, enable_submit2
     method = current_task1 if active_tab == "Task A" else current_task2
 
@@ -282,6 +297,8 @@ def redesign(participant, scenario, prompt, sim_radio, like_radio, dislike_radio
         responses_memory[participant][method][counter]["satisfied_img"] = f"round {counter}, {like_radio}"
         responses_memory[participant][method][counter]["unsatisfied_img"] = f"round {counter}, {dislike_radio}"
         
+        save_response_to_sheet(participant, method, scenario, active_tab, counter, like_image, dislike_image)
+
         enable_submit = True if sim_radio in ["Satisfied", "Very Satisfied"] or enable_submit else False
 
         history_prompts = [[v["prompt"]] for v in responses_memory[participant][method].values()]
@@ -337,18 +354,7 @@ def save_response(participant, scenario, prompt, sim_radio, like_radio, dislike_
         responses_memory[participant][method][counter]["unsatisfied_img"] = f"round {counter}, {dislike_radio}"
         
         try:
-            # save responses in google sheet
-            gc = gspread.service_account(filename='credentials.json')
-            sheet = gc.open("DiverseGen-phase3").sheet1 
-            
-            for i, entry in responses_memory[participant][method].items():
-                sheet.append_row([participant, scenario, f"{active_tab}, {method}", i, entry["prompt"], entry["sim_radio"], entry["response"], entry["satisfied_img"], entry["unsatisfied_img"]])
-            
-            # save images in google drive
-            creds = Credentials.from_service_account_file('credentials.json',scopes=scopes)
-            save_image(creds, like_image, dislike_image, f"{participant}_{scenario}_{active_tab}_{method}")
-
-            display_info_message("✅ Your answer is saved!")
+            save_response_to_sheet(participant, method, scenario, active_tab, counter, like_image, dislike_image)
 
             # reset global variables
             if method == METHODS[0]:
@@ -579,12 +585,12 @@ with gr.Blocks(theme=gr.themes.Soft(font=[gr.themes.GoogleFont("Inconsolata"), "
 
     redesign_btn1.click(
         fn=redesign, 
-        inputs=[participant, scenario, prompt1, sim_radio1, like_radio1, dislike_radio1, images_method1, history_images1, active_tab], 
+        inputs=[participant, scenario, prompt1, sim_radio1, like_radio1, dislike_radio1, images_method1, history_images1, active_tab, like_image1, dislike_image1], 
         outputs=[sim_radio1, dislike_radio1, like_radio1, images_method1, history_images1, example1.dataset, prompt1, next_btn1, redesign_btn1, submit_btn1]
     )
     redesign_btn2.click(
         fn=redesign, 
-        inputs=[participant, scenario, prompt2, sim_radio2, like_radio2, dislike_radio2, images_method2, history_images2, active_tab], 
+        inputs=[participant, scenario, prompt2, sim_radio2, like_radio2, dislike_radio2, images_method2, history_images2, active_tab, like_image2, dislike_image2], 
         outputs=[sim_radio2, dislike_radio2, like_radio2, images_method2, history_images2, example2.dataset, prompt2, next_btn2, redesign_btn2, submit_btn2]
     )
     submit_btn1.click(fn=save_response,