fix messed up navigation

app.py

@@ -7,6 +7,8 @@ import streamlit as st
 
 from huggingface_hub import HfFileSystem
 
+COMMENT_ID_COL = 'COMMENT ID'
+
 
 # Function to get user ID from URL
 def get_user_id_from_url():
@@ -32,12 +34,22 @@ hf_fs = HfFileSystem(token=HF_TOKEN)
 def save_data(data):
     user_file_path = f"datasets/ijundi/mod-gen-eval-test/{data['user_id']}"
     hf_fs.mkdir(user_file_path)
-    with hf_fs.open(f"{user_file_path}/{data['COMMENT ID']}.json", "w") as f:
+    with hf_fs.open(f"{user_file_path}/{data[COMMENT_ID_COL]}.json", "w") as f:
         f.write(json.dumps(data))
 
 
 #################################### Streamlit App ####################################
 
+# Function to navigate rows
+def navigate(index_change):
+    st.session_state.current_index += index_change
+    print(st.session_state.current_index)
+    # Loop around if navigating beyond the start or end
+    if st.session_state.current_index < 0:
+        st.session_state.current_index = len(st.session_state.data) - 1
+    elif st.session_state.current_index >= len(st.session_state.data):
+        st.session_state.current_index = 0
+
 
 # Title of the app
 st.title("Moderation Feedback Collection")
@@ -51,28 +63,17 @@ if 'current_index' not in st.session_state:
     st.session_state.current_index = -1
 
 
-# Function to navigate rows
-def navigate(index_change):
-    st.session_state.current_index += index_change
-    print(st.session_state.current_index)
-    # Loop around if navigating beyond the start or end
-    if st.session_state.current_index < 0:
-        st.session_state.current_index = len(st.session_state.data) - 1
-    elif st.session_state.current_index >= len(st.session_state.data):
-        st.session_state.current_index = 0
-
+st.write(f"Page {st.session_state.current_index + 1} out of {len(st.session_state.data)}")
 
-# Retrieve user ID from URL
-user_id = get_user_id_from_url()
 if st.session_state.current_index == -1:
-    st.session_state.user_id = st.text_input('Please enter your user ID to proceed', value=user_id)
-else:
-    # Display current row data
-    st.write(f"Data for row {st.session_state.current_index + 1}:")
-    st.dataframe(st.session_state.data.iloc[[st.session_state.current_index]])
+    st.session_state.user_id = st.text_input('Please enter your user ID to proceed', value=get_user_id_from_url())
 
-    # Creating the form
+elif st.session_state.current_index <= len(st.session_state.data):
+# Creating the form
     with st.form("feedback_form"):
+        # Display current row data
+        st.dataframe(st.session_state.data.iloc[[st.session_state.current_index]])
+
         st.write("Please provide your feedback:")
 
         q1_reply_comment = st.selectbox("Is the reply comment fitting and does it make sense as a reply to the preceding comment? (0-4)", options=[0, 1, 2, 3, 4])
@@ -83,14 +84,13 @@ else:
         q6_actions_clear = st.selectbox("Does the reply make it clear what actions need to be done? (0-4)", options=[0, 1, 2, 3, 4])
         other_comments = st.text_area("Other: free text?")
 
-        submitted = st.form_submit_button("Submit")
-
-        if submitted:
-            # Save the feedback
+        # https://discuss.streamlit.io/t/run-streamlit-from-pycharm/21624/10
+        # st.form_submit_button("Submit", on_click=lambda: navigate(1))
+        if st.form_submit_button("Submit"):
             data = {
-                'user_id': st.session_state,
+                'user_id': st.session_state.user_id,
                 'index': st.session_state.current_index,
-                'COMMENT ID': st.session_state.data.iloc[st.session_state.current_index]['COMMENT ID'],
+                COMMENT_ID_COL: st.session_state.data.iloc[st.session_state.current_index][COMMENT_ID_COL],
                 'q1_reply_comment': q1_reply_comment,
                 'q2_moderator_function': q2_moderator_function,
                 'q3_issues_in_comment': q3_issues_in_comment,
@@ -102,6 +102,9 @@ else:
             save_data(data)
             st.success("Feedback submitted successfully!")
 
+else:
+    st.write("Finished all data points!")
+
 # Navigation buttons
 col1, col2 = st.columns(2)
 # with col1:

requirements.txt

@@ -1,4 +1,2 @@
 huggingface_hub==0.21.4
-requests==2.31.0
-datasets==2.18.0
 streamlit==1.32.0