Update app.py

app.py

@@ -144,6 +144,14 @@ with gr.Blocks() as demo:
         dataset_selector = gr.Dropdown(label="Select Dataset", interactive=True)
         message_box = gr.Textbox(interactive=False, label="Message")
 
+    # Dataset Viewer and Pagination Controls at the Bottom
+    with gr.Column():
+        gr.Markdown("### Dataset Viewer")
+        dataset_html = gr.HTML()
+        with gr.Row():
+            prev_button = gr.Button("Previous Page")
+            next_button = gr.Button("Next Page")
+
     # Tabs as before
     with gr.Tabs():
         with gr.TabItem("Create / Upload Dataset"):
@@ -217,7 +225,7 @@ with gr.Blocks() as demo:
             add_button.click(
                 add_entry,
                 inputs=[image_input, prompt_input, current_dataset_name, datasets],
-                outputs=[datasets, current_page_number, gr.HTML.update(), entry_selector, message_box]
+                outputs=[datasets, current_page_number, dataset_html, entry_selector, message_box]
             )
 
         with gr.TabItem("Edit / Delete Entry"):
@@ -266,7 +274,7 @@ with gr.Blocks() as demo:
             edit_button.click(
                 edit_entry,
                 inputs=[entry_selector, new_prompt_input, current_dataset_name, datasets, current_page_number],
-                outputs=[datasets, gr.HTML.update(), entry_selector, new_prompt_input, message_box]
+                outputs=[datasets, dataset_html, entry_selector, new_prompt_input, message_box]
             )
 
             def delete_entry(entry_option, current_dataset_name, datasets, current_page_number):
@@ -285,7 +293,7 @@ with gr.Blocks() as demo:
             delete_button.click(
                 delete_entry,
                 inputs=[entry_selector, current_dataset_name, datasets, current_page_number],
-                outputs=[datasets, gr.HTML.update(), entry_selector, selected_image, selected_prompt, message_box]
+                outputs=[datasets, dataset_html, entry_selector, selected_image, selected_prompt, message_box]
             )
 
         with gr.TabItem("Download Dataset"):
@@ -311,14 +319,6 @@ with gr.Blocks() as demo:
                 outputs=[download_output, message_box]
             )
 
-    # Dataset Viewer and Pagination Controls at the Bottom
-    with gr.Column():
-        gr.Markdown("### Dataset Viewer")
-        dataset_html = gr.HTML()
-        with gr.Row():
-            prev_button = gr.Button("Previous Page")
-            next_button = gr.Button("Next Page")
-
     def select_dataset(dataset_name, datasets):
         if dataset_name in datasets:
             dataset = datasets[dataset_name]