Update app.py

app.py

@@ -203,19 +203,19 @@ def main():
         else:
             st.session_state.data = manual_data_entry()
 
-    if st.session_state.data is not None:
+     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")
+        # 열 이름을 수정 가능하게 설정
+        column_config = {col: st.column_config.TextColumn(col, max_chars=50) for col in st.session_state.data.columns}
         
-        # 데이터 편집
-        edited_data = st.data_editor(st.session_state.data, num_rows="dynamic", key="data_editor")
-        
-        # 열 이름 업데이트
-        edited_data.columns = edited_columns['column_name']
+        edited_data = st.data_editor(
+            st.session_state.data,
+            num_rows="dynamic",
+            column_config=column_config,
+            key="data_editor"
+        )
         
         if st.button("데이터 분석 시작", key="start_analysis") or st.session_state.analysis_performed:
             if not st.session_state.analysis_performed: