Update app.py

app.py

@@ -41,7 +41,7 @@ def get_user_datasets(username: str):
         return gr.Dropdown.update(choices=sorted(dataset_ids), value=None, interactive=True)
     except Exception as e:
         gr.Warning(f"Could not fetch datasets for user '{username}'. Error: {e}")
-        return gr.Dropdown.update(choices=[], value=None, interactive=False)
+        return gr.update(choices=[], value=None, interactive=False)
 
 def get_dataset_splits(dataset_id: str):
     """Gets all available splits for a selected dataset."""
@@ -52,12 +52,12 @@ def get_dataset_splits(dataset_id: str):
         return gr.Dropdown.update(choices=splits, value=splits[0] if splits else None, interactive=True)
     except Exception as e:
         gr.Warning(f"Could not fetch splits for dataset '{dataset_id}'. Error: {e}")
-        return gr.Dropdown.update(choices=[], value=None, interactive=False)
+        return gr.update(choices=[], value=None, interactive=False)
 
 def get_split_columns(dataset_id: str, split: str):
     """Gets all columns for a selected split by loading one row."""
     if not dataset_id or not split:
-        return gr.Dropdown.update(choices=[], value=None, interactive=False)
+        return gr.update(choices=[], value=None, interactive=False)
     try:
         # Stream one row to get column names without downloading the whole dataset
         dataset_sample = load_dataset(dataset_id, split=split, streaming=True)
@@ -66,10 +66,10 @@ def get_split_columns(dataset_id: str, split: str):
         # Heuristically find the best text column
         preferred_cols = ['text', 'content', 'instruction', 'question', 'document', 'prompt']
         best_col = next((col for col in preferred_cols if col in columns), columns[0] if columns else None)
-        return gr.Dropdown.update(choices=columns, value=best_col, interactive=True)
+        return gr.update(choices=columns, value=best_col, interactive=True)
     except Exception as e:
         gr.Warning(f"Could not fetch columns for split '{split}'. Error: {e}")
-        return gr.Dropdown.update(choices=[], value=None, interactive=False)
+        return gr.update(choices=[], value=None, interactive=False)
 
 # --- Main Atlas Generation Logic ---
 def generate_atlas(