add scene idx

app.py

@@ -56,18 +56,17 @@ def display_version_info():
     return version_info
 
 
-def run_simulation(scene, model, mode, prompt, history, request: gr.Request):
+def run_simulation(scene, episode, model, mode, prompt, history, request: gr.Request):
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     scene_desc = SCENE_CONFIGS.get(scene, {}).get("description", scene)
     user_ip = request.client.host if request else "unknown"
     session_id = request.session_hash
-    
     if not is_request_allowed(user_ip):
         log_submission(scene, prompt, model, user_ip, "IP blocked temporarily")
         raise gr.Error("Too many requests from this IP. Please wait and try again one minute later.")
     
     # 提交任务到后端
-    submission_result = submit_to_backend(scene, prompt, mode, model, user_ip)
+    submission_result = submit_to_backend(scene, episode, prompt, mode, model, user_ip)
     if submission_result.get("status") != "pending":
         log_submission(scene, prompt, model, user_ip, "Submission failed")
         raise gr.Error(f"Submission failed: {submission_result.get('message', 'unknown issue')}")
@@ -324,18 +323,19 @@ with gr.Blocks(title="InternNav Model Inference Demo", css=custom_css) as demo:
                     history_slots.append((slot, accordion, video, detail_md))
     gr.Examples(
         examples=[
-            ["demo1", "rdp", "vlnPE", "Walk past the left side of the bed and stop in the doorway."],
-            ["demo2", "rdp", "vlnPE", "Walk through the bathroom, past the sink and toilet. Stop in front of the counter with the two suitcase."],
-            ["demo3", "rdp", "vlnPE", "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", "rdp", "vlnPE", "Walk out of bathroom and stand on white bath mat."],
-            ["demo5", "rdp", "vlnPE", "Walk straight through the double wood doors, follow the red carpet straight to the next doorway and stop where the carpet splits off."]
+            ["demo1", '1', "rdp", "vlnPE", "Walk past the left side of the bed and stop in the doorway."],
+            ["demo2", '1', "rdp", "vlnPE", "Walk through the bathroom, past the sink and toilet. Stop in front of the counter with the two suitcase."],
+            ["demo3", '1', "rdp", "vlnPE", "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", '1', "rdp", "vlnPE", "Walk out of bathroom and stand on white bath mat."],
+            ["demo5", '1', "rdp", "vlnPE", "Walk straight through the double wood doors, follow the red carpet straight to the next doorway and stop where the carpet splits off."]
         ],
-        inputs=[scene_dropdown, model_dropdown, mode_dropdown, prompt_input],
+        inputs=[scene_dropdown, episode_dropdown, model_dropdown, mode_dropdown, prompt_input],
         label="Navigation Task Examples"
     )
+    
     submit_btn.click(
         fn=run_simulation,
-        inputs=[scene_dropdown, model_dropdown, mode_dropdown, prompt_input, history_state],
+        inputs=[scene_dropdown, episode_dropdown, model_dropdown, mode_dropdown, prompt_input, history_state],
         outputs=[video_output, history_state],
         queue=True,
         api_name="run_simulation"

backend_api.py

@@ -6,13 +6,19 @@ import json
 from typing import Optional
 from config import API_ENDPOINTS
 
-def submit_to_backend(scene: str, prompt: str, mode: str, model_type: str, user: str = "Gradio-user") -> dict:
+def submit_to_backend(scene: str, episode: str, prompt: str, mode: str, model_type: str, user: str = "Gradio-user") -> dict:
     job_id = str(uuid.uuid4())
+
+    scene_index = scene.split("_")[-1]
+    episode_index = episode.split("_")[-1]
+
     data = {
         "model_type": model_type,
         "instruction": prompt,
         "episode_type": scene,
         "mode": mode,
+        "scene_index": scene_index,
+        "episode_index": episode_index,
     }
     payload = {
         "user": user,