Update app.py

app.py

@@ -85,8 +85,8 @@ def generate_atlas(
     model_name: str,
     umap_neighbors: int,
     umap_min_dist: float,
-    request: gr.Request,  # <<< STEP 1: The Request object comes AFTER your main inputs.
-    progress=gr.Progress(track_tqdm=True) # <<< STEP 2: The Progress object comes LAST as a default argument.
+    request: gr.Request,
+    progress=gr.Progress(track_tqdm=True)
 ):
     """
     Loads data, computes embeddings, and serves the Embedding Atlas UI.
@@ -136,12 +136,14 @@ def generate_atlas(
     mount_path = f"/{uuid.uuid4().hex}"
     atlas_app = make_server(atlas_dataset, static_path=static_path, duckdb_uri="wasm")
     
-    # --- This now uses the correct request object to mount the app ---
     logging.info(f"Mounting FastAPI app at path: {mount_path}")
     request.app.mount(mount_path, atlas_app)
 
     progress(1.0, desc="Done!")
-    iframe_html = f"<iframe src='{mount_path}' width='100%' height='800px' frameborder='0'></iframe>"
+    
+    # --- THE FINAL FIX: Add a trailing slash to the iframe src ---
+    iframe_html = f"<iframe src='{mount_path}/' width='100%' height='800px' frameborder='0'></iframe>"
+    
     return gr.HTML(iframe_html)
 
 # --- Gradio UI Definition ---
@@ -177,7 +179,6 @@ with gr.Blocks(theme=gr.themes.Soft(), title="Embedding Atlas Explorer") as app:
     split_input.change(fn=get_split_columns, inputs=[dataset_input, split_input], outputs=text_column_input)
     
     # --- Button Click Event ---
-    # The 'inputs' list does NOT include 'request' or 'progress'. Gradio handles them automatically.
     generate_button.click(
         fn=generate_atlas,
         inputs=[dataset_input, split_input, text_column_input, sample_size_input, model_input, umap_neighbors_input, umap_min_dist_input],
@@ -187,4 +188,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__":
-    app.launch(debug=True)
+    app.launch(debug=True)```