Update app.py

app.py

@@ -136,9 +136,12 @@ with gr.Blocks() as demo:
     datasets = gr.State({})
     current_dataset_name = gr.State("")
     current_page_number = gr.State(0)
-    dataset_selector = gr.Dropdown(label="Select Dataset", interactive=True)
+    # Place top-level components in a Column
+    with gr.Column():
+        dataset_selector = gr.Dropdown(label="Select Dataset", interactive=True)
+        message_box = gr.Textbox(interactive=False, label="Message")
+    # Define entry_selector but do not place it in the interface yet
     entry_selector = gr.Dropdown(label="Select Entry to Edit/Delete")
-    message_box = gr.Textbox(interactive=False, label="Message")
 
     # Tabs as before
     with gr.Tabs():
@@ -220,7 +223,8 @@ with gr.Blocks() as demo:
             with gr.Column():
                 selected_image = gr.Image(label="Selected Image", interactive=False, type="numpy")
                 selected_prompt = gr.Textbox(label="Current Prompt", interactive=False)
-                entry_selector = gr.Dropdown(label="Select Entry to Edit/Delete")
+                # Include entry_selector in the layout
+                entry_selector
                 new_prompt_input = gr.Textbox(label="New Prompt (for Edit)")
                 with gr.Row():
                     edit_button = gr.Button("Edit Entry")