Update app.py

app.py

@@ -67,12 +67,18 @@ def load_question_generator():
 # CSV 데이터 로드 함수
 @st.cache_data
 #def load_data(data_file='/train.csv'):
-def load_data(data_file='/processed_mathqa2.csv'):
+def load_data(data_file='/processed_mathqa2.csv', selected_indexes=None):
     try:
         file_path = os.path.join(data_path, data_file.lstrip('/'))
         df = pd.read_csv(file_path)
         logger.info(f"Data loaded successfully from {file_path}")
+
+        if selected_indexes is not None:
+            df = df.loc[selected_indexes]  # Filter rows by index
+            logger.info(f"Data filtered to selected indexes: {selected_indexes}")
+        
         return df
+        
     except FileNotFoundError:
         st.error(f"파일을 찾을 수 없습니다: {data_file}")
         logger.error(f"File not found: {data_file}")
@@ -80,7 +86,10 @@ def load_data(data_file='/processed_mathqa2.csv'):
 
 def start_quiz():
     """퀴즈 시작 및 초기화"""
-    df = load_data()
+
+    selected_indexes = [6400, 24318, 18158, 6715, 21794, 7026, 3520, 7259, 7295, 7401]
+
+    df = load_data(selected_indexes=selected_indexes)
     if df is None or df.empty:
         st.error("데이터를 불러올 수 없습니다. 데이터셋을 확인해주세요.")
         return