import logging
import os
from fastapi import APIRouter, Request, Response

# Import des sous-routeurs
from api import prediction, management 

logger = logging.getLogger(__name__)

# Routeur principal
router = APIRouter()

# --- Suppression du Middleware --- 
# Le middleware doit être appliqué à l'instance FastAPI (app) dans main.py

# --- Inclusion des sous-routeurs ---
router.include_router(prediction.router)       # Routes de prédiction (/batch_predict)
router.include_router(management.router)       # Routes de gestion (/manage/load_model, /manage/update_model)
# router.include_router(management.public_router) # Routes d'info publiques (/models, /models/{id}) - Supprimé par l'utilisateur