app.py

from fastapi import FastAPI
from pydantic import BaseModel

app = FastAPI()

class NumberInput(BaseModel):
    number: float

@app.post("/square")
def compute_square(data: NumberInput):
    result = data.number ** 2
    return {"input": data.number, "square": result}