Update app.py

app.py

@@ -416,6 +416,7 @@ with gr.Blocks() as demo:
         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)]
+        dataset_viewer.update(value=html_content)  # Update dataset_viewer component
         return gr.update(value=html_content), gr.update(choices=entry_options)
 
     dataset_selector.change(
@@ -431,6 +432,7 @@ with gr.Blocks() as demo:
         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)]
+        dataset_viewer.update(value=html_content)  # Update dataset_viewer component
         return gr.update(value=html_content), gr.update(choices=entry_options)
 
     add_button.click(
@@ -446,6 +448,7 @@ with gr.Blocks() as demo:
         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)]
+        dataset_viewer.update(value=html_content)  # Update dataset_viewer component
         return gr.update(value=html_content), gr.update(choices=entry_options)
 
     edit_button.click(
@@ -461,6 +464,7 @@ with gr.Blocks() as demo:
         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)]
+        dataset_viewer.update(value=html_content)  # Update dataset_viewer component
         return gr.update(value=html_content), gr.update(choices=entry_options)
 
     delete_button.click(