Update app.py

app.py

@@ -1524,19 +1524,26 @@ with gr.Blocks(css=css, title="DataSynth — Analytics Hub") as demo:
             tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
             df.to_csv(tmp.name, index=False)
             # prepare choices for date and target selectors
-            #cols = df.columns.tolist()
+            prepare_updates = update_column_lists(df)
             # save df to temp csv for persistence (store path in state)
             #tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
             #df.to_csv(tmp.name, index=False)
             
-            return (md, df.head(100), tmp.name, prof, corr_plot_fig)
+            return (md, df.head(100), tmp.name, prof, corr_plot_fig, *prepare_updates)
         except Exception as e:
             return (gr.update(value=f"Error loading file: {e}"), 
                    pd.DataFrame(), None, None, None)
 
-    profile_btn.click(fn=_load_and_profile,
-                      inputs=[upload],
-                      outputs=[profile_md, sample_table, df_state, profile_state, corr_plot])
+    profile_btn.click(
+        fn=_load_and_profile,
+        inputs=[upload],
+        outputs=[
+            profile_md, sample_table, df_state, profile_state, corr_plot,
+            available_cols, rename_col_old, dtype_col, num_impute_cols,
+            cat_impute_cols, replace_col, split_col, date_col_selected, pivot_index,
+            text_ops_cols, groupby_cols, agg_columns, pivot_columns, pivot_values, join_key
+        ]
+    )
 
     #def _view_data(uploaded):
      #   try: