| """ |
| src/ui/examples.py — Demo image loader for the Gradio Examples component. |
| |
| Reads demo images from the configured assets directory and pairs each |
| image with a contextually appropriate natural-language command. |
| """ |
|
|
| from pathlib import Path |
|
|
| from src.utils.config import DEMO_IMAGES_DIR |
|
|
|
|
| |
| _IMAGE_COMMANDS: dict[str, str] = { |
| "assembly_line.jpg": "Pick up the defective board and place it in the red bin", |
| "construction_site.jpg": "Move the hard hat next to the gloves", |
| "desk_scene.jpg": "Pick up the bottle and place it in the box", |
| "garage_tools.jpg": "Place the wrench back into the toolbox", |
| "grocery_shelf.jpg": "Move the water bottle into the shopping basket", |
| "kitchen_counter.jpg": "Pick up the apple and place it in the bowl", |
| "lab_equipment.jpg": "Move the bottle next to the microscope", |
| "medical_tray.jpg": "Pick up the scissors and place them on the tray", |
| "office_supplies.jpg": "Move the phone next to the laptop", |
| "warehouse_boxes.jpg": "Move the closest box onto the hand truck", |
| } |
|
|
|
|
| def get_demo_examples() -> list[list]: |
| """ |
| Return a list of ``[image_path, command]`` pairs for Gradio Examples. |
| |
| Returns: |
| Sorted list of ``[str, str]`` pairs, one per demo image found on disk. |
| Returns an empty list if the demo images directory does not exist. |
| """ |
| demo_dir = Path(DEMO_IMAGES_DIR) |
| if not demo_dir.exists(): |
| return [] |
|
|
| examples: list[list] = [] |
| for img_path in sorted(demo_dir.glob("*.jpg")): |
| cmd = _IMAGE_COMMANDS.get(img_path.name, "Inspect the closest object") |
| examples.append([str(img_path), cmd]) |
|
|
| return examples |
|
|
