1 files changed, 6 insertions, 0 deletions
diff --git a/main.py b/main.py
index fdbdea9..65b8271 100644
--- a/main.py
+++ b/main.py
@@ -2,6 +2,7 @@ from contextlib import asynccontextmanager
 import asyncio
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
+from sqlalchemy import text
 from database import engine, Base
 from config import settings
 from routers import pantry, meals, recipes, menus, grocery
@@ -11,6 +12,11 @@ from routers import pantry, meals, recipes, menus, grocery
 async def lifespan(app: FastAPI):
     # Startup
     Base.metadata.create_all(bind=engine)
+    with engine.connect() as conn:
+        result = conn.execute(text("PRAGMA table_info(menu_plans)"))
+        if 'notes' not in [row[1] for row in result]:
+            conn.execute(text("ALTER TABLE menu_plans ADD COLUMN notes TEXT"))
+            conn.commit()
     yield
     # Shutdown
     pass