Added Boss Fight scenario support to reset endpoint

server.py

@@ -77,11 +77,20 @@ def format_semantic_obs(env_instance):
 # Track the previous veto count to know if the CURRENT action was vetoed
 server_state = {"last_veto_count": 0}
 
+
+
+class ResetRequest(BaseModel):
+    scenario_id: int = 0  # Default to 0 (normal random episode)
+
 @app.post("/reset")
-def reset():
-    obs, info = env.reset()
-    server_state["last_veto_count"] = env.veto_count  # Reset our tracker
-    # FIXED: Pass the 'env' object directly
+def reset(req: ResetRequest = None):
+    # If no request body is sent, default to scenario 0
+    scenario = req.scenario_id if req else 0
+    
+    # Pass the scenario option to your original environment
+    obs, info = env.reset(options={"scenario": scenario})
+    
+    server_state["last_veto_count"] = env.veto_count
     return {"observation": format_semantic_obs(env)}
 
 @app.post("/step")