Update app.py

app.py

@@ -38,9 +38,17 @@ def manage_session_state():
     if 'filtered_data' not in st.session_state:
         st.session_state.filtered_data = None
 
+
+SAMPLE_DATA_FILES = [
+    {"name": "과목별 노력과 성취도", "file": "subject.xlsx"},
+    {"name": "채점", "file": "score.xlsx"},
+    {"name": "출석일수와 성적", "file": "attendance.xlsx"}
+]
+
 # 데이터 로드
 @st.cache_data
 def load_data(file):
+    file_path = os.path.join("sample_data", file_name)
     file_extension = file.name.split('.')[-1].lower()
     if file_extension == 'csv':
         data = pd.read_csv(file)
@@ -208,12 +216,21 @@ def main():
     manage_session_state()
 
     if st.session_state.data is None:
-        data_input_method = st.radio("데이터 입력 방법 선택:", ("파일 업로드", "수동 입력"), key="data_input_method")
+        data_input_method = st.radio("데이터 입력 방법 선택:", ("파일 업로드", "예시 데이터 사용", "수동 입력"), key="data_input_method")
         
         if data_input_method == "파일 업로드":
             uploaded_file = st.file_uploader("CSV, XLS, 또는 XLSX 파일을 선택하세요", type=["csv", "xls", "xlsx"], key="file_uploader")
             if uploaded_file is not None:
                 st.session_state.data = load_data(uploaded_file)
+        elif data_input_method == "예시 데이터 사용":
+            sample_choice = st.selectbox(
+                "예시 데이터 선택",
+                options=[sample["name"] for sample in SAMPLE_DATA_FILES],
+                format_func=lambda x: x
+            )
+            if st.button("선택한 예시 데이터 로드"):
+                selected_file = next(sample["file"] for sample in SAMPLE_DATA_FILES if sample["name"] == sample_choice)
+                st.session_state.data = load_sample_data(selected_file)
         else:
             st.session_state.data = manual_data_entry()