Spaces:
Sleeping
Sleeping
File size: 1,928 Bytes
dfbb493 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 | from fastapi import FastAPI, HTTPException
import sys
import os
# Add src to path so standard imports work
sys.path.insert(0, os.path.dirname(__file__))
from env import SpectraQualEnv
from models import PCBAction, StepResult
app = FastAPI(
title="SpectraQual OpenEnv API",
description="REST API for automated OpenEnv space evaluation",
version="1.0.0"
)
# Initialize a default environment instance
# In a real deployed evaluator, they may instantiate isolated environments
# but for the "ping space URL" test, a global instance is standard.
env_instance = SpectraQualEnv(task_id="task_easy")
@app.get("/")
def health_check():
"""Returns 200 to pass automated ping test."""
return {"status": "ok", "environment": "SpectraQual"}
@app.post("/reset")
def reset_env() -> StepResult:
"""Reset the environment and return initial observation."""
try:
return env_instance.reset()
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.post("/step")
def step_env(action: PCBAction) -> StepResult:
"""Take a step in the environment."""
try:
if env_instance.state()["done"]:
# If done, returning an error or auto-resetting depends on the logic.
# Best practice: raise 400 that episode is done.
raise HTTPException(status_code=400, detail="Episode is done. Call /reset first.")
return env_instance.step(action)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.get("/state")
def get_state():
"""Return the internal state of the environment."""
try:
return env_instance.state()
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
if __name__ == "__main__":
import uvicorn
# Typically run via Docker CMD: uvicorn src.api:app --host 0.0.0.0 --port 7860
uvicorn.run(app, host="0.0.0.0", port=7860)
|