Add custom task input support and update URLs to HF Space

- reset() now accepts task="custom" with user-defined tasks, meetings, and energy
- Validates custom input (1-10 tasks, bounds clamping on all params)
- Existing easy/medium/hard scenarios unchanged
- Default URL updated from localhost to HF Space

client.py

@@ -23,7 +23,7 @@ class RhythmEnv(EnvClient[RhythmAction, RhythmObservation, RhythmState]):
     Client for the RhythmEnv Environment.
 
     Example:
-        >>> async with RhythmEnv(base_url="http://localhost:8000") as client:
+        >>> async with RhythmEnv(base_url="https://InosLihka-rhythm-env.hf.space") as client:
         ...     result = await client.reset(task="easy")
         ...     result = await client.step(RhythmAction(action_type=ActionType.START_TASK, task_id=0))
     """

inference.py

@@ -56,7 +56,7 @@ IMAGE_NAME = os.getenv("IMAGE_NAME")
 API_KEY = os.getenv("HF_TOKEN") or os.getenv("API_KEY")
 API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
 MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen2.5-72B-Instruct")
-BASE_URL = os.getenv("RHYTHM_ENV_URL", "http://localhost:8000")
+BASE_URL = os.getenv("RHYTHM_ENV_URL", "https://InosLihka-rhythm-env.hf.space")
 BENCHMARK = "rhythm_env"
 TASKS = ["easy", "medium", "hard"]
 MAX_STEPS = 20

server/rhythm_environment.py

@@ -253,10 +253,14 @@ class RhythmEnvironment(Environment):
         **kwargs: Any,
     ) -> RhythmObservation:
         task_name = kwargs.get("task", "easy")
-        if task_name not in TASK_CONFIGS:
-            task_name = "easy"
 
-        config = TASK_CONFIGS[task_name]
+        if task_name == "custom":
+            config = self._parse_custom_config(kwargs)
+        elif task_name in TASK_CONFIGS:
+            config = TASK_CONFIGS[task_name]
+        else:
+            task_name = "easy"
+            config = TASK_CONFIGS[task_name]
 
         # Deep-copy tasks so mutations don't affect the template
         self._tasks = [dict(t) for t in config["tasks"]]
@@ -499,6 +503,43 @@ class RhythmEnvironment(Environment):
                 return False
         return True
 
+    @staticmethod
+    def _parse_custom_config(kwargs: Dict[str, Any]) -> Dict[str, Any]:
+        """Parse and validate a custom task configuration from reset kwargs."""
+        raw_tasks = kwargs.get("tasks")
+        if not raw_tasks or not isinstance(raw_tasks, list):
+            raise ValueError("Custom mode requires a 'tasks' list with at least 1 task.")
+        if len(raw_tasks) > 10:
+            raise ValueError("Maximum 10 tasks allowed.")
+
+        tasks = []
+        for i, t in enumerate(raw_tasks):
+            if not isinstance(t, dict):
+                raise ValueError(f"Task {i} must be a dict.")
+            tasks.append({
+                "id": i,
+                "name": str(t.get("name", f"Task {i}")),
+                "description": str(t.get("description", "")),
+                "effort": max(0.05, min(1.0, float(t.get("effort", 0.3)))),
+                "progress": 0.0,
+                "deadline": max(1, min(MAX_STEPS, int(t.get("deadline", MAX_STEPS - 2)))),
+                "importance": max(0.1, min(1.0, float(t.get("importance", 0.5)))),
+            })
+
+        meetings = kwargs.get("meetings", [])
+        if not isinstance(meetings, list):
+            meetings = []
+        meetings = [int(m) for m in meetings if 0 <= int(m) < MAX_STEPS]
+
+        initial_energy = max(0.1, min(1.0, float(kwargs.get("initial_energy", 0.8))))
+
+        return {
+            "scenario": "Custom task configuration.",
+            "tasks": tasks,
+            "meetings": meetings,
+            "initial_energy": initial_energy,
+        }
+
     def _compute_mode(self) -> str:
         """Compute hidden internal mode (not exposed to agent)."""
         if (