Update app.py

app.py

@@ -399,4 +399,56 @@ with gr.Blocks() as demo:
     # Hide dataset_html
     dataset_html.visible = False
 
+    # Update all components when a dataset is selected
+    def update_all_components(current_dataset_name, datasets):
+        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)]
+        return gr.update(value=html_content), gr.update(choices=entry_options)
+
+    dataset_selector.change(
+        update_all_components,
+        inputs=[current_dataset_name, datasets],
+        outputs=[dataset_viewer, entry_selector]
+    )
+
+    # Update all components when an entry is added
+    def update_all_components_after_add(current_dataset_name, datasets):
+        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)]
+        return gr.update(value=html_content), gr.update(choices=entry_options)
+
+    add_button.click(
+        update_all_components_after_add,
+        inputs=[current_dataset_name, datasets],
+        outputs=[dataset_viewer, entry_selector]
+    )
+
+    # Update all components when an entry is edited
+    def update_all_components_after_edit(current_dataset_name, datasets):
+        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)]
+        return gr.update(value=html_content), gr.update(choices=entry_options)
+
+    edit_button.click(
+        update_all_components_after_edit,
+        inputs=[current_dataset_name, datasets],
+        outputs=[dataset_viewer, entry_selector]
+    )
+
+    # Update all components when an entry is deleted
+    def update_all_components_after_delete(current_dataset_name, datasets):
+        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)]
+        return gr.update(value=html_content), gr.update(choices=entry_options)
+
+    delete_button.click(
+        update_all_components_after_delete,
+        inputs=[current_dataset_name, datasets],
+        outputs=[dataset_viewer, entry_selector]
+    )
+
 demo.launch()