Needed to update recommender for season and episode props

app.py

@@ -14,6 +14,7 @@ from zoneinfo import ZoneInfo
 from sentence_transformers import SentenceTransformer
 from apscheduler.schedulers.background import BackgroundScheduler
 import math
+from contextlib import asynccontextmanager # Import this for lifespan
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -22,17 +23,6 @@ logger = logging.getLogger(__name__)
 # Load environment variables
 load_dotenv()
 
-# FastAPI app setup
-app = FastAPI()
-
-app.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_credentials=True,
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
 # Configuration
 SEED = 42
 random.seed(SEED)
@@ -120,7 +110,7 @@ class Recommender:
         self.reply_weight = reply_weight
         self.reply_like_weight = reply_like_weight
 
-    async def fetch_existing_post_ids() -> set:
+    async def fetch_existing_post_ids(self) -> set:
         supabase = get_supabase_client()
         page_size = 1000
         page = 0
@@ -345,6 +335,32 @@ def sync_background_update():
     loop.run_until_complete(background_update())
     loop.close()
 
+# Lifespan context manager
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    # Startup event
+    logger.info("Starting up application...")
+    await recommender.update_data()
+    scheduler.add_job(sync_background_update, 'interval', seconds=UPDATE_INTERVAL)
+    scheduler.start()
+    yield
+    # Shutdown event
+    logger.info("Shutting down application...")
+    scheduler.shutdown()
+    logger.info("Scheduler shut down")
+
+# FastAPI app setup with lifespan
+app = FastAPI(lifespan=lifespan)
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+
 @app.get("/recommend/feed")
 async def get_recommendations_handler(user_id: str = Query(...)):
     try:
@@ -365,17 +381,6 @@ async def get_recommendations_handler(user_id: str = Query(...)):
     except Exception as e:
         logger.error(f"Error generating recommendations: {str(e)}")
         return {"status": "error", "message": str(e)}
-    
-@app.on_event("startup")
-async def startup_event():
-    await recommender.update_data()
-    scheduler.add_job(sync_background_update, 'interval', seconds=UPDATE_INTERVAL)
-    scheduler.start()
-
-@app.on_event("shutdown")
-async def shutdown_event():
-    scheduler.shutdown()
-    logger.info("Scheduler shut down")
 
 @app.get("/")
 async def health_check():