Update src/streamlit_app.py

src/streamlit_app.py

@@ -999,6 +999,7 @@ def view_topic_page():
                 # Update the user's progress to the next comment they should see.
                 update_user_progress(user_id, topic_id, next_comment_id)
 
+                st.session_state._voting_in_progress = False
                 st.rerun() # Rerun to update UI
 
             except Exception as e:
@@ -1007,23 +1008,26 @@ def view_topic_page():
                 if local_con:
                     local_con.close()
 
-        st.session_state._voting_in_progress = False
+        st.session_state._voting_in_progress = st.session_state.get("_voting_in_progress", False)
         col1, col2, col3, col4 = st.columns(4)
         user_id = find_or_create_user(user_email) # Ensure user exists
 
         col1.markdown("*Personally I...*")
         if col2.button("Agree", disabled=st.session_state.get('_voting_in_progress', False)):
             # Set a flag immediately to disable buttons until next render
-            st.session_state._voting_in_progress = True
-            handle_vote("agree", current_comment_id, topic_id, user_id)
+            if st.session_state._voting_in_progress == False:
+                st.session_state._voting_in_progress = True
+                handle_vote("agree", current_comment_id, topic_id, user_id)
         if col3.button("Neutral", disabled=st.session_state.get('_voting_in_progress', False)):
             # Set a flag immediately to disable buttons until next render
-            st.session_state._voting_in_progress = True
-            handle_vote("neutral", current_comment_id, topic_id, user_id)
+            if st.session_state._voting_in_progress == False:
+                st.session_state._voting_in_progress = True
+                handle_vote("neutral", current_comment_id, topic_id, user_id)
         if col4.button("Disagree", disabled=st.session_state.get('_voting_in_progress', False)):
             # Set a flag immediately to disable buttons until next render
-            st.session_state._voting_in_progress = True
-            handle_vote("disagree", current_comment_id, topic_id, user_id)
+            if st.session_state._voting_in_progress == False:
+                st.session_state._voting_in_progress = True
+                handle_vote("disagree", current_comment_id, topic_id, user_id)
 
     else:
         st.info("No more comments to vote on in this topic." if "No more comments" in current_comment_content else current_comment_content)