Your Name commited on
Commit
5e99860
·
1 Parent(s): 65111eb

Auto-initialize database tables on startup

Browse files
Files changed (1) hide show
  1. backend/main.py +5 -0
backend/main.py CHANGED
@@ -22,6 +22,11 @@ async def lifespan(app: FastAPI):
22
  settings = get_settings()
23
  logging.basicConfig(level=settings.log_level)
24
  logger.info("Starting AutoForge backend")
 
 
 
 
 
25
  yield
26
  await engine.dispose()
27
  logger.info("AutoForge backend stopped")
 
22
  settings = get_settings()
23
  logging.basicConfig(level=settings.log_level)
24
  logger.info("Starting AutoForge backend")
25
+
26
+ # Automatically initialize database schema for Hugging Face Spaces
27
+ async with engine.begin() as conn:
28
+ await conn.run_sync(Base.metadata.create_all)
29
+
30
  yield
31
  await engine.dispose()
32
  logger.info("AutoForge backend stopped")