Update app.py

app.py

@@ -204,17 +204,34 @@ def main():
             st.session_state.data = manual_data_entry()
 
     if st.session_state.data is not None:
-        st.subheader("데이터 미리보기 및 수정")
-        st.write("데이터를 확인하고 필요한 경우 수정하세요. 열 이름을 직접 클릭하여 수정할 수 있습니다:")
+        st.subheader("열 이름 수정")
+        st.write("열 이름을 확인하고 필요한 경우 수정하세요:")
+        
+        # 열 이름 편집을 위한 데이터프레임 생성
+        column_names = pd.DataFrame({'현재 열 이름': st.session_state.data.columns})
+        edited_column_names = st.data_editor(
+            column_names,
+            num_rows="fixed",
+            key="column_name_editor",
+            column_config={
+                "현재 열 이름": st.column_config.TextColumn(
+                    "열 이름",
+                    help="새로운 열 이름을 입력하세요",
+                    max_chars=50
+                )
+            }
+        )
         
-        # 열 이름을 수정 가능하게 설정
-        column_config = {col: st.column_config.TextColumn(col, max_chars=50) for col in st.session_state.data.columns}
+        # 수정된 열 이름 적용
+        st.session_state.data.columns = edited_column_names['현재 열 이름']
+
+        st.subheader("데이터 미리보기 및 수정")
+        st.write("데이터를 확인하고 필요한 경우 수정하세요:")
         
         edited_data = st.data_editor(
             st.session_state.data,
             num_rows="dynamic",
-            column_config=column_config,
-            key="data_editor"
+            key="main_data_editor"  # 여기서 키를 변경했습니다
         )
         
         if st.button("데이터 분석 시작", key="start_analysis") or st.session_state.analysis_performed:
@@ -224,5 +241,4 @@ def main():
             perform_analysis()
 
 if __name__ == "__main__":
-    main()
     main()