Fix KeyError: 0 :))))

app.py

@@ -25,9 +25,9 @@ if 'df' not in st.session_state:
     st.session_state.df = df
 
 if 'idx' not in st.session_state:
-    st.session_state.idx = 1
+    st.session_state.idx = 0
 
-st.markdown(f"<h3 style='text-align: center;'>Sample {st.session_state.idx}/{len(df)}</h3>", unsafe_allow_html=True)
+st.markdown(f"<h3 style='text-align: center;'>Sample {st.session_state.idx + 1}/{len(df)}</h3>", unsafe_allow_html=True)
 
 col_1, col_2, col_3, col_4, col_5, col_6, col_7, col_8, col_9, col_10 = st.columns([1, 1, 1, 1, 1, 1, 1, 1, 1, 1])
 
@@ -39,9 +39,9 @@ btn_save = col_3.button(label=':heavy_check_mark: Save change', use_container_wi
 if len(df) != 0:
     index = st.session_state.idx
 
-    txt_context = st.text_area(height=300, label='Your context:', value=st.session_state.df.loc[index - 1, 'context'])
-    txt_question = st.text_area(height=100, label='Your question:', value=st.session_state.df.loc[index - 1, 'question'])
-    txt_answer = st.text_area(height=100, label='Your answer:', value=st.session_state.df.loc[index - 1, 'answer'])
+    txt_context = st.text_area(height=300, label='Your context:', value=st.session_state.df['context'][index])
+    txt_question = st.text_area(height=100, label='Your question:', value=st.session_state.df['question'][index])
+    txt_answer = st.text_area(height=100, label='Your answer:', value=st.session_state.df['answer'][index])
 
     if txt_answer.strip() and txt_context.strip():
         highlighted_context = re.sub(re.escape(txt_answer), "<mark>" + txt_answer + "</mark>", txt_context, flags=re.IGNORECASE)
@@ -49,18 +49,18 @@ if len(df) != 0:
 
 
     if btn_next:
-        if index - 1 < len(df) - 1:
+        if index < len(df) - 1:
             st.session_state.idx += 1
             st.rerun()
 
     if btn_save:
-        st.session_state.df.loc[st.session_state.idx - 1, 'context'] = txt_context
-        st.session_state.df.loc[st.session_state.idx - 1, 'question'] = txt_question
-        st.session_state.df.loc[st.session_state.idx - 1, 'answer'] = txt_answer
+        st.session_state.df['context'][index] = txt_context
+        st.session_state.df['question'][index] = txt_question
+        st.session_state.df['answer'][index] = txt_answer
         csv_file = convert_df(df=st.session_state.df)
         btn_download = col_4.download_button(data=csv_file, label=':arrow_down_small: Download file', use_container_width=True, file_name="large_df.csv", mime="text/csv")
 
     if btn_previous:
-        if index - 1 > 0:
+        if index > 0:
             st.session_state.idx -= 1
             st.rerun()