from routes.distributions import router as distributions_router
from routes.ui import router as ui_router
from routes.parameters import router as parameters_router
from fastapi import FastAPI
from fastapi.middleware import Middleware
from middleware.validation import validate_json
from config.database import Base, engine
from routes.scenarios import router as scenarios_router
from routes.costs import router as costs_router
from routes.consumption import router as consumption_router
from routes.production import router as production_router
from routes.equipment import router as equipment_router
from routes.reporting import router as reporting_router
from routes.simulations import router as simulations_router
from routes.maintenance import router as maintenance_router

# Initialize database schema
Base.metadata.create_all(bind=engine)

app = FastAPI()
# Register validation middleware
app.middleware("http")(validate_json)

# Include API routers
app.include_router(scenarios_router)
app.include_router(parameters_router)
app.include_router(distributions_router)
app.include_router(costs_router)
app.include_router(consumption_router)
app.include_router(simulations_router)
app.include_router(production_router)
app.include_router(equipment_router)
app.include_router(maintenance_router)
app.include_router(reporting_router)
app.include_router(ui_router)