main.py

from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware
from contextlib import asynccontextmanager
from app.api.routes import router
from app.core.config import REPORTS_DIR
from app.core.database import init_db
from app.core.logging_config import setup_logging, TracingMiddleware
from app.core.timeout_monitor import start_timeout_monitor, stop_timeout_monitor

# 1. Initialize enhanced logging system
setup_logging()

# 2. Lifespan context manager for startup/shutdown
@asynccontextmanager
async def lifespan(app: FastAPI):
    # Start timeout monitor on startup
    await start_timeout_monitor()
    yield
    # Cleanup on shutdown
    await stop_timeout_monitor()

app = FastAPI(title="AI Automation Factory", lifespan=lifespan)

# 3. Inject tracing middleware
app.add_middleware(TracingMiddleware)

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

init_db()

app.mount("/static", StaticFiles(directory="static"), name="static")
app.mount("/reports", StaticFiles(directory="reports"), name="reports")

app.include_router(router)

@app.get("/")
async def root():
    from fastapi.responses import FileResponse
    return FileResponse("static/index.html")

if __name__ == "__main__":
    import uvicorn
    # Use string import for reload compatibility
    uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)