Reset "continue" button after a new sample is loaded

app.py

@@ -83,13 +83,17 @@ def reset_submit_buttons():
     return tuple(gr.Button(value="Submit", interactive=True) for _ in range(n_models))
 
 
+def reset_continue_button():
+    return gr.Button(value=f"0/{n_models} forms submitted", interactive=False)
+
+
 def next_sample(current_sample_ind, shuffled_idx):
     if current_sample_ind == n_samples:
         return None
 
     current_sample_ind += 1
     updated_view = update_commit_view(shuffled_idx[current_sample_ind])
-    return (current_sample_ind,) + updated_view + reset_answers() + reset_submit_buttons()
+    return (current_sample_ind,) + updated_view + reset_answers() + reset_submit_buttons() + (reset_continue_button(),)
 
 
 with open("head.html") as head_file:
@@ -210,7 +214,7 @@ with gr.Blocks(theme=gr.themes.Soft(), head=head_html, css="style_overrides.css"
                     submit_buttons.append(gr.Button(value="Submit"))
 
             n_forms_submitted = gr.Number(visible=False, value=0, precision=0)
-            continue_btn = gr.Button(value=f"0/{n_models} forms submitted", interactive=False)
+            continue_btn = reset_continue_button()
 
             session_val = gr.Textbox(info='Session', interactive=False, container=True, show_label=False,
                                      label='session')
@@ -247,10 +251,10 @@ with gr.Blocks(theme=gr.themes.Soft(), head=head_html, css="style_overrides.css"
     questions_list = list(chain.from_iterable(questions))
 
     skip_btn.click(next_sample, inputs=[current_sample_sld, shuffled_idx_val],
-                   outputs=[current_sample_sld] + commit_view + questions_list + submit_buttons)
+                   outputs=[current_sample_sld] + commit_view + questions_list + submit_buttons + [continue_btn])
 
     continue_btn.click(next_sample, inputs=[current_sample_sld, shuffled_idx_val],
-                       outputs=[current_sample_sld] + commit_view + questions_list + submit_buttons)
+                       outputs=[current_sample_sld] + commit_view + questions_list + submit_buttons + [continue_btn])
 
 
     def submit_for_model(current_sample, n_forms_submitted_val, *args):