main.py

# main.py
from fastapi import FastAPI
from routes import router as trainer_router
from auth import router as auth_router
from extraction_routes import router as extraction_router  # NEW IMPORT

app = FastAPI(title="LongTrainer Bot API with Auth and Custom Prompts")

app.include_router(trainer_router)
app.include_router(auth_router)
app.include_router(extraction_router, prefix="/extract")  # NEW LINE

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