准星

gradio-web/config.py

@@ -24,6 +24,25 @@ SESSION_TIMEOUT = 300  # Session超时时间（秒），如果30秒内没有exec
 # 兜底执行次数配置
 EXECUTE_LIMIT_OFFSET = 4  # 兜底执行次数 = non_demonstration_task_length + EXECUTE_LIMIT_OFFSET
 
+TASK_NAME_LIST = [
+    "BinFill",
+    "StopCube",
+    "PickXtimes",
+    "SwingXtimes",
+    "ButtonUnmask",
+    "VideoUnmask",
+    "VideoUnmaskSwap",
+    "ButtonUnmaskSwap",
+    "PickHighlight",
+    "VideoRepick",
+    "VideoPlaceButton",
+    "VideoPlaceOrder",
+    "MoveCube",
+    "InsertPeg",
+    "PatternLock",
+    "RouteStick",
+]
+
 
 # 应该显示demonstration videos的环境ID列表
 DEMO_VIDEO_ENV_IDS = [

gradio-web/image_utils.py

@@ -702,16 +702,17 @@ def draw_coordinate_axes(img, position="right", rotate_180=False, env_id=None):
 
 
 def draw_marker(img, x, y):
-    """Draws a red circle and cross at (x, y)."""
+    """Draws a fluorescent yellow circle and cross at (x, y)."""
     if isinstance(img, np.ndarray):
         img = Image.fromarray(img)
     
     img = img.copy()
     draw = ImageDraw.Draw(img)
     r = 5
+    marker_color = (204, 255, 0)
     # Circle
-    draw.ellipse((x-r, y-r, x+r, y+r), outline="red", width=2)
+    draw.ellipse((x-r, y-r, x+r, y+r), outline=marker_color, width=2)
     # Cross
-    draw.line((x-r, y, x+r, y), fill="red", width=2)
-    draw.line((x, y-r, x, y+r), fill="red", width=2)
+    draw.line((x-r, y, x+r, y), fill=marker_color, width=2)
+    draw.line((x, y-r, x, y+r), fill=marker_color, width=2)
     return img

gradio-web/test/test_reference_action_callbacks.py

@@ -25,6 +25,10 @@ class _FakeOptionSession:
         return Image.new("RGB", (24, 24), color=(0, 0, 0))
 
 
+def _is_fluorescent_yellow(pixel):
+    return pixel[0] > 180 and pixel[1] > 200 and pixel[2] < 80
+
+
 def test_on_reference_action_success_updates_option_and_coords(monkeypatch, reload_module):
     config = reload_module("config")
     callbacks = reload_module("gradio_callbacks")
@@ -48,7 +52,7 @@ def test_on_reference_action_success_updates_option_and_coords(monkeypatch, relo
 
     assert img_update.get("__type__") == "update"
     assert isinstance(img_update.get("value"), Image.Image)
-    assert img_update["value"].getpixel((5, 6)) != (0, 0, 0)
+    assert _is_fluorescent_yellow(img_update["value"].getpixel((5, 6)))
     assert img_update.get("elem_classes") == config.get_live_obs_elem_classes()
     assert option_update.get("value") == 2
     assert coords_text == "5, 6"
@@ -117,7 +121,7 @@ def test_on_map_click_clears_wait_state_and_restores_action_prompt(monkeypatch,
 
     assert img_update.get("__type__") == "update"
     assert isinstance(img_update.get("value"), Image.Image)
-    assert img_update["value"].getpixel((5, 6)) != (0, 0, 0)
+    assert _is_fluorescent_yellow(img_update["value"].getpixel((5, 6)))
     assert img_update.get("elem_classes") == config.get_live_obs_elem_classes()
     assert coords_text == "5, 6"
     assert log_text == config.UI_TEXT["log"]["action_selection_prompt"]

gradio-web/test/test_ui_native_layout_contract.py

@@ -97,6 +97,25 @@ def test_render_header_goal_capitalizes_display_value(reload_module):
     assert ui_layout.render_header_goal("") == "—"
 
 
+def test_header_task_dropdown_uses_task_name_list_order(reload_module):
+    config = reload_module("config")
+    ui_layout = reload_module("ui_layout")
+
+    ui_layout.user_manager.env_choices = list(config.TASK_NAME_LIST)
+    demo = ui_layout.create_ui_blocks()
+
+    try:
+        cfg = demo.get_config_file()
+        header_task_comp = next(
+            comp
+            for comp in cfg.get("components", [])
+            if comp.get("props", {}).get("elem_id") == "header_task"
+        )
+        assert header_task_comp.get("props", {}).get("choices") == config.TASK_NAME_LIST
+    finally:
+        demo.close()
+
+
 def test_native_ui_config_contains_phase_machine_and_precheck_chain(reload_module):
     ui_layout = reload_module("ui_layout")
     demo = ui_layout.create_ui_blocks()

gradio-web/test/test_user_manager_random_flow.py

@@ -94,3 +94,22 @@ def test_init_session_fails_when_metadata_root_missing(monkeypatch, reload_modul
     assert success is False
     assert "No available environments" in msg
     assert status is None
+
+
+def test_env_choices_follow_task_name_list_order(monkeypatch, reload_module, tmp_path):
+    metadata_root = tmp_path / "metadata"
+    _write_metadata(metadata_root, "VideoPlaceButton", [0])
+    _write_metadata(metadata_root, "BinFill", [0])
+    _write_metadata(metadata_root, "PatternLock", [0])
+    _write_metadata(metadata_root, "StopCube", [0])
+    monkeypatch.setenv("ROBOMME_METADATA_ROOT", str(metadata_root))
+
+    user_manager_mod = reload_module("user_manager")
+    manager = user_manager_mod.UserManager()
+
+    assert manager.env_choices == [
+        "BinFill",
+        "StopCube",
+        "VideoPlaceButton",
+        "PatternLock",
+    ]

gradio-web/user_manager.py

@@ -4,6 +4,7 @@ import random
 import threading
 from pathlib import Path
 
+from config import TASK_NAME_LIST
 from state_manager import clear_task_start_time, get_task_start_time
 
 
@@ -16,7 +17,7 @@ class UserManager:
         self.lock = threading.Lock()
 
         self.env_to_episodes = self._load_env_episode_pool()
-        self.env_choices = sorted(self.env_to_episodes.keys())
+        self.env_choices = self._build_env_choices()
 
         # Session-local progress only (no disk persistence)
         self.session_progress = {}
@@ -61,6 +62,12 @@ class UserManager:
         print(f"Loaded random env pool: {len(env_to_episodes)} envs from metadata root {metadata_root}")
         return env_to_episodes
 
+    def _build_env_choices(self):
+        available_envs = set(self.env_to_episodes.keys())
+        ordered_choices = [env_id for env_id in TASK_NAME_LIST if env_id in available_envs]
+        remaining_choices = sorted(available_envs - set(ordered_choices))
+        return ordered_choices + remaining_choices
+
     def _ensure_session_entry(self, uid):
         if uid not in self.session_progress:
             self.session_progress[uid] = {