Update app.py

app.py

@@ -49,33 +49,10 @@ def load_data(file):
     else:
         st.error("지원되지 않는 파일 형식입니다. CSV, XLS, 또는 XLSX 파일을 업로드해주세요.")
         return None
-
-    # 빈 열이 있는 경우 라벨링 처리
+    
+    # 빈 열 이름에 기본값 부여
     if data.columns.isnull().any():
-        new_columns = []
-        for i, col in enumerate(data.columns):
-            if pd.isnull(col):
-                new_columns.append(f'Column_{i+1}')
-            else:
-                new_columns.append(col)
-        data.columns = new_columns
-        
-        st.warning("일부 열 이름이 비어 있어서 자동으로 이름을 부여했습니다.")
-        
-        # 사용자에게 열 이름 수정 기회 제공
-        st.write("현재 열 이름:")
-        st.write(data.columns.tolist())
-        
-        if st.checkbox("열 이름을 수정하시겠습니까?"):
-            new_column_names = st.text_input("새로운 열 이름을 쉼표로 구분하여 입력하세요:", 
-                                             value=",".join(data.columns))
-            new_column_names = [name.strip() for name in new_column_names.split(",")]
-            
-            if len(new_column_names) == len(data.columns):
-                data.columns = new_column_names
-                st.success("열 이름이 성공적으로 업데이트되었습니다.")
-            else:
-                st.error("입력한 열 이름의 개수가 현재 열 개수와 일치하지 않습니다.")
+        data.columns = [f'Column_{i+1}' if pd.isnull(col) else col for i, col in enumerate(data.columns)]
     
     return data
 
@@ -228,9 +205,18 @@ def main():
 
     if st.session_state.data is not None:
         st.subheader("데이터 미리보기 및 수정")
-        st.write("데이터를 확인하고 필요한 경우 수정하세요:")
+        st.write("데이터를 확인하고 필요한 경우 수정하세요. 열 이름도 직접 수정할 수 있습니다:")
+        
+        # 열 이름 편집을 위한 데이터 프레임 생성
+        column_df = pd.DataFrame({'column_name': st.session_state.data.columns})
+        edited_columns = st.data_editor(column_df, num_rows="dynamic", key="column_editor")
+        
+        # 데이터 편집
         edited_data = st.data_editor(st.session_state.data, num_rows="dynamic", key="data_editor")
         
+        # 열 이름 업데이트
+        edited_data.columns = edited_columns['column_name']
+        
         if st.button("데이터 분석 시작", key="start_analysis") or st.session_state.analysis_performed:
             if not st.session_state.analysis_performed:
                 st.session_state.processed_data = preprocess_data(edited_data)