Update app.py

app.py

@@ -393,7 +393,7 @@ with gr.Blocks() as demo:
     next_button.click(
         fn=lambda current_page_number, datasets, current_dataset_name: change_page("next", current_page_number, datasets, current_dataset_name),
         inputs=[current_page_number, datasets, current_dataset_name],
-        outputs=[current_page_number, dataset_viewer, message_box]
+        outputs=[current_page_number, dataset_html, message_box]
     )
 
     # Initialize dataset_selector
@@ -411,7 +411,7 @@ with gr.Blocks() as demo:
 
     # Update all components when a dataset is selected
     def update_all_components(current_dataset_name, datasets):
-        time.sleep(0.5)  # Add a small delay
+        time.sleep(0.3)  # Add a small delay
         dataset = datasets[current_dataset_name]
         html_content = display_dataset_html(dataset, page_number=0)
         entry_options = [f"{idx}: {entry['prompt'][:30]}" for idx, entry in enumerate(dataset)]
@@ -425,7 +425,7 @@ with gr.Blocks() as demo:
 
     # Update all components when an entry is added
     def update_all_components_after_add(current_dataset_name, datasets):
-        time.sleep(0.1)  # Add a small delay
+        time.sleep(0.3)  # Add a small delay
         dataset = datasets[current_dataset_name]
         html_content = display_dataset_html(dataset, page_number=0)
         entry_options = [f"{idx}: {entry['prompt'][:30]}" for idx, entry in enumerate(dataset)]
@@ -439,7 +439,7 @@ with gr.Blocks() as demo:
 
     # Update all components when an entry is edited
     def update_all_components_after_edit(current_dataset_name, datasets):
-        time.sleep(0.1)  # Add a small delay
+        time.sleep(0.3)  # Add a small delay
         dataset = datasets[current_dataset_name]
         html_content = display_dataset_html(dataset, page_number=0)
         entry_options = [f"{idx}: {entry['prompt'][:30]}" for idx, entry in enumerate(dataset)]
@@ -453,7 +453,7 @@ with gr.Blocks() as demo:
 
     # Update all components when an entry is deleted
     def update_all_components_after_delete(current_dataset_name, datasets):
-        time.sleep(0.1)  # Add a small delay
+        time.sleep(0.3)  # Add a small delay
         dataset = datasets[current_dataset_name]
         html_content = display_dataset_html(dataset, page_number=0)
         entry_options = [f"{idx}: {entry['prompt'][:30]}" for idx, entry in enumerate(dataset)]