# uvicorn app:app --reload

from fastapi import Body, FastAPI, UploadFile, File
from fastapi.responses import JSONResponse
from fastapi.middleware.cors import CORSMiddleware
from detect import DengueDetector
import traceback

# Inicializar detector e preditor
detector = DengueDetector()
app = FastAPI()

# --- CORS ---
origins = ["https://previdengue.vercel.app", "*"]
app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"]
)

# --- Rotas ---
@app.get("/")
def health_check():
    return {"status": "ok", "message": "API de Dengue rodando!"}

# --- Rota de detecção ---
@app.post("/detect/")
async def detect(file: UploadFile = File(...)):
    try:
        content = await file.read()
        result = detector.detect_image(content)  
        return JSONResponse(content=result)
    except Exception as e:
        return JSONResponse(status_code=500, content={"error": str(e)})