Update app.py

app.py

@@ -180,29 +180,72 @@ def perform_analysis():
     else:
         st.warning("분석할 데이터가 없습니다. 데이터를 먼저 로드하고 전처리해주세요.")
 
+
+# state 유지하도록 추가 
+
+def update_filtered_data():
+    st.session_state.filtered_data = apply_slicers(st.session_state.processed_data)
+
+def create_slicers(data):
+    for col in st.session_state.categorical_columns:
+        if data[col].nunique() <= 10:
+            st.session_state.slicers[col] = st.multiselect(
+                f"{col} 선택", 
+                options=sorted(data[col].unique()), 
+                default=sorted(data[col].unique()),
+                key=f"slicer_{col}",
+                on_change=update_filtered_data
+            )
+
+def apply_slicers(data):
+    filtered_data = data.copy()
+    for col, selected_values in st.session_state.slicers.items():
+        if selected_values:
+            filtered_data = filtered_data[filtered_data[col].isin(selected_values)]
+    return filtered_data
+
+def perform_analysis():
+    if 'filtered_data' not in st.session_state or st.session_state.filtered_data is None:
+        st.session_state.filtered_data = st.session_state.processed_data.copy()
+    
+    st.header("탐색적 데이터 분석")
+    
+    # 슬라이서 생성
+    create_slicers(st.session_state.processed_data)
+    
+    # 요약 통계
+    st.write("요약 통계:")
+    st.write(st.session_state.filtered_data.describe())
+
+    # 상관관계 히트맵
+    st.subheader("상관관계 히트맵")
+    plot_correlation_heatmap(st.session_state.filtered_data)
+    
+    # 사용자가 선택한 두 변수에 대한 산점도 및 회귀 분석
+    st.subheader("두 변수 간의 관계 분석")
+    x_var = st.selectbox("X축 변수 선택", options=st.session_state.numeric_columns, key='x_var')
+    y_var = st.selectbox("Y축 변수 선택", options=[col for col in st.session_state.numeric_columns if col != x_var], key='y_var')
+
+    if x_var and y_var:
+        plot_scatter_with_regression(st.session_state.filtered_data, x_var, y_var)
+
 def main():
     st.title("인터랙티브 EDA 툴킷")
     
     manage_session_state()
 
-    if st.session_state.data is None:
-        data_input_method = st.radio("데이터 입력 방법 선택:", ("파일 업로드", "수동 입력"), key="data_input_method")
-        
-        if data_input_method == "파일 업로드":
-            uploaded_file = st.file_uploader("CSV, XLS, 또는 XLSX 파일을 선택하세요", type=["csv", "xls", "xlsx"], key="file_uploader")
-            if uploaded_file is not None:
-                st.session_state.data = load_data(uploaded_file)
-        else:
-            st.session_state.data = manual_data_entry()
-    
+    if 'data' not in st.session_state or st.session_state.data is None:
+        # ... (데이터 로드 부분)
+
     if st.session_state.data is not None:
         st.subheader("데이터 미리보기 및 수정")
         st.write("데이터를 확인하고 필요한 경우 수정하세요:")
         edited_data = st.data_editor(st.session_state.data, num_rows="dynamic", key="data_editor")
         
-        if st.button("데이터 분석 시작", key="start_analysis") or st.session_state.analysis_performed:
-            if not st.session_state.analysis_performed:
+        if st.button("데이터 분석 시작", key="start_analysis") or ('analysis_performed' in st.session_state and st.session_state.analysis_performed):
+            if 'analysis_performed' not in st.session_state or not st.session_state.analysis_performed:
                 st.session_state.processed_data = preprocess_data(edited_data)
+                st.session_state.analysis_performed = True
             perform_analysis()
 
 if __name__ == "__main__":