Continue fix KeyError: 0 :)))))

app.py

@@ -39,15 +39,14 @@ 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['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])
+    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])
 
     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
@@ -57,7 +56,10 @@ if len(df) != 0:
         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)
+
+        df = st.session_state.df.copy()
+
+        csv_file = convert_df(df=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: