Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import pandas as pd
 from datasets import load_dataset, get_dataset_split_names
-from huggingface_hub import HfApi, HfFolder
+from huggingface_hub import HfApi
 import os
 import pathlib
 import uuid
@@ -33,7 +33,10 @@ def get_user_datasets(username: str):
     if not username:
         return gr.Dropdown.update(choices=[], value=None, interactive=False)
     try:
-        datasets = hf_api.list_datasets(author=username, cardData=True)
+        # --- THIS IS THE FIX ---
+        # Replace deprecated 'cardData=True' with 'full=True'
+        datasets = hf_api.list_datasets(author=username, full=True)
+        
         dataset_ids = [d.id for d in datasets if not d.private]
         return gr.Dropdown.update(choices=sorted(dataset_ids), value=None, interactive=True)
     except Exception as e:
@@ -61,7 +64,7 @@ def get_split_columns(dataset_id: str, split: str):
         first_row = next(iter(dataset_sample))
         columns = list(first_row.keys())
         # Heuristically find the best text column
-        preferred_cols = ['text', 'content', 'instruction', 'question', 'document']
+        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)
     except Exception as e:
@@ -130,8 +133,6 @@ def generate_atlas(
     mount_path = f"/{uuid.uuid4().hex}"
     atlas_app = make_server(atlas_dataset, static_path=static_path, duckdb_uri="wasm")
     
-    # --- THIS IS THE FIX ---
-    # Call mount_gradio_app on the Blocks instance `app`
     app.mount_gradio_app(atlas_app, path=mount_path)
 
     progress(1.0, desc="Done!")
@@ -199,6 +200,4 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Embedding Atlas Explorer") as app:
     app.load(fn=get_user_datasets, inputs=[hf_user_input], outputs=[dataset_input])
 
 if __name__ == "__main__":
-    # To run locally, you might need to log in to Hugging Face Hub
-    # HfFolder.save_token("YOUR_HF_TOKEN")
     app.launch()