check directory

app.py

@@ -1,6 +1,7 @@
 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")
 
@@ -13,6 +14,14 @@ 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 = './main/'
+
+# 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)