from fastapi import FastAPI, UploadFile, File, HTTPException
from fastapi.middleware.cors import CORSMiddleware

# Import routers
from image_route import router as image_enhance_router
from pdf_route import router as pdf_to_md_router

app = FastAPI(
    title="PDF Processing API",
    description="API for converting PDF to PNG and enhancing images",
    version="1.0.0"
)

# Add CORS middleware
app.add_middleware(
    CORSMiddleware,
    allow_origins=["https://readmytable.vercel.app"],  # Allows all origins
    allow_credentials=True,
    allow_methods=["*"],  # Allows all methods
    allow_headers=["*"],  # Allows all headers
)

# Include routers
app.include_router(image_enhance_router, prefix="/image", tags=["image"])
app.include_router(pdf_to_md_router, prefix="/pdf", tags=["pdf"])

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=7860)