Add Download button

app.py

@@ -1,7 +1,6 @@
 import pandas as pd
 import streamlit as st
 import re
-import os
 
 st.set_page_config(page_icon='🍃', page_title='MRC for Legal Document Dataset checker', layout='wide', initial_sidebar_state="collapsed")
 
@@ -14,14 +13,6 @@ if 'idx' not in st.session_state:
 
 st.markdown(f"<h4 style='text-align: center;'>Sample {st.session_state.idx + 1}/{len(df)}</h4>", unsafe_allow_html=True)
 
-root_folder = './'
-
-# Lấy danh sách các thư mục và tập tin trong thư mục gốc
-contents = os.listdir(root_folder)
-
-# Hiển thị tiêu đề
-st.title("Danh sách thư mục và tập tin trong thư mục gốc")
-
 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])
 
 btn_previous = col_1.button(label=':arrow_backward: Previous sample', use_container_width=True)
@@ -55,5 +46,5 @@ if len(df) != 0:
         df['context'][st.session_state.idx] = txt_context
         df['question'][st.session_state.idx] = txt_question
         df['answer'][st.session_state.idx] = txt_answer
-
+        btn_download = col_4.download_button(data=df, label=':arrow_down_small: Download file', use_container_width=True, file_name="checked.csv", mime="text/csv")
         df.to_csv(path_or_buf='./Legal_AbstractiveA.csv', index=None)