Update app/db.py

app/db.py

@@ -101,3 +101,11 @@ async def get_async_session() -> AsyncGenerator[AsyncSession, None]:
 
 async def get_user_db(session: AsyncSession = Depends(get_async_session)):
     yield SQLAlchemyUserDatabase(session, User)
+
+
+async def connect() -> None:
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.create_all, checkfirst=True)
+async def disconnect() -> None:
+    if engine:
+        await engine.dispose()