Update app.py

app.py

@@ -78,6 +78,48 @@ def create_component_to_add_target_func(selected_files, dfs, i):
 #         st.text(content)
 #         st.write(f1(3))
 
+def set_filtered_data(df,i):
+    action = "data_filter"
+    col_to_filter = st.selectbox("Select the field to Filter on ", df.columns.values,
+                                 key= action + "_col_filter_" + str(i))
+    filter_operation = st.selectbox("Operation ",
+                                    ['Greater Than', 'Equals', 'Less Than', "In", "In Between"],
+                                    key=action + "_col_filter_op_" + str(i))
+    selected_filter_vals = None
+    
+    if filter_operation:
+        if filter_operation == 'In':
+            selected_filter_vals = st.multiselect("Select Values to Filter on ", df[col_to_filter].unique(),
+                                                  key=action + "_col_filter_val_" + str(i))
+            if selected_filter_vals:
+                filtered_df = df[df[col_to_filter].isin(selected_filter_vals)]
+        elif filter_operation == 'Equals':
+            selected_filter_vals = st.text_input("Enter a numeric value",
+                                                 key=action + "_col_filter_val_" + str(i))
+            if selected_filter_vals:
+                filtered_df = df[df[col_to_filter] == selected_filter_vals]
+        elif filter_operation == 'Greater Than':
+            selected_filter_vals = st.text_input("Enter a numeric value",
+                                                 key=action + "_col_filter_val_" + str(i))
+            if selected_filter_vals:
+                filtered_df = df[df[col_to_filter] > selected_filter_vals]
+        elif filter_operation == 'Less Than':
+            selected_filter_vals = st.text_input("Enter a numeric value",
+                                                 key=action + "_col_filter_val_" + str(i))
+            if selected_filter_vals:
+                filtered_df = df[df[col_to_filter] < selected_filter_vals]
+        elif filter_operation == 'In Between':
+            selected_filter_vals = st.select_slider("Select range",
+                                                    (df[col_to_filter].min(), df[col_to_filter].max()),
+                                                    key=action + "_col_filter_val_" + str(i))
+            if selected_filter_vals:
+                filtered_df = df[df[col_to_filter] < selected_filter_vals]
+        
+        if selected_filter_vals:
+            set_filtered_data_session_object(filtered_df,selected_files[i])
+#                 st.write(df.shape)
+#                 st.write( st.session_state['filtered_data'][selected_files[i]].shape)
+
 
 def create_component_for_analysis_for_single_df(selected_files, dfs, i):
     st.subheader(selected_files[i])
@@ -86,46 +128,7 @@ def create_component_for_analysis_for_single_df(selected_files, dfs, i):
     filter_data = st.checkbox("Analyse on Filtered Data",key="filter_data_check"+str(i))
     
     if filter_data:
-        action = "data_filter"
-        col_to_filter = st.selectbox("Select the field to Filter on ", df.columns.values,
-                                     key= action + "_col_filter_" + str(i))
-        filter_operation = st.selectbox("Operation ",
-                                        ['Greater Than', 'Equals', 'Less Than', "In", "In Between"],
-                                        key=action + "_col_filter_op_" + str(i))
-        selected_filter_vals = None
-        
-        if filter_operation:
-            if filter_operation == 'In':
-                selected_filter_vals = st.multiselect("Select Values to Filter on ", df[col_to_filter].unique(),
-                                                      key=action + "_col_filter_val_" + str(i))
-                if selected_filter_vals:
-                    filtered_df = df[df[col_to_filter].isin(selected_filter_vals)]
-            elif filter_operation == 'Equals':
-                selected_filter_vals = st.text_input("Enter a numeric value",
-                                                     key=action + "_col_filter_val_" + str(i))
-                if selected_filter_vals:
-                    filtered_df = df[df[col_to_filter] == selected_filter_vals]
-            elif filter_operation == 'Greater Than':
-                selected_filter_vals = st.text_input("Enter a numeric value",
-                                                     key=action + "_col_filter_val_" + str(i))
-                if selected_filter_vals:
-                    filtered_df = df[df[col_to_filter] > selected_filter_vals]
-            elif filter_operation == 'Less Than':
-                selected_filter_vals = st.text_input("Enter a numeric value",
-                                                     key=action + "_col_filter_val_" + str(i))
-                if selected_filter_vals:
-                    filtered_df = df[df[col_to_filter] < selected_filter_vals]
-            elif filter_operation == 'In Between':
-                selected_filter_vals = st.select_slider("Select range",
-                                                        (df[col_to_filter].min(), df[col_to_filter].max()),
-                                                        key=action + "_col_filter_val_" + str(i))
-                if selected_filter_vals:
-                    filtered_df = df[df[col_to_filter] < selected_filter_vals]
-            
-            if selected_filter_vals:
-                set_filtered_data_session_object(filtered_df,selected_files[i])
-#                 st.write(df.shape)
-#                 st.write( st.session_state['filtered_data'][selected_files[i]].shape)
+        set_filtered_data()
 
     analysis_actions = st.multiselect("What analysis do you wish to do?",
                                       ['Summary of Data', 'Sample Data','Get Profile' ,'Univariate Analysis',