import subprocess
import sys

def main():
    # 1. 定义数据集id和你要查看的片段索引
    repo_id = "lerobot/svla_so101_pickplace"
    episode_index = 0      # 从0开始，0代表数据集的第一个演示片段
    host = "127.0.0.1"     # 服务运行在本机
    port = 9091            # 网页服务端口，避免与常见端口冲突

    # 2. 构建并执行命令
    command = [
        sys.executable,                     # 获取当前Python解释器路径
        "-m", "lerobot.scripts.visualize_dataset_html",  # 以模块方式运行脚本
        "--repo-id", repo_id,
        "--episodes", str(episode_index),   # 只显示指定的片段
        "--host", host,
        "--port", str(port)
    ]

    print(f"🚀 正在启动可视化服务，数据集：{repo_id}，片段：{episode_index}")
    print(f"🌐 服务启动后，请在浏览器中打开：http://{host}:{port}")
    print("⚠️  按 Ctrl+C 可停止服务")

    # 3. 运行命令
    try:
        subprocess.run(command, check=True)
    except KeyboardInterrupt:
        print("\n👋 已停止可视化服务。")
    except subprocess.CalledProcessError as e:
        print(f"❌ 命令执行出错，退出码：{e.returncode}")
        sys.exit(1)

if __name__ == "__main__":
    main()