Update app.py

app.py

@@ -67,13 +67,11 @@ def get_split_columns(dataset_id: str, split: str):
     if not dataset_id or not split:
         return gr.update(choices=[], value=None, interactive=False)
     try:
-        # This is the most robust method: stream one row and get its keys.
         dataset_sample = load_dataset(dataset_id, split=split, streaming=True)
         first_row = next(iter(dataset_sample))
         columns = list(first_row.keys())
         logging.info(f"Found columns: {columns}")
         
-        # 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)
         logging.info(f"Best default column chosen: {best_col}")
@@ -144,7 +142,10 @@ def generate_atlas(
     mount_path = f"/{uuid.uuid4().hex}"
     atlas_app = make_server(atlas_dataset, static_path=static_path, duckdb_uri="wasm")
     
-    app.mount_gradio_app(atlas_app, path=mount_path)
+    # --- THIS IS THE FINAL FIX ---
+    # Call mount_gradio_app from the gradio module 'gr'
+    # and pass the 'app' Blocks instance as the first argument.
+    gr.mount_gradio_app(app, atlas_app, path=mount_path)
 
     progress(1.0, desc="Done!")
     iframe_html = f"<iframe src='{mount_path}' width='100%' height='800px' frameborder='0'></iframe>"
@@ -177,12 +178,12 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Embedding Atlas Explorer") as app:
             gr.Markdown("### 3. Explore Atlas")
             output_html = gr.HTML("<div style='display:flex; justify-content:center; align-items:center; height:800px; border: 1px solid #ddd; border-radius: 5px;'><p>Atlas will be displayed here after generation.</p></div>")
 
-    # --- Chained Event Listeners for Dynamic UI (CORRECTED LOGIC) ---
+    # --- Chained Event Listeners for Dynamic UI (Corrected Logic) ---
     hf_user_input.submit(fn=get_user_datasets, inputs=hf_user_input, outputs=dataset_input)
     
     dataset_input.change(fn=get_dataset_splits, inputs=dataset_input, outputs=split_input)
     
-    # This is the critical fix: The columns are populated only AFTER a split is chosen.
+    # The columns are populated only AFTER a split is chosen.
     split_input.change(fn=get_split_columns, inputs=[dataset_input, split_input], outputs=text_column_input)
     
     # --- Button Click Event ---