app/main.py

from contextlib import asynccontextmanager
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from fastapi.staticfiles import StaticFiles
from app.api.routes import router
from app.db.database import init_db
from custom_logger import logger_config as logger

@asynccontextmanager
async def lifespan(app: FastAPI):
    logger.info("="*60)
    logger.info("OCR Runner API Server Starting Up")
    logger.info("="*60)
    logger.info("Worker will start automatically on first upload")
    logger.info("Image files will be deleted after successful processing")
    logger.info("="*60)
    
    await init_db()
    yield
    logger.info("OCR Runner API Server Shutting Down")

app = FastAPI(title="OCR Runner API", version="2.0.0", lifespan=lifespan)

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

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