from fastapi import FastAPI
from pydantic import BaseModel

app = FastAPI(title="Grid2Op OpenEnv API")

# -------------------------------
# Request schema
# -------------------------------
class ScenarioRequest(BaseModel):
    input_root: str
    output_dir: str


# -------------------------------
# Health check
# -------------------------------
@app.get("/")
def home():
    return {"message": "Grid2Op API is running 🚀"}


# -------------------------------
# Run Scenario Curriculum
# -------------------------------
@app.post("/run-scenario")
def run_scenario(req: ScenarioRequest):
    try:
        from scenario_curriculum import build_scenario_curriculum

        outputs = build_scenario_curriculum(
            req.input_root,
            req.output_dir,
            with_plots=False
        )

        return {
            "status": "success",
            "outputs": str(outputs)
        }

    except Exception as e:
        return {
            "status": "error",
            "message": str(e)
        }