File size: 953 Bytes
61d9463
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from fastapi import FastAPI
import os
from backend.app.routes import users_route, events_route, travels_route
import asyncio
from backend.app.tasks.worker_tasks import run_embedding_worker_loop

# avoiding the conflictions
os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'

app = FastAPI()

@app.on_event("startup")
async def startup_event():
    print("Application startup...")
    # Schedule the background task for embedding worker loop
    asyncio.create_task(run_embedding_worker_loop(interval_seconds=30))

@app.on_event("shutdown")
async def shutdown_event():
    print("Application shutdown...")
    users_route.user_db.db.close()
    events_route.event_db.db.close()
    travels_route.travel_db.db.close()

# Include routers
app.include_router(events_route.router, prefix="/events", tags=["events"])
app.include_router(travels_route.router, prefix="/travels", tags=["travels"])
app.include_router(users_route.router, prefix="/users", tags=["users"])