Update app.py

app.py

@@ -39,30 +39,29 @@ 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=df['context'][index])
-    txt_question = st.text_area(height=100, label='Your question:', value=df['question'][index])
-    txt_answer = st.text_area(height=100, label='Your answer:', value=df['answer'][index])
+    print(f"Length: {len(st.session_state.df)}")
+    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)
         st.markdown(highlighted_context, unsafe_allow_html=True)
 
+
     if btn_next:
         if index < len(df) - 1:
             st.session_state.idx += 1
-            st.rerun()
+            # st.rerun()
 
     if btn_save:
         st.session_state.df['context'][index] = txt_context
         st.session_state.df['question'][index] = txt_question
         st.session_state.df['answer'][index] = txt_answer
-
-        df = st.session_state.df.copy()
-
-        csv_file = convert_df(df=df)
+        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 > 0:
             st.session_state.idx -= 1
-            st.rerun()
+            # st.rerun()