Update app.py

app.py

@@ -525,49 +525,55 @@ def transform_cols(df: pd.DataFrame, cols: List[str], method="log"):
 # -----------------------------
 
 def update_column_lists(df):
-    """Update all column dropdowns and checkboxes based on current dataframe"""
-    if df is None or df.empty:
-        return (
-            gr.CheckboxGroup.update(choices=[]),  # available_cols
-            gr.Dropdown.update(choices=[]),       # rename_col_old
-            gr.Dropdown.update(choices=[]),       # dtype_col
-            gr.CheckboxGroup.update(choices=[]),  # num_impute_cols
-            gr.CheckboxGroup.update(choices=[]),  # cat_impute_cols
-            gr.Dropdown.update(choices=[]),       # replace_col
-            gr.Dropdown.update(choices=[]),       # split_col
-            gr.Dropdown.update(choices=[]),       # date_col_selected
-            gr.Dropdown.update(choices=[]),       # pivot_index
-            gr.CheckboxGroup.update(choices=[]),  # text_ops_cols
-            gr.Dropdown.update(choices=[]),       # groupby_cols
-            gr.Dropdown.update(choices=[]),       # agg_columns
-            gr.Dropdown.update(choices=[]),       # pivot_columns
-            gr.Dropdown.update(choices=[]),       # pivot_values
-            gr.Dropdown.update(choices=[]),       # join_key
-        )
-    
-    cols = df.columns.tolist()
-    numeric_cols = df.select_dtypes(include=[np.number]).columns.tolist()
-    categorical_cols = df.select_dtypes(include=["object", "category"]).columns.tolist()
-    datetime_cols = df.select_dtypes(include=["datetime64"]).columns.tolist()
-    
+    """
+    Return a tuple of gr.update(...) values that correspond exactly to the
+    prepare_widget_outputs order:
+    [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]
+    """
+    # safe column list getter - ensure strings, flatten MultiIndex if present
+    def get_columns_safe(df):
+        if df is None:
+            return []
+        cols = list(df.columns)
+        # handle MultiIndex columns
+        if hasattr(df.columns, "levels") and hasattr(df.columns, "names"):
+            try:
+                cols = ['_'.join(map(str, c)) if isinstance(c, (list, tuple)) else str(c) for c in cols]
+            except Exception:
+                cols = [str(c) for c in cols]
+        else:
+            cols = [str(c) for c in cols]
+        return cols
+
+    cols = get_columns_safe(df)
+    single_value = cols[0] if cols else None
+    multi_value = cols if cols else []
+
+    # Choose reasonable defaults for UI elements:
+    # - For components that expect a list/multiselect: value=multi_value
+    # - For single-select dropdowns: value=single_value
+    # NOTE: adjust ordering if your outputs wiring order differs.
     return (
-        gr.CheckboxGroup.update(choices=cols),  # available_cols
-        gr.Dropdown.update(choices=cols),       # rename_col_old
-        gr.Dropdown.update(choices=cols),       # dtype_col
-        gr.CheckboxGroup.update(choices=numeric_cols),  # num_impute_cols
-        gr.CheckboxGroup.update(choices=categorical_cols),  # cat_impute_cols
-        gr.Dropdown.update(choices=cols),       # replace_col
-        gr.Dropdown.update(choices=categorical_cols),  # split_col
-        gr.Dropdown.update(choices=cols),       # date_col_selected
-        gr.Dropdown.update(choices=cols),       # pivot_index
-        gr.CheckboxGroup.update(choices=categorical_cols),  # text_ops_cols
-        gr.Dropdown.update(choices=cols),       # groupby_cols
-        gr.Dropdown.update(choices=numeric_cols),  # agg_columns
-        gr.Dropdown.update(choices=categorical_cols),  # pivot_columns
-        gr.Dropdown.update(choices=numeric_cols),  # pivot_values
-        gr.Dropdown.update(choices=cols),       # join_key
+        gr.update(choices=cols, value=multi_value),   # available_cols (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=single_value),  # rename_col_old (Dropdown)
+        gr.update(choices=cols, value=single_value),  # dtype_col (Dropdown for selecting column to change dtype)
+        gr.update(choices=cols, value=multi_value),   # num_impute_cols (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=multi_value),   # cat_impute_cols (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=multi_value),   # replace_col (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=multi_value),   # split_col (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=single_value),  # date_col_selected (Dropdown)
+        gr.update(choices=cols, value=single_value),  # pivot_index (Dropdown)
+        gr.update(choices=cols, value=multi_value),   # text_ops_cols (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=multi_value),   # groupby_cols (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=multi_value),   # agg_columns (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=multi_value),   # pivot_columns (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=multi_value),   # pivot_values (CheckboxGroup / Multiselect)
+        gr.update(choices=cols, value=single_value)   # join_key (Dropdown)
     )
 
+
 def apply_column_operations(df, selected_cols, rename_mapping, dtype_conversions):
     """Apply column selection, renaming, and type conversions"""
     df = df.copy()