Update app.py

app.py

@@ -35,6 +35,8 @@ def manage_session_state():
         st.session_state.slicers = {}
     if 'analysis_performed' not in st.session_state:
         st.session_state.analysis_performed = False
+    if 'filtered_data' not in st.session_state:
+        st.session_state.filtered_data = None
 
 # 데이터 로드
 @st.cache_data
@@ -95,14 +97,18 @@ def preprocess_data(data):
     
     return data
 
+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:  # 고유값이 10개 이하인 경우에만 슬라이서 생성
+        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}"
+                key=f"slicer_{col}",
+                on_change=update_filtered_data
             )
 
 def apply_slicers(data):
@@ -151,61 +157,9 @@ def plot_scatter_with_regression(data, x_var, y_var):
     st.write(f"상관계수: {r_value:.4f}")
     st.write(f"p-value: {p_value:.4f}")
     st.write(f"표준 오차: {std_err:.4f}")
-    
-def perform_analysis():
-    if st.session_state.processed_data is not None and not st.session_state.processed_data.empty:
-        st.header("탐색적 데이터 분석")
-        
-        # 슬라이서 생성 및 적용
-        create_slicers(st.session_state.processed_data)
-        filtered_data = apply_slicers(st.session_state.processed_data)
-        
-        # 요약 통계
-        st.write("요약 통계:")
-        st.write(filtered_data.describe())
-
-        # 상관관계 히트맵
-        st.subheader("상관관계 히트맵")
-        plot_correlation_heatmap(filtered_data)
-        
-        # 사용자가 선택한 두 변수에 대한 산점도 및 회귀 분석
-        st.subheader("두 변수 간의 관계 분석")
-        st.session_state.x_var = st.selectbox("X축 변수 선택", options=st.session_state.numeric_columns, key='x_var')
-        st.session_state.y_var = st.selectbox("Y축 변수 선택", options=[col for col in st.session_state.numeric_columns if col != st.session_state.x_var], key='y_var')
-
-        if st.session_state.x_var and st.session_state.y_var:
-            plot_scatter_with_regression(filtered_data, st.session_state.x_var, st.session_state.y_var)
-
-        st.session_state.analysis_performed = True
-    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:
+    if st.session_state.filtered_data is None:
         st.session_state.filtered_data = st.session_state.processed_data.copy()
     
     st.header("탐색적 데이터 분석")
@@ -234,16 +188,23 @@ def main():
     
     manage_session_state()
 
-    if 'data' not in st.session_state or st.session_state.data is None:
-        # ... (데이터 로드 부분)
+    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 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 ('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:
+        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)
                 st.session_state.analysis_performed = True
             perform_analysis()