add a submit all button to finish wandb run

app.py

@@ -97,7 +97,7 @@ if uploaded_file:
             print(f'Correct label for image {i}: {correct_label}')
 
             # Add a button to confirm feedback and log it
-            if st.button(f"Submit Feedback for Image {i}", key=f'submit-{i}'):
+            if st.button(f"Add feedback for Image {i}", key=f'add-{i}'):
                 feedback_table.add_data(
                     wandb.Image(image),
                     uploaded_file.type,
@@ -106,9 +106,9 @@ if uploaded_file:
                     correct_label,
                     label2id[correct_label],
                 )
-                st.success(f"Feedback for Image {i} submitted!")
 
-    print(feedback_table)
-    run = st.session_state.wandb_run
-    run.log({'feedback_table': feedback_table})
-    run.finish()
+    if st.button("Submit all feedback?", key=f'submit'):
+        run = st.session_state.wandb_run
+        run.log({'feedback_table': feedback_table})
+        run.finish()
+        st.success(f"Feedback for Image {i} submitted!")