Update app.py

app.py

@@ -411,7 +411,8 @@ with gr.Blocks() as demo:
 
     # Update all components when a dataset is selected
     def update_all_components(current_dataset_name, datasets):
-        time.sleep(1)  # Add a small delay
+        while current_dataset_name not in datasets:
+            time.sleep(0.1)  # Wait until dataset is loaded
         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 +426,8 @@ 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.5)  # Add a small delay
+        while current_dataset_name not in datasets:
+            time.sleep(0.1)  # Wait until dataset is loaded
         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 +441,8 @@ 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.5)  # Add a small delay
+        while current_dataset_name not in datasets:
+            time.sleep(0.1)  # Wait until dataset is loaded
         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 +456,8 @@ 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
+        while current_dataset_name not in datasets:
+            time.sleep(0.1)  # Wait until dataset is loaded
         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)]