42 lines
1.0 KiB
Python
42 lines
1.0 KiB
Python
from contextlib import asynccontextmanager
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from app.core.database import init_db, Base
|
|
from app.api.books import router as books_router
|
|
from app.api.chat import router as chat_router
|
|
from app.api.models import router as models_router
|
|
|
|
from app.models.book import Book, Chapter, Segment
|
|
from app.models.chat import Conversation, Message
|
|
from app.models.ai_model import AiModel
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
await init_db()
|
|
yield
|
|
|
|
|
|
app = FastAPI(title="BookMate", version="0.1.0", lifespan=lifespan)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
app.include_router(books_router, prefix="/api")
|
|
app.include_router(chat_router, prefix="/api")
|
|
app.include_router(models_router, prefix="/api")
|
|
|
|
|
|
@app.get("/api/health")
|
|
async def health():
|
|
return {"status": "ok"} |