Upload folder using huggingface_hub

envs/warehouse_env/server/app.py

@@ -39,6 +39,51 @@ app = create_app(warehouse_env, WarehouseAction, WarehouseObservation, env_name=
 
 
 # Add custom render endpoints
+@app.post("/set-difficulty")
+async def set_difficulty(request: dict):
+    """Change the difficulty level and reset the environment."""
+    try:
+        difficulty = int(request.get("difficulty", 2))
+        if difficulty < 1 or difficulty > 5:
+            return JSONResponse(
+                status_code=400,
+                content={"error": "Difficulty must be between 1 and 5"}
+            )
+
+        # Recreate the warehouse environment with new difficulty
+        global warehouse_env
+        warehouse_env = WarehouseEnvironment(
+            difficulty_level=difficulty,
+            grid_width=None,
+            grid_height=None,
+            num_packages=None,
+            max_steps=None,
+            random_seed=None,
+        )
+
+        # Reset the environment
+        observation = warehouse_env.reset()
+
+        return JSONResponse(content={
+            "success": True,
+            "difficulty": difficulty,
+            "grid_size": (warehouse_env.grid_width, warehouse_env.grid_height),
+            "num_packages": warehouse_env.num_packages,
+            "max_steps": warehouse_env.max_steps,
+            "observation": {
+                "step_count": observation.step_count,
+                "packages_delivered": observation.packages_delivered,
+                "total_packages": observation.total_packages,
+                "robot_position": observation.robot_position,
+            }
+        })
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={"error": f"Failed to set difficulty: {str(e)}"}
+        )
+
+
 @app.get("/render")
 async def render():
     """Get ASCII visualization of warehouse state."""

envs/warehouse_env/server/demo.html

@@ -196,6 +196,7 @@
                 <option value="4">Level 4 - Hard (15×15, 5 packages)</option>
                 <option value="5">Level 5 - Expert (20×20, 8 packages)</option>
             </select>
+            <button class="btn-secondary" id="applyDifficulty">Apply</button>
         </div>
 
         <div class="stats">
@@ -419,6 +420,34 @@
         document.getElementById('stopBtn').addEventListener('click', stopAutoPlay);
         document.getElementById('resetBtn').addEventListener('click', reset);
         document.getElementById('resetManualBtn').addEventListener('click', reset);
+        document.getElementById('applyDifficulty').addEventListener('click', async () => {
+            stopAutoPlay();
+            const difficulty = parseInt(document.getElementById('difficulty').value);
+            showMessage(`Changing to difficulty level ${difficulty}...`, 'info');
+
+            try {
+                const response = await fetch(`${baseUrl}/set-difficulty`, {
+                    method: 'POST',
+                    headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify({ difficulty: difficulty })
+                });
+                const data = await response.json();
+
+                if (data.success) {
+                    showMessage(
+                        `Difficulty changed! Grid: ${data.grid_size[0]}×${data.grid_size[1]}, ` +
+                        `Packages: ${data.num_packages}, Max Steps: ${data.max_steps}`,
+                        'success'
+                    );
+                    updateStats(data.observation);
+                    await refreshVisualization();
+                } else {
+                    showMessage('Failed to change difficulty: ' + data.error, 'error');
+                }
+            } catch (error) {
+                showMessage('Error changing difficulty: ' + error.message, 'error');
+            }
+        });
 
         // Initialize on load
         reset();

server/app.py

@@ -39,6 +39,51 @@ app = create_app(warehouse_env, WarehouseAction, WarehouseObservation, env_name=
 
 
 # Add custom render endpoints
+@app.post("/set-difficulty")
+async def set_difficulty(request: dict):
+    """Change the difficulty level and reset the environment."""
+    try:
+        difficulty = int(request.get("difficulty", 2))
+        if difficulty < 1 or difficulty > 5:
+            return JSONResponse(
+                status_code=400,
+                content={"error": "Difficulty must be between 1 and 5"}
+            )
+
+        # Recreate the warehouse environment with new difficulty
+        global warehouse_env
+        warehouse_env = WarehouseEnvironment(
+            difficulty_level=difficulty,
+            grid_width=None,
+            grid_height=None,
+            num_packages=None,
+            max_steps=None,
+            random_seed=None,
+        )
+
+        # Reset the environment
+        observation = warehouse_env.reset()
+
+        return JSONResponse(content={
+            "success": True,
+            "difficulty": difficulty,
+            "grid_size": (warehouse_env.grid_width, warehouse_env.grid_height),
+            "num_packages": warehouse_env.num_packages,
+            "max_steps": warehouse_env.max_steps,
+            "observation": {
+                "step_count": observation.step_count,
+                "packages_delivered": observation.packages_delivered,
+                "total_packages": observation.total_packages,
+                "robot_position": observation.robot_position,
+            }
+        })
+    except Exception as e:
+        return JSONResponse(
+            status_code=500,
+            content={"error": f"Failed to set difficulty: {str(e)}"}
+        )
+
+
 @app.get("/render")
 async def render():
     """Get ASCII visualization of warehouse state."""

server/demo.html

@@ -196,6 +196,7 @@
                 <option value="4">Level 4 - Hard (15×15, 5 packages)</option>
                 <option value="5">Level 5 - Expert (20×20, 8 packages)</option>
             </select>
+            <button class="btn-secondary" id="applyDifficulty">Apply</button>
         </div>
 
         <div class="stats">
@@ -419,6 +420,34 @@
         document.getElementById('stopBtn').addEventListener('click', stopAutoPlay);
         document.getElementById('resetBtn').addEventListener('click', reset);
         document.getElementById('resetManualBtn').addEventListener('click', reset);
+        document.getElementById('applyDifficulty').addEventListener('click', async () => {
+            stopAutoPlay();
+            const difficulty = parseInt(document.getElementById('difficulty').value);
+            showMessage(`Changing to difficulty level ${difficulty}...`, 'info');
+
+            try {
+                const response = await fetch(`${baseUrl}/set-difficulty`, {
+                    method: 'POST',
+                    headers: { 'Content-Type': 'application/json' },
+                    body: JSON.stringify({ difficulty: difficulty })
+                });
+                const data = await response.json();
+
+                if (data.success) {
+                    showMessage(
+                        `Difficulty changed! Grid: ${data.grid_size[0]}×${data.grid_size[1]}, ` +
+                        `Packages: ${data.num_packages}, Max Steps: ${data.max_steps}`,
+                        'success'
+                    );
+                    updateStats(data.observation);
+                    await refreshVisualization();
+                } else {
+                    showMessage('Failed to change difficulty: ' + data.error, 'error');
+                }
+            } catch (error) {
+                showMessage('Error changing difficulty: ' + error.message, 'error');
+            }
+        });
 
         // Initialize on load
         reset();