Changes to expose end points

app.py

@@ -1,56 +1,115 @@
 """
-app.py — quick local episode runner using the heuristic baseline agent.
+HF Spaces server — DebugOps Environment (OpenEnv compatible)
 
-Usage
-    python app.py
-
-This runs a single episode of the Simple task using the rule-based agent
-and prints a per-step log. It is intentionally lightweight and dependency-free
-(no LLM API key required) so it works out of the box.
+Exposes:
+  POST /reset
+  POST /step
+  GET  /state
 """
+
 from __future__ import annotations
-from typing import Any, Dict, List
+from typing import Dict, Any, Optional
+
+import os
+import uvicorn
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+
+# Import your environments
+from tasks.task_simple import create_env as create_simple
+from tasks.task_multi_service import create_env as create_multi
+from tasks.task_critical import create_env as create_critical
+
+
+app = FastAPI(title="DebugOps AI Environment", version="1.0.0")
+
+# Global env instance
+_env = None
+
+class ResetRequest(BaseModel):
+    task: str = "simple"   # simple | multi_service | critical
+
+
+class StepRequest(BaseModel):
+    action: str
+
+def create_env(task: str):
+    if task == "simple":
+        return create_simple()
+    elif task == "multi_service":
+        return create_multi()
+    elif task == "critical":
+        return create_critical()
+    else:
+        raise ValueError(f"Invalid task: {task}")
+
+
+def get_env():
+    global _env
+    if _env is None:
+        raise HTTPException(status_code=400, detail="Call /reset first")
+    return _env
+
+@app.get("/")
+def root():
+    return {
+        "name": "DebugOps AI Environment",
+        "description": "Production debugging RL environment (OpenEnv compatible)",
+        "endpoints": ["/reset", "/step", "/state", "/health"],
+    }
+
+
+@app.get("/health")
+def health():
+    return {"status": "ok"}
+
+
+@app.post("/reset")
+def reset(request: ResetRequest):
+    global _env
+    try:
+        _env = create_env(request.task)
+        obs = _env.reset()
 
-from tasks.task_simple import create_env
-from agent.baseline import act, reset_history
+        return {
+            "observation": obs,
+            "done": False,
+        }
 
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
-def run_episode() -> List[Dict[str, Any]]:
-    env   = create_env()
-    state = env.reset()
-    reset_history()
 
-    logs: List[Dict[str, Any]] = []
-    done  = False
-    step  = 0
+@app.post("/step")
+def step(request: StepRequest):
+    env = get_env()
 
-    print("[app.py] Starting simple-task episode with heuristic baseline agent.")
+    try:
+        obs, reward, done, info = env.step(request.action)
 
-    while not done:
-        action = act(state)
-        next_state, reward, done, info = env.step(action)
+        return {
+            "observation": obs,
+            "reward": float(round(reward, 3)),
+            "done": done,
+            "info": info,
+        }
 
-        logs.append({
-            "step":   step,
-            "state":  state,
-            "action": action,
-            "reward": round(reward, 3),
-            "info":   info,
-        })
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=str(e))
 
-        print(
-            f"  step={step} action={action:<15} reward={round(reward, 3):>8.3f} "
-            f"resolved={str(info.get('resolved', False)).lower()}"
-        )
 
-        state = next_state
-        step += 1
+@app.get("/state")
+def state():
+    env = get_env()
+    try:
+        return env.state()
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=str(e))
 
-    success = any(e["info"].get("resolved", False) for e in logs)
-    print(f"\n[app.py] Episode finished — steps={step} success={str(success).lower()}")
-    return logs
+def main():
+    port = int(os.getenv("PORT", 7860))
+    uvicorn.run(app, host="0.0.0.0", port=port)
 
 
 if __name__ == "__main__":
-    logs = run_episode()
-    print(f"Steps: {len(logs)}")
+    main()