config.py

# config.py
# 配置相关：API、场景等
import os

BACKEND_URL = os.getenv("BACKEND_URL", "http://47.95.6.204:51001")
API_ENDPOINTS = {
    "submit_task": f"{BACKEND_URL}/predict/video",
    "query_status": f"{BACKEND_URL}/predict/task",
    "get_result": f"{BACKEND_URL}//predict"
}

SCENE_CONFIGS = {
    "demo1": {
        "description": "Demo 1",
        "objects": ["bedroom", "kitchen", "living room", ""],
        "preview_image": "./assets/scene_1.png",
        "glb_path": "scene_assets/scene1_no_ceiling.glb",
        "scene_name": "17DRP5sb8fy",
        "default_instruction": "Walk past the left side of the bed and stop in the doorway."
    },
    "demo2": {
        "description": "Demo 2", 
        "objects": ["office", "meeting room", "corridor"],
        "preview_image": "./assets/scene_2.png",
        "glb_path": "scene_assets/scene2_no_ceiling.glb",
        "scene_name": "r1Q1Z4BcV1o",
        "default_instruction": "Walk through the bathroom, past the sink and toilet. Stop in front of the counter with the two suitcase."
    },
    "demo3": {
        "description": "Demo 3",
        "objects": ["garage", "workshop", "storage"],
        "preview_image": "./assets/scene_3.png",
        "glb_path": "scene_assets/scene3_no_ceiling.glb",
        "scene_name": "dhjEzFoUFzH",
        "default_instruction": "Do a U-turn. Walk forward through the kitchen, heading to the black door. Walk out of the door and take a right onto the deck. Walk out on to the deck and stop."
    },
    "demo4": {
        "description": "Demo 4",
        "objects": ["garden", "patio", "pool"],
        "preview_image": "./assets/scene_4.png",
        "glb_path": "scene_assets/scene4_no_ceiling.glb",
        "scene_name": "demo4",  # 没有对应的图片文件，保持原名
        "default_instruction": "Walk out of bathroom and stand on white bath mat."
    },
    "demo5": {
        "description": "Demo 5",
        "objects": ["library", "hall", "lounge"],
        "preview_image": "./assets/scene_5.png",
        "glb_path": "scene_assets/scene4_no_ceiling.glb",  # 只有scene1-4的GLB文件
        "scene_name": "demo5",  # 没有对应的图片文件，保持原名
        "default_instruction": "Walk straight through the double wood doors, follow the red carpet straight to the next doorway and stop where the carpet splits off."
    },
}

EPISODE_CONFIGS = {
    "episode_1": {
        "description": "1",
    },
    "episode_2": {
        "description": "2",
    },
    "episode_3": {
        "description": "3",
    },
    "episode_4": {
        "description": "4",
    }
}

MODEL_CHOICES = ["rdp", "cma"]
MODE_CHOICES = ["vlnPE", "vlnCE"]