JUNGU commited on
Commit
96b9255
·
verified ·
1 Parent(s): c1f61c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -45,10 +45,20 @@ SAMPLE_DATA_FILES = [
45
  {"name": "출석일수와 성적", "file": "attendance.xlsx"}
46
  ]
47
 
 
 
 
 
 
 
 
 
 
 
 
48
  # 데이터 로드
49
  @st.cache_data
50
  def load_data(file):
51
- file_path = os.path.join("sample_data", file_name)
52
  file_extension = file.name.split('.')[-1].lower()
53
  if file_extension == 'csv':
54
  data = pd.read_csv(file)
 
45
  {"name": "출석일수와 성적", "file": "attendance.xlsx"}
46
  ]
47
 
48
+ def load_sample_data(file_name):
49
+ # 예시 데이터 파일 경로
50
+ file_path = os.path.join("sample_data", file_name)
51
+ if file_name.endswith('.csv'):
52
+ return pd.read_csv(file_path)
53
+ elif file_name.endswith(('.xls', '.xlsx')):
54
+ return pd.read_excel(file_path)
55
+ else:
56
+ st.error("지원되지 않는 파일 형식입니다.")
57
+ return None
58
+
59
  # 데이터 로드
60
  @st.cache_data
61
  def load_data(file):
 
62
  file_extension = file.name.split('.')[-1].lower()
63
  if file_extension == 'csv':
64
  data = pd.read_csv(file)