Update app.py

app.py

@@ -5,7 +5,7 @@ import re
 st.set_page_config(layout='wide')
 
 def load_data():
-    return pd.read_csv(filepath_or_buffer='./Legal_AbstractiveA.csv')
+    return pd.read_csv(filepath_or_buffer='./data.csv')
 
 df = load_data()
 
@@ -30,17 +30,17 @@ if btn_next:
 
 st.markdown(f"<h3 style='text-align: center;'>Sample: {st.session_state.idx+1}/{len(df)}</h3>", unsafe_allow_html=True)
 
-context = st.text_area(label='Your context: ', value=df['context'][st.session_state.idx], height=300)
-question = st.text_area(label='Your question: ', value=df['question'][st.session_state.idx], height=100)
-answer = st.text_area(label='Your answer: ', value=df['answer'][st.session_state.idx], height=100)
+context = st.text_area(label='Your context: ', value=df['contexts'][st.session_state.idx], height=300)
+question = st.text_area(label='Your question: ', value=df['questions'][st.session_state.idx], height=100)
+answer = st.text_area(label='Your answer: ', value=df['answers'][st.session_state.idx], height=100)
 
 if answer.strip() and context.strip():
     highlighted_context = re.sub(re.escape(answer), "<mark>" + answer + "</mark>", context, flags=re.IGNORECASE)
     st.markdown(highlighted_context, unsafe_allow_html=True)
 
 if btn_save:
-    df.loc[st.session_state.idx, 'context'] = context
-    df.loc[st.session_state.idx, 'question'] = question
-    df.loc[st.session_state.idx, 'answer'] = answer
+    df.loc[st.session_state.idx, 'contexts'] = context
+    df.loc[st.session_state.idx, 'questions'] = question
+    df.loc[st.session_state.idx, 'answers'] = answer
     
-    df.to_csv('./Legal_AbstractiveA.csv', index=False)
+    df.to_csv('./data.csv', index=False)