Update app.py

app.py

@@ -79,19 +79,23 @@ def push_to_hf_hub(csv_path):
         print(f"Error pushing to HF: {e}")
 
 # Function to save user choice
-def save_choice(selected_answer, not_enough_info):
+def save_choice():
     sample = dataset.iloc[st.session_state.index]
-    st.session_state.results.append({
-        "username": st.session_state.username,
-        "id": sample["id"],
-        "video_id": sample["video_id"],
-        "answer1": sample["answer1"],
-        "answer2": sample["answer2"],
-        "selected_answer": selected_answer,
-        "target": sample["target"],
-        "not_enough_info": not_enough_info
-    })
-    st.session_state.index += 1
+    selected_answer = st.session_state.get("selected_answer", None)
+    not_enough_info = st.session_state.get("not_enough_info", False)
+    
+    if selected_answer is not None:
+        st.session_state.results.append({
+            "username": st.session_state.username,
+            "id": sample["id"],
+            "video_id": sample["video_id"],
+            "answer1": sample["answer1"],
+            "answer2": sample["answer2"],
+            "selected_answer": selected_answer,
+            "target": sample["target"],
+            "not_enough_info": not_enough_info
+        })
+        st.session_state.index += 1
 
     # Save results and push to Hugging Face Hub if all samples are labeled
     if st.session_state.index >= len(dataset):
@@ -136,6 +140,7 @@ with st.form("annotation_form"):
     )
 
     # Independent checkbox for insufficient information
-    not_enough_info = st.checkbox("The frame does not provide enough information to answer the question.", key="not_enough
-::contentReference[oaicite:12]{index=12}
- 
+    not_enough_info = st.checkbox("The frame does not provide enough information to answer the question.", key="not_enough_info")
+
+    # Submit button
+    submit_button = st.form_submit_button("Next", on_click=save_choice)