app.py

# main.py
from fastapi import FastAPI
from pydantic import BaseModel
from inference import detect_page

app = FastAPI()

class DetectRequest(BaseModel):
    texts: list[str] = []
    images: list[str] = []

@app.get("/")
def root():
    return {"status": "running"}

@app.post("/detect")
def detect_endpoint(data: DetectRequest):
    return detect_page(data.texts, data.images)