Update app.py

app.py

@@ -8,6 +8,7 @@ from PIL import Image
 import uuid
 import tempfile
 import numpy as np
+import time
 
 # Function to save dataset to zip
 
@@ -401,6 +402,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.1)  # 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)]
@@ -414,6 +416,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
         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)]
@@ -427,6 +430,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
         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)]
@@ -440,6 +444,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.5)  # 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)]