Spaces:
Running
Running
Your Name commited on
Commit ·
5e99860
1
Parent(s): 65111eb
Auto-initialize database tables on startup
Browse files- 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")
|