Fix

app.py

@@ -108,6 +108,35 @@ def _load_traj_entries(repo_id: str, directory: str) -> list:
     return entries
 
 
+def _load_instruction_file(repo_id: str, filepath: str) -> Optional[str]:
+    """Download a metadata JSON file and extract its instruction field."""
+    local_path = hf_hub_download(
+        repo_id=repo_id,
+        filename=filepath,
+        repo_type="dataset",
+    )
+
+    with open(local_path, "r", encoding="utf-8") as file:
+        data = json.load(file)
+
+    instruction: Optional[str] = None
+    if isinstance(data, dict):
+        instruction = data.get("instruction")
+    elif isinstance(data, list):
+        for entry in data:
+            if isinstance(entry, dict) and "instruction" in entry:
+                instruction = entry["instruction"]
+                break
+
+    if instruction is None:
+        return None
+
+    if isinstance(instruction, str):
+        return instruction.strip()
+
+    return str(instruction)
+
+
 def _format_annotation(index: int, annotation) -> str:
     prefix = f"Step {index + 1}"
     if isinstance(annotation, str):
@@ -122,10 +151,14 @@ def _format_annotation(index: int, annotation) -> str:
 
 def _prepare_gallery_items(
     repo_id: str, directory: Optional[str]
-) -> tuple[list[tuple[str, str]], list[str]]:
-    """Prepare gallery items and status messages for a directory."""
+) -> tuple[list[tuple[str, str]], list[str], str]:
+    """Prepare gallery items, status messages, and instruction text for a directory."""
     if not directory:
-        return [], ["Select a fourth-level folder to view screenshots."]
+        return (
+            [],
+            ["Select a fifth-level folder to view screenshots."],
+            "ℹ️ Instruction will appear once a fifth-level folder is selected.",
+        )
 
     prefix = f"{directory}/"
     files = [path for path in _list_repo_files(repo_id) if path.startswith(prefix)]
@@ -170,6 +203,55 @@ def _prepare_gallery_items(
             "⚠️ Mismatch between images and annotations; displaying available pairs."
         )
 
+    folder_name = directory.rsplit("/", 1)[-1]
+    metadata_files = [
+        path
+        for path in files
+        if "/" not in path[len(prefix) :]
+        and path.lower().endswith(".json")
+        and not path.lower().endswith(".jsonl")
+    ]
+
+    instruction_markdown = ""
+    if metadata_files:
+        metadata_files.sort(
+            key=lambda path: (
+                0
+                if path.rsplit("/", 1)[-1].rsplit(".", 1)[0] == folder_name
+                else 1,
+                path.rsplit("/", 1)[-1].lower(),
+            )
+        )
+        metadata_path = metadata_files[0]
+        metadata_filename = metadata_path.rsplit("/", 1)[-1]
+        try:
+            instruction_value = _load_instruction_file(repo_id, metadata_path)
+            if instruction_value:
+                instruction_markdown = f"**Instruction:** {instruction_value}"
+                status_lines.append("📋 Instruction: loaded")
+            else:
+                instruction_markdown = (
+                    f"⚠️ Instruction missing in `{metadata_filename}`."
+                )
+                status_lines.append("⚠️ Instruction: missing in metadata.")
+        except json.JSONDecodeError as error:
+            status_lines.append(
+                f"⚠️ Failed to parse `{metadata_filename}`: {error}"
+            )
+            instruction_markdown = (
+                f"⚠️ Unable to parse instruction from `{metadata_filename}`."
+            )
+        except Exception as error:  # pragma: no cover - unexpected IO errors
+            status_lines.append(
+                f"⚠️ Error loading `{metadata_filename}`: {error}"
+            )
+            instruction_markdown = (
+                f"⚠️ Unable to load instruction from `{metadata_filename}`."
+            )
+    else:
+        instruction_markdown = "⚠️ Instruction file not found in this folder."
+        status_lines.append("⚠️ Instruction: file not found.")
+
     gallery_items: list[tuple[str, str]] = []
     for idx, url in enumerate(image_urls):
         caption = "No annotation available"
@@ -177,7 +259,7 @@ def _prepare_gallery_items(
             caption = _format_annotation(idx, annotations[idx])
         gallery_items.append((url, caption))
 
-    return gallery_items, status_lines
+    return gallery_items, status_lines, instruction_markdown
 
 
 def _dropdown_update(
@@ -209,6 +291,8 @@ def refresh_repo(repo_id: str):
             gr.update(choices=[], value=None, interactive=False),
             gr.update(choices=[], value=None, interactive=False),
             gr.update(choices=[], value=None, interactive=False),
+            gr.update(choices=[], value=None, interactive=False),
+            gr.update(value=""),
             gr.update(value=[]),
             gr.update(value=""),
             gr.update(value=f"❌ Unable to load repo `{repo_id}`: {error}"),
@@ -221,6 +305,8 @@ def refresh_repo(repo_id: str):
             gr.update(choices=[], value=None, interactive=False),
             gr.update(choices=[], value=None, interactive=False),
             gr.update(choices=[], value=None, interactive=False),
+            gr.update(choices=[], value=None, interactive=False),
+            gr.update(value=""),
             gr.update(value=[]),
             gr.update(value=""),
             gr.update(value=f"❌ Unexpected error loading `{repo_id}`: {error}"),
@@ -255,12 +341,29 @@ def refresh_repo(repo_id: str):
     )
     fourth_value = fourth_dirs[0] if fourth_dirs else None
 
-    target_directory = (
-        _build_path(top_value, second_value, third_value, fourth_value)
+    fifth_dirs = (
+        _get_subdirectories(
+            repo_id, _build_path(top_value, second_value, third_value, fourth_value)
+        )
         if fourth_value
+        else []
+    )
+    fifth_value = fifth_dirs[0] if fifth_dirs else None
+
+    target_directory = (
+        _build_path(
+            top_value,
+            second_value,
+            third_value,
+            fourth_value,
+            fifth_value,
+        )
+        if fifth_value
         else None
     )
-    gallery_items, gallery_status = _prepare_gallery_items(repo_id, target_directory)
+    gallery_items, gallery_status, instruction_markdown = _prepare_gallery_items(
+        repo_id, target_directory
+    )
 
     first_dropdown_update = _dropdown_update(
         choices=top_dirs,
@@ -294,17 +397,29 @@ def refresh_repo(repo_id: str):
         empty_info="No fourth-level folders under the selection",
     )
 
+    fifth_dropdown_update = _dropdown_update(
+        choices=fifth_dirs,
+        value=fifth_value,
+        label="Fifth-level folders",
+        filled_info="Choose a fifth-level folder",
+        empty_info="No fifth-level folders under the selection",
+    )
+
+    instruction_update = gr.update(value=instruction_markdown)
     gallery_update = gr.update(value=gallery_items)
     annotation_update = gr.update(value="\n".join(gallery_status))
+    status_update = gr.update(value="\n".join(status_lines))
 
     return (
         first_dropdown_update,
         second_dropdown_update,
         third_dropdown_update,
         fourth_dropdown_update,
+        fifth_dropdown_update,
+        instruction_update,
         gallery_update,
         annotation_update,
-        gr.update(value="\n".join(status_lines)),
+        status_update,
     )
 
 
@@ -333,10 +448,29 @@ def update_second_dropdown(repo_id: str, top_level_dir: str):
                 filled_info="Choose a fourth-level folder",
                 empty_info="Select a higher-level folder first",
             )
+            empty_fifth = _dropdown_update(
+                choices=[],
+                value=None,
+                label="Fifth-level folders",
+                filled_info="Choose a fifth-level folder",
+                empty_info="Select a higher-level folder first",
+            )
+            instruction_update = gr.update(
+                value="ℹ️ Select a top-level folder to display its instruction."
+            )
+            gallery_update = gr.update(value=[])
             annotation_update = gr.update(
                 value="Select a top-level folder to load screenshots and annotations."
             )
-            return empty_second, empty_third, empty_fourth, gr.update(value=[]), annotation_update
+            return (
+                empty_second,
+                empty_third,
+                empty_fourth,
+                empty_fifth,
+                instruction_update,
+                gallery_update,
+                annotation_update,
+            )
 
         second_dirs = _get_subdirectories(repo_id, top_level_dir)
         second_value = second_dirs[0] if second_dirs else None
@@ -349,18 +483,40 @@ def update_second_dropdown(repo_id: str, top_level_dir: str):
         third_value = third_dirs[0] if third_dirs else None
 
         fourth_dirs = (
-            _get_subdirectories(repo_id, _build_path(top_level_dir, second_value, third_value))
+            _get_subdirectories(
+                repo_id, _build_path(top_level_dir, second_value, third_value)
+            )
             if third_value
             else []
         )
         fourth_value = fourth_dirs[0] if fourth_dirs else None
 
-        target_directory = (
-            _build_path(top_level_dir, second_value, third_value, fourth_value)
+        fifth_dirs = (
+            _get_subdirectories(
+                repo_id,
+                _build_path(top_level_dir, second_value, third_value, fourth_value),
+            )
             if fourth_value
+            else []
+        )
+        fifth_value = fifth_dirs[0] if fifth_dirs else None
+
+        target_directory = (
+            _build_path(
+                top_level_dir,
+                second_value,
+                third_value,
+                fourth_value,
+                fifth_value,
+            )
+            if fifth_value
             else None
         )
-        gallery_items, gallery_status = _prepare_gallery_items(repo_id, target_directory)
+        (
+            gallery_items,
+            gallery_status,
+            instruction_markdown,
+        ) = _prepare_gallery_items(repo_id, target_directory)
 
         return (
             _dropdown_update(
@@ -384,6 +540,14 @@ def update_second_dropdown(repo_id: str, top_level_dir: str):
                 filled_info="Choose a fourth-level folder",
                 empty_info="No fourth-level folders under the selection",
             ),
+            _dropdown_update(
+                choices=fifth_dirs,
+                value=fifth_value,
+                label="Fifth-level folders",
+                filled_info="Choose a fifth-level folder",
+                empty_info="No fifth-level folders under the selection",
+            ),
+            gr.update(value=instruction_markdown),
             gr.update(value=gallery_items),
             gr.update(value="\n".join(gallery_status)),
         )
@@ -411,10 +575,19 @@ def update_second_dropdown(repo_id: str, top_level_dir: str):
             filled_info="Choose a fourth-level folder",
             empty_info="Unable to load subdirectories",
         )
+        empty_fifth = _dropdown_update(
+            choices=[],
+            value=None,
+            label="Fifth-level folders",
+            filled_info="Choose a fifth-level folder",
+            empty_info="Unable to load subdirectories",
+        )
         return (
             empty_second,
             empty_third,
             empty_fourth,
+            empty_fifth,
+            gr.update(value="Unable to load instruction."),
             gr.update(value=[]),
             gr.update(value="Unable to load screenshots or annotations."),
         )
@@ -438,10 +611,28 @@ def update_third_dropdown(repo_id: str, top_level_dir: str, second_level_dir: st
                 filled_info="Choose a fourth-level folder",
                 empty_info="Select higher-level folders first",
             )
+            empty_fifth = _dropdown_update(
+                choices=[],
+                value=None,
+                label="Fifth-level folders",
+                filled_info="Choose a fifth-level folder",
+                empty_info="Select higher-level folders first",
+            )
+            instruction_update = gr.update(
+                value="ℹ️ Select higher-level folders to display the instruction."
+            )
+            gallery_update = gr.update(value=[])
             annotation_update = gr.update(
                 value="Select higher-level folders to load screenshots and annotations."
             )
-            return empty_third, empty_fourth, gr.update(value=[]), annotation_update
+            return (
+                empty_third,
+                empty_fourth,
+                empty_fifth,
+                instruction_update,
+                gallery_update,
+                annotation_update,
+            )
 
         third_dirs = _get_subdirectories(
             repo_id, _build_path(top_level_dir, second_level_dir)
@@ -457,12 +648,32 @@ def update_third_dropdown(repo_id: str, top_level_dir: str, second_level_dir: st
         )
         fourth_value = fourth_dirs[0] if fourth_dirs else None
 
-        target_directory = (
-            _build_path(top_level_dir, second_level_dir, third_value, fourth_value)
+        fifth_dirs = (
+            _get_subdirectories(
+                repo_id,
+                _build_path(top_level_dir, second_level_dir, third_value, fourth_value),
+            )
             if fourth_value
+            else []
+        )
+        fifth_value = fifth_dirs[0] if fifth_dirs else None
+
+        target_directory = (
+            _build_path(
+                top_level_dir,
+                second_level_dir,
+                third_value,
+                fourth_value,
+                fifth_value,
+            )
+            if fifth_value
             else None
         )
-        gallery_items, gallery_status = _prepare_gallery_items(repo_id, target_directory)
+        (
+            gallery_items,
+            gallery_status,
+            instruction_markdown,
+        ) = _prepare_gallery_items(repo_id, target_directory)
 
         return (
             _dropdown_update(
@@ -479,6 +690,14 @@ def update_third_dropdown(repo_id: str, top_level_dir: str, second_level_dir: st
                 filled_info="Choose a fourth-level folder",
                 empty_info="No fourth-level folders under the selection",
             ),
+            _dropdown_update(
+                choices=fifth_dirs,
+                value=fifth_value,
+                label="Fifth-level folders",
+                filled_info="Choose a fifth-level folder",
+                empty_info="No fifth-level folders under the selection",
+            ),
+            gr.update(value=instruction_markdown),
             gr.update(value=gallery_items),
             gr.update(value="\n".join(gallery_status)),
         )
@@ -502,9 +721,18 @@ def update_third_dropdown(repo_id: str, top_level_dir: str, second_level_dir: st
             filled_info="Choose a fourth-level folder",
             empty_info="Unable to load subdirectories",
         )
+        empty_fifth = _dropdown_update(
+            choices=[],
+            value=None,
+            label="Fifth-level folders",
+            filled_info="Choose a fifth-level folder",
+            empty_info="Unable to load subdirectories",
+        )
         return (
             empty_third,
             empty_fourth,
+            empty_fifth,
+            gr.update(value="Unable to load instruction."),
             gr.update(value=[]),
             gr.update(value="Unable to load screenshots or annotations."),
         )
@@ -516,7 +744,7 @@ def update_fourth_dropdown(
     second_level_dir: str,
     third_level_dir: str,
 ):
-    """Update the fourth dropdown and gallery when the third-level changes."""
+    """Update the fourth and fifth dropdowns when the third-level changes."""
     try:
         if not top_level_dir or not second_level_dir or not third_level_dir:
             empty_fourth = _dropdown_update(
@@ -526,10 +754,27 @@ def update_fourth_dropdown(
                 filled_info="Choose a fourth-level folder",
                 empty_info="Select higher-level folders first",
             )
+            empty_fifth = _dropdown_update(
+                choices=[],
+                value=None,
+                label="Fifth-level folders",
+                filled_info="Choose a fifth-level folder",
+                empty_info="Select higher-level folders first",
+            )
+            instruction_update = gr.update(
+                value="ℹ️ Select higher-level folders to display the instruction."
+            )
+            gallery_update = gr.update(value=[])
             annotation_update = gr.update(
                 value="Select higher-level folders to load screenshots and annotations."
             )
-            return empty_fourth, gr.update(value=[]), annotation_update
+            return (
+                empty_fourth,
+                empty_fifth,
+                instruction_update,
+                gallery_update,
+                annotation_update,
+            )
 
         fourth_dirs = _get_subdirectories(
             repo_id,
@@ -537,17 +782,34 @@ def update_fourth_dropdown(
         )
         fourth_value = fourth_dirs[0] if fourth_dirs else None
 
+        fifth_dirs = (
+            _get_subdirectories(
+                repo_id,
+                _build_path(
+                    top_level_dir, second_level_dir, third_level_dir, fourth_value
+                ),
+            )
+            if fourth_value
+            else []
+        )
+        fifth_value = fifth_dirs[0] if fifth_dirs else None
+
         target_directory = (
             _build_path(
                 top_level_dir,
                 second_level_dir,
                 third_level_dir,
                 fourth_value,
+                fifth_value,
             )
-            if fourth_value
+            if fifth_value
             else None
         )
-        gallery_items, gallery_status = _prepare_gallery_items(repo_id, target_directory)
+        (
+            gallery_items,
+            gallery_status,
+            instruction_markdown,
+        ) = _prepare_gallery_items(repo_id, target_directory)
 
         return (
             _dropdown_update(
@@ -557,6 +819,14 @@ def update_fourth_dropdown(
                 filled_info="Choose a fourth-level folder",
                 empty_info="No fourth-level folders under the selection",
             ),
+            _dropdown_update(
+                choices=fifth_dirs,
+                value=fifth_value,
+                label="Fifth-level folders",
+                filled_info="Choose a fifth-level folder",
+                empty_info="No fifth-level folders under the selection",
+            ),
+            gr.update(value=instruction_markdown),
             gr.update(value=gallery_items),
             gr.update(value="\n".join(gallery_status)),
         )
@@ -574,44 +844,163 @@ def update_fourth_dropdown(
             filled_info="Choose a fourth-level folder",
             empty_info="Unable to load subdirectories",
         )
+        empty_fifth = _dropdown_update(
+            choices=[],
+            value=None,
+            label="Fifth-level folders",
+            filled_info="Choose a fifth-level folder",
+            empty_info="Unable to load subdirectories",
+        )
         return (
             empty_fourth,
+            empty_fifth,
+            gr.update(value="Unable to load instruction."),
             gr.update(value=[]),
             gr.update(value="Unable to load screenshots or annotations."),
         )
 
 
-def update_gallery(
+def update_fifth_dropdown(
     repo_id: str,
     top_level_dir: str,
     second_level_dir: str,
     third_level_dir: str,
     fourth_level_dir: str,
 ):
-    """Update the image gallery when the fourth-level selection changes."""
+    """Update the fifth dropdown and gallery when the fourth-level changes."""
     try:
-        if not all([top_level_dir, second_level_dir, third_level_dir, fourth_level_dir]):
+        if not all(
+            [top_level_dir, second_level_dir, third_level_dir, fourth_level_dir]
+        ):
+            empty_fifth = _dropdown_update(
+                choices=[],
+                value=None,
+                label="Fifth-level folders",
+                filled_info="Choose a fifth-level folder",
+                empty_info="Select higher-level folders first",
+            )
+            instruction_update = gr.update(
+                value="ℹ️ Select higher-level folders to display the instruction."
+            )
+            gallery_update = gr.update(value=[])
             annotation_update = gr.update(
-                value="Select all folder levels to load screenshots and annotations."
+                value="Select higher-level folders to load screenshots and annotations."
             )
-            return gr.update(value=[]), annotation_update
+            return empty_fifth, instruction_update, gallery_update, annotation_update
 
-        directory = _build_path(
-            top_level_dir,
-            second_level_dir,
-            third_level_dir,
-            fourth_level_dir,
+        fifth_dirs = _get_subdirectories(
+            repo_id,
+            _build_path(
+                top_level_dir,
+                second_level_dir,
+                third_level_dir,
+                fourth_level_dir,
+            ),
+        )
+        fifth_value = fifth_dirs[0] if fifth_dirs else None
+
+        target_directory = (
+            _build_path(
+                top_level_dir,
+                second_level_dir,
+                third_level_dir,
+                fourth_level_dir,
+                fifth_value,
+            )
+            if fifth_value
+            else None
+        )
+        (
+            gallery_items,
+            gallery_status,
+            instruction_markdown,
+        ) = _prepare_gallery_items(repo_id, target_directory)
+
+        return (
+            _dropdown_update(
+                choices=fifth_dirs,
+                value=fifth_value,
+                label="Fifth-level folders",
+                filled_info="Choose a fifth-level folder",
+                empty_info="No fifth-level folders under the selection",
+            ),
+            gr.update(value=instruction_markdown),
+            gr.update(value=gallery_items),
+            gr.update(value="\n".join(gallery_status)),
         )
-        gallery_items, gallery_status = _prepare_gallery_items(repo_id, directory)
-        return gr.update(value=gallery_items), gr.update(value="\n".join(gallery_status))
     except Exception as error:
         print(
-            "[update_gallery] Error for "
+            "[update_fifth_dropdown] Error for "
             f"{repo_id}/{top_level_dir}/{second_level_dir}/{third_level_dir}/{fourth_level_dir}: {error}",
             flush=True,
         )
         print(traceback.format_exc(), flush=True)
-        return gr.update(value=[]), gr.update(value="Unable to load screenshots or annotations.")
+        empty_fifth = _dropdown_update(
+            choices=[],
+            value=None,
+            label="Fifth-level folders",
+            filled_info="Choose a fifth-level folder",
+            empty_info="Unable to load subdirectories",
+        )
+        return (
+            empty_fifth,
+            gr.update(value="Unable to load instruction."),
+            gr.update(value=[]),
+            gr.update(value="Unable to load screenshots or annotations."),
+        )
+
+
+def update_gallery(
+    repo_id: str,
+    top_level_dir: str,
+    second_level_dir: str,
+    third_level_dir: str,
+    fourth_level_dir: str,
+    fifth_level_dir: str,
+):
+    """Update the image gallery when the fifth-level selection changes."""
+    try:
+        directory = (
+            _build_path(
+                top_level_dir,
+                second_level_dir,
+                third_level_dir,
+                fourth_level_dir,
+                fifth_level_dir,
+            )
+            if all(
+                [
+                    top_level_dir,
+                    second_level_dir,
+                    third_level_dir,
+                    fourth_level_dir,
+                    fifth_level_dir,
+                ]
+            )
+            else None
+        )
+        (
+            gallery_items,
+            gallery_status,
+            instruction_markdown,
+        ) = _prepare_gallery_items(repo_id, directory)
+        return (
+            gr.update(value=instruction_markdown),
+            gr.update(value=gallery_items),
+            gr.update(value="\n".join(gallery_status)),
+        )
+    except Exception as error:
+        print(
+            "[update_gallery] Error for "
+            f"{repo_id}/{top_level_dir}/{second_level_dir}/{third_level_dir}/{fourth_level_dir}/{fifth_level_dir}: {error}",
+            flush=True,
+        )
+        print(traceback.format_exc(), flush=True)
+        return (
+            gr.update(value="Unable to load instruction."),
+            gr.update(value=[]),
+            gr.update(value="Unable to load screenshots or annotations."),
+        )
 
 
 with gr.Blocks(title="HF Dataset Explorer") as demo:
@@ -635,6 +1024,8 @@ Provide a dataset repository ID (e.g. `org/dataset`) to list its top-level folde
     second_level_dropdown = gr.Dropdown(label="Second-level folders", interactive=False)
     third_level_dropdown = gr.Dropdown(label="Third-level folders", interactive=False)
     fourth_level_dropdown = gr.Dropdown(label="Fourth-level folders", interactive=False)
+    fifth_level_dropdown = gr.Dropdown(label="Fifth-level folders", interactive=False)
+    instruction_display = gr.Markdown()
     image_gallery = gr.Gallery(label="Images", columns=4)
     annotation_status = gr.Markdown()
 
@@ -646,6 +1037,8 @@ Provide a dataset repository ID (e.g. `org/dataset`) to list its top-level folde
             second_level_dropdown,
             third_level_dropdown,
             fourth_level_dropdown,
+            fifth_level_dropdown,
+            instruction_display,
             image_gallery,
             annotation_status,
             status_display,
@@ -659,6 +1052,8 @@ Provide a dataset repository ID (e.g. `org/dataset`) to list its top-level folde
             second_level_dropdown,
             third_level_dropdown,
             fourth_level_dropdown,
+            fifth_level_dropdown,
+            instruction_display,
             image_gallery,
             annotation_status,
         ],
@@ -670,6 +1065,8 @@ Provide a dataset repository ID (e.g. `org/dataset`) to list its top-level folde
         outputs=[
             third_level_dropdown,
             fourth_level_dropdown,
+            fifth_level_dropdown,
+            instruction_display,
             image_gallery,
             annotation_status,
         ],
@@ -685,12 +1082,31 @@ Provide a dataset repository ID (e.g. `org/dataset`) to list its top-level folde
         ],
         outputs=[
             fourth_level_dropdown,
+            fifth_level_dropdown,
+            instruction_display,
             image_gallery,
             annotation_status,
         ],
     )
 
     fourth_level_dropdown.change(
+        update_fifth_dropdown,
+        inputs=[
+            repo_id_input,
+            folder_dropdown,
+            second_level_dropdown,
+            third_level_dropdown,
+            fourth_level_dropdown,
+        ],
+        outputs=[
+            fifth_level_dropdown,
+            instruction_display,
+            image_gallery,
+            annotation_status,
+        ],
+    )
+
+    fifth_level_dropdown.change(
         update_gallery,
         inputs=[
             repo_id_input,
@@ -698,8 +1114,9 @@ Provide a dataset repository ID (e.g. `org/dataset`) to list its top-level folde
             second_level_dropdown,
             third_level_dropdown,
             fourth_level_dropdown,
+            fifth_level_dropdown,
         ],
-        outputs=[image_gallery, annotation_status],
+        outputs=[instruction_display, image_gallery, annotation_status],
     )
 
     demo.load(
@@ -710,6 +1127,8 @@ Provide a dataset repository ID (e.g. `org/dataset`) to list its top-level folde
             second_level_dropdown,
             third_level_dropdown,
             fourth_level_dropdown,
+            fifth_level_dropdown,
+            instruction_display,
             image_gallery,
             annotation_status,
             status_display,