from pydantic import BaseModel
from fastapi import FastAPI


# Initialize FastAPI app
app = FastAPI(title="PDF Data Extraction")

class Item(BaseModel):
    name: str
    description: str = None

@app.post("/predict/")
async def predict(item: Item):
    return {"result": f"Vous avez envoyé : {item.name}"}

@app.get("/")
async def root():
    return {"message": "Bienvenue sur mon API !"}